Mutually reduces multiple tensors of identical type and shape and scatters the result.
tf.raw_ops.CollectiveReduceScatterV2(
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 .
|