Searched defs:all_reduce_coalesced (Results 1 – 3 of 3) sorted by relevance
144 std::vector<at::Tensor> all_reduce_coalesced( in all_reduce_coalesced() function
332 def all_reduce_coalesced( function
2516 def all_reduce_coalesced(tensors, op=ReduceOp.SUM, group=None, async_op=False): function