Searched refs:test_inference_mode (Results 1 – 3 of 3) sorted by relevance
/aosp_15_r20/external/pytorch/test/ | ||
H A D | test_fake_tensor.py | 1872 def test_inference_mode(self): member in FakeTensorDispatchCache |
/aosp_15_r20/external/pytorch/test/functorch/ | ||
H A D | test_aotdispatch.py | 3542 def test_inference_mode(self): member in TestAOTAutograd |
/aosp_15_r20/external/pytorch/test/dynamo/ | ||
H A D | test_misc.py | 5905 def test_inference_mode(self): member in MiscTests |