An op that demultiplexes a tensor to be sharded by XLA to a list of partitioned
tf.raw_ops.TPUPartitionedOutput(
inputs, num_splits, partition_dim=0, name=None
)
outputs outside the XLA computation.
Returns | |
---|---|
A list of num_splits Tensor objects with the same type as inputs .
|