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 …]
|