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
)
Args |
|---|
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).
|
Returns |
|---|
A Tensor. Has the same type as input.
|
