Searched defs:broadcast_recv (Results 1 – 2 of 2) sorted by relevance
59 broadcast_recv = _collective_ops.broadcast_recv variable in CollectiveOpsV190 def broadcast_recv(shape, dtype, group_size, group_key, instance_key, *args, member in CollectiveOpsV2
346 def broadcast_recv(shape, function