Mutually reduces multiple tensors of identical type and shape.
View aliases
Compat aliases for migration
See Migration guide for more details.
tf.raw_ops.CollectiveReduceV2(
input, group_size, group_key, instance_key, ordering_token, merge_op, final_op,
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_
|
A Tensor of type int32 .
|
group_
|
A Tensor of type int32 .
|
instance_
|
A Tensor of type int32 .
|
ordering_
|
A list of Tensor objects with type resource .
|
merge_
|
A string from: "Min", .
|
final_
|
A string from: "Id", .
|
communication_
|
An optional string . Defaults to "auto" .
|
timeout_
|
An optional float . Defaults to 0 .
|
name
|
A name for the operation (optional). |
Returns | |
---|---|
A Tensor . Has the same type as input .
|