tf.raw_ops.CollectiveReduceV3

Mutually reduces multiple tensors of identical type and shape.

View aliases

Compat aliases for migration

See Migration guide for more details.

tf.compat.v1.raw_ops.CollectiveReduceV3

tf.raw_ops.CollectiveReduceV3(
    input,
    communicator,
    group_assignment,
    reduction,
    timeout_seconds=0,
    name=None
)

Args
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.
reduction	A string from: "Min", "Max", "Mul", "Add".
timeout_seconds	An optional float. Defaults to 0.
name	A name for the operation (optional).

Returns
A Tensor. Has the same type as input.