Searched defs:functional_call (Results 1 – 6 of 6) 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=''): argument
98 def test_functional_call(self, functional_call): argument
106 def test_functional_call_with_jit(self, functional_call): argument
128 def test_functional_call_with_data_parallel(self, functional_call): argument
139 def test_functional_call_with_data_parallel_error(self, functional_call): argument
153 def test_functional_call_with_gradient(self, functional_call): argument
177 def test_functional_batch_norm(self, functional_call): argument
198 def test_circular_references(self, functional_call): argument
223 def test_reparametrized_module_change_parametrization_original(self, functional_call): argument
245 def test_reparametrize_module_fail_reset_to_original(self, functional_call): argument
[all …]
|
| H A D | test_proxy_tensor.py | 1651 def functional_call(*args, **kwargs): function
|
| /aosp_15_r20/external/pytorch/torch/_functorch/ |
| H A D | functional_call.py | 12 def functional_call( function
|
| H A D | aot_autograd.py | 909 def functional_call(named_params, named_buffers, *args, **kwargs): function
|
| /aosp_15_r20/external/pytorch/torch/nn/utils/ |
| H A D | stateless.py | 186 def functional_call( function
|
| /aosp_15_r20/external/pytorch/torch/_functorch/_aot_autograd/ |
| H A D | traced_function_transforms.py | 848 def functional_call(*args, **kwargs): function
|