Searched defs:torch_func (Results 1 – 2 of 2) sorted by relevance
67 def deco_binary_ufunc(torch_func): argument289 def deco_unary_ufunc(torch_func): argument
1696 def _test_reduction_function_with_numpy(self, torch_func, np_func, device, dtype, argument