Searched defs:torchfn (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/pytorch/test/xpu/ |
| H A D | test_gemm.py | 903 def fn(torchfn, *args, test_out=False, **kwargs): argument
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_reductions.py | 660 def _testCSelection(self, torchfn, mathfn): argument
1151 def _test_minmax_helper(self, torchfn, reffn, device, dtype, skip_indices=False): argument
|
| H A D | test_binary_ufuncs.py | 2984 def _test_cop(self, torchfn, mathfn, dtype, device): argument
|
| H A D | test_linalg.py | 5076 def fn(torchfn, *args, test_out=False, **kwargs): argument
8243 def fn(torchfn, *args): argument
|
| H A D | test_torch.py | 1113 def torchfn(t1, t2, t3): function
|