Receives a tensor value broadcast from another device.
tf.raw_ops.CollectiveBcastRecv(
T, group_size, group_key, instance_key, shape,
communication_hint='auto', timeout_seconds=0, name=None
)
Args |
|---|
T
|
A tf.DType from: tf.bool, tf.float32, tf.half, tf.float64, tf.int32, tf.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).
|
Returns |
|---|
A Tensor of type T.
|
