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