code: def partial_tile(
      tensor,
      tile_assignment,
      use_sharding_op=False,
      unspecified_dims=None,
  ):
      """Returns a tensor that has tiled sharding.
  
      Args:
        tensor: A tf.Tensor to shard.
        tile_assignment: An np.ndarray describing the topology of the tiling and
          which device will compute which part of the topology. It must have one
          more dimension than tensor, and the last dimension represents partially
          replicated tiles.
        use_sharding_op: If true, adds a sharding op to set the sharding.
        unspecified_dims: An optional list of dimensions unspecified.
      """
      return Sharding.partial_tile(tile_assignment).apply_to_tensor(
          tensor,
          use_sharding_op=use_sharding_op,
          unspecified_dims=unspecified_dims or [],
      )
  
imports: import numpy
  from tensorflow.compiler.tf2xla.python import xla
  from local_xla.xla import xla_data_pb2
  from tensorflow.core.framework import attr_value_pb2
label_desc: Use image node_id 6 for calling a global function with example usage: partial_tile(tensor, tile_assignment, use_sharding_op, unspecified_dims) and returns: Sharding