631f99ebc3b7e44f5b86c2313d789e7df7258a7a,tensorflow/compiler/xla/experimental/xla_sharding/xla_sharding.py,,partial_tile,#Any#Any#Any#,316

Before Change

      replicated tiles.
    use_sharding_op: If true, adds a sharding op to set the sharding.
  
  if use_sharding_op:
    tensor = tf2xla.sharding(tensor)
  Sharding.partial_tile(tile_assignment).apply_to_tensor(tensor)
  return tensor


def get_op_sharding(op):
  Returns sharding attribute of an op.

After Change

      replicated tiles.
    use_sharding_op: If true, adds a sharding op to set the sharding.
  
  return Sharding.partial_tile(tile_assignment).apply_to_tensor(
      tensor, use_sharding_op=use_sharding_op)


def get_op_sharding(op):
  Returns sharding attribute of an op.

Instances