Searched defs:numpy_fn (Results 1 – 3 of 3) sorted by relevance
59 numpy_fn, argument105 def _TestGradient(self, nccl_reduce, numpy_fn): argument
238 numpy_fn, argument
2650 def test_fn(torch_fn, numpy_fn, steps): argument