tf.raw_ops.CollectiveReduce

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.CollectiveReduce

tf.raw_ops.CollectiveReduce(
    input, group_size, group_key, instance_key, merge_op, final_op, subdiv_offsets,
    wait_for=[], communication_hint='auto', timeout_seconds=0, name=None
)

Args
input	A Tensor. Must be one of the following types: float32, half, float64, int32, int64.
group_size	An int.
group_key	An int.
instance_key	An int.
merge_op	A string from: "Min", "Max", "Mul", "Add".
final_op	A string from: "Id", "Div".
subdiv_offsets	A list of ints.
wait_for	An optional list of ints. Defaults to [].
communication_hint	An optional string. Defaults to "auto".
timeout_seconds	An optional float. Defaults to 0.
name	A name for the operation (optional).

Returns
A Tensor. Has the same type as input.