W3cubDocs

/TensorFlow 2.4

tf.raw_ops.CollectiveBcastRecv

Receives a tensor value broadcast from another device.

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.

© 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/CollectiveBcastRecv