An op that demultiplexes a tensor to be sharded by XLA to a list of partitioned
View aliases
Compat aliases for migration
See Migration guide for more details.
tf.raw_ops.TPUPartitionedOutputV2(
inputs, num_splits, partition_dims, name=None
)
outputs outside the XLA computation. Supports ND sharding.
Returns | |
---|---|
A list of num_ Tensor objects with the same type as inputs .
|