Home
last modified time | relevance | path

Searched refs:test_inference_mode (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/test/
H A Dtest_fake_tensor.py1872 def test_inference_mode(self): member in FakeTensorDispatchCache
/aosp_15_r20/external/pytorch/test/functorch/
H A Dtest_aotdispatch.py3542 def test_inference_mode(self): member in TestAOTAutograd
/aosp_15_r20/external/pytorch/test/dynamo/
H A Dtest_misc.py5905 def test_inference_mode(self): member in MiscTests