code: def broadcast_recv_v2(
      shape,
      dtype,
      group_size,
      group_key,
      instance_key,
      communication_hint="auto",
      timeout=0,
  ):
      """Receives a broadcasts tensor, across devices.
  
      Args:
        shape: an int tensor.  Shape of the tensor to be received.
        dtype: Type of the tensor to be received.
        group_size: an int32 tensor.  One plus the number of receiving tensors, i.e.
            the total number of devices participating.  Each tensor must reside on a
            different device.
        group_key: an int32 tensor identifying the group of devices.
        instance_key: an int32 tensor identifying the participating group of Ops.
        communication_hint: preferred collective communication.  The implementation
          may fall back to another mechanism.  Options include `auto`, `ring`, and
          `nccl`.
        timeout: If set to a non zero, set a completion timeout to detect staleness.
          If the timer goes off, a DeadlineExceededError is raised.
          The timeout value in seconds. This feature is experimental.
  
      Returns:
        An Op implementing the broadcast receive.
      """
      return gen_collective_ops.collective_bcast_recv_v2(
          T=dtype,
          group_size=group_size,
          group_key=group_key,
          instance_key=instance_key,
          shape=shape,
          communication_hint=communication_hint.lower(),
          timeout_seconds=timeout,
      )
  
file_path: tensorflow-upstream/tensorflow/python/ops/collective_ops.py
imports: from tensorflow.python.ops import gen_collective_ops
label_desc: Use image node_id 9 for calling a global function with example usage: broadcast_recv_v2(shape, dtype, group_size, group_key, instance_key, communication_hint, timeout) and returns: gen_collective_ops