tf.raw_ops.CollectiveReduce

Mutually reduces multiple tensors of identical type and shape.

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
)

input A Tensor. Must be one of the following types: bfloat16, 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).

A Tensor. Has the same type as input.