Docs4dev
Docs
tensorflow / 2.9 / raw_ops / collectivereducev2.html

tf.raw_ops.CollectiveReduceV2

Mutually reduces multiple tensors of identical type and shape.

tf.raw_ops.CollectiveReduceV2(
    input,
    group_size,
    group_key,
    instance_key,
    ordering_token,
    merge_op,
    final_op,
    communication_hint='auto',
    timeout_seconds=0,
    max_subdivs_per_device=-1,
    name=None
)
Args
input A Tensor. Must be one of the following types: bfloat16, float32, half, float64, int32, int64.
group_size A Tensor of type int32.
group_key A Tensor of type int32.
instance_key A Tensor of type int32.
ordering_token A list of Tensor objects with type resource.
merge_op A string from: "Min", "Max", "Mul", "Add".
final_op A string from: "Id", "Div".
communication_hint An optional string. Defaults to "auto".
timeout_seconds An optional float. Defaults to 0.
max_subdivs_per_device An optional int. Defaults to -1.
name A name for the operation (optional).
Returns
A Tensor. Has the same type as input.

© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.9/api_docs/python/tf/raw_ops/CollectiveReduceV2