Searched refs:test_multiple_outputs (Results 1 – 4 of 4) sorted by relevance
/aosp_15_r20/external/pytorch/test/ | ||
H A D | test_tensorexpr.py | 1652 def test_multiple_outputs(self): member in TestTensorExprFuser |
H A D | test_legacy_vmap.py | 90 def test_multiple_outputs(self): member in TestVmapAPILegacy |
/aosp_15_r20/external/pytorch/test/functorch/ | ||
H A D | test_eager_transforms.py | 2614 def test_multiple_outputs(self, device): member in TestJvp |
H A D | test_vmap.py | 174 def test_multiple_outputs(self): member in TestVmapAPI |