shard-shaped tensor to be consumed by later manually-partitioned ops. If the
shape is not evenly partitionable, the padding region will be masked with 0s.
The conversion can happen partially in subgroups, by specifying the dim
attribute, where only that dim will be converted.
  }];

  let arguments = (ins
    TF_Tensor:$input,

    StrAttr:$manual_sharding,