ML Community Day is November 9! Join us for updates from TensorFlow, JAX, and more Learn more

tf.raw_ops.CollectiveAllToAllV3

Mutually exchanges multiple tensors of identical type and shape.

input A Tensor. Must be one of the following types: bfloat16, float32, half, float64, int32, int64.
communicator A Tensor of type resource.
group_assignment A Tensor of type int32.
timeout_seconds An optional float. Defaults to 0.
name A name for the operation (optional).

A Tensor. Has the same type as input.