Home
last modified time | relevance | path

Searched refs:_test_export_same_as_eager (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/test/export/
H A Dtest_torchbind.py131 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 Dtest_export.py285 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 …]