Searched refs:_run_call_with_mock_module (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_stateless.py | 41 def _run_call_with_mock_module(self, module, functional_call, device='cpu', prefix=''): member in TestStatelessFunctionalAPI
100 self._run_call_with_mock_module(module, functional_call)
113 self._run_call_with_mock_module(jit_module, functional_call)
120 self._run_call_with_mock_module(traced_module, functional_call)
132 self._run_call_with_mock_module(dp_module, functional_call, device='cuda', prefix='module')
|