Searched refs:broadcast_recv (Results 1 – 4 of 4) sorted by relevance
4 name: "collective.broadcast_recv"
137 def broadcast_recv(shape, dtype, group_size, group_key, instance_key): function
103 out1 = collective_ops.broadcast_recv(c1.shape, c1.dtype,
289 return collective_ops.broadcast_recv(