tf.raw_ops.CollectiveGatherV2

Mutually accumulates multiple tensors of identical type and shape.

Args
input A Tensor. Must be one of the following types: 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.
communication_hint An optional string. Defaults to "auto".
timeout_seconds An optional float. Defaults to 0.
name A name for the operation (optional).
Returns
A Tensor. Has the same type as input.

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