Searched refs:_test_export_same_as_eager (Results 1 – 2 of 2) sorted by relevance
/aosp_15_r20/external/pytorch/test/export/ |
H A D | test_torchbind.py | 131 def _test_export_same_as_eager( member in TestExportTorchbind 172 ep = self._test_export_same_as_eager( 222 ep = self._test_export_same_as_eager( 256 ep = self._test_export_same_as_eager( 291 ep = self._test_export_same_as_eager( 336 self._test_export_same_as_eager( 346 ep = self._test_export_same_as_eager( 398 self._test_export_same_as_eager( 413 self._test_export_same_as_eager( 430 ep = self._test_export_same_as_eager( [all …]
|
H A D | test_export.py | 285 def _test_export_same_as_eager(self, f, args, kwargs=None): member in TestExport 348 self._test_export_same_as_eager(f, inp) 2654 self._test_export_same_as_eager(list_tensor_map, inps) 2694 self._test_export_same_as_eager(kw_func, args, kwargs) 2707 self._test_export_same_as_eager(kw_func, args, kwargs) 2725 self._test_export_same_as_eager(kw_func, args, kwargs1) 2726 self._test_export_same_as_eager(kw_func, args, kwargs2) 2728 self._test_export_same_as_eager(kw_func2, args, kwargs3) 2737 self._test_export_same_as_eager(kw_func, args) 2747 self._test_export_same_as_eager(kw_func, args, kwargs) [all …]
|