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',
    timeout_seconds=0,
    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". timeout_seconds An optional float. Defaults to 0. name A name for the operation (optional).