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', 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".
  • name: A name for the operation (optional).

Returns:

A Tensor. Has the same type as input.