Searched defs:nccl_reduce (Results 1 – 1 of 1) sorted by relevance
61 nccl_reduce, argument108 def _TestGradient(self, nccl_reduce, numpy_fn): argument