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,
is_stateless=False,
max_subdivs_per_device=-1,
name=None
)
is_stateless
means each op does not need control dependencies to other
collective ops. In this case, keys that are unique at runtime
(e.g. instance_key
) should be used to distinguish collective groups.
Returns | |
---|---|
A Tensor . Has the same type as input .
|