Searched defs:nccl_fun (Results 1 – 1 of 1) sorted by relevance
37 def _NcclAllReduce(nccl_fun, tensors, devices): argument41 def _NcclReduce(nccl_fun, tensors, devices): argument