Searched refs:_assert_module_parameters_and_buffer_are (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_modules.py | 28 def _assert_module_parameters_and_buffer_are(self, module, device, dtype): member in TestModule
82 self._assert_module_parameters_and_buffer_are(m, device, dtype)
140 self._assert_module_parameters_and_buffer_are(m, device, dtype)
165 self._assert_module_parameters_and_buffer_are(m, device, dtype)
172 self._assert_module_parameters_and_buffer_are(m, "cpu", dtype)
177 self._assert_module_parameters_and_buffer_are(m, device, dtype)
196 self._assert_module_parameters_and_buffer_are(m, torch.device("cuda:1"), dtype)
|