tf.raw_ops.CollectiveGather

Mutually accumulates multiple tensors of identical type and shape.

tf.raw_ops.CollectiveGather(
    input, group_size, group_key, instance_key, shape, communication_hint='auto',
    name=None
)

input A Tensor. Must be one of the following types: float32, half, float64, int32, int64.
group_size An int.
group_key An int.
instance_key An int.
shape A tf.TensorShape or list of ints.
communication_hint An optional string. Defaults to "auto".
name A name for the operation (optional).

A Tensor. Has the same type as input.