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