Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/test/jit/
H A Dtest_list_dict.py2627 def _compare_eager_and_script(self, fn, input_list, script_input_list=None): argument
2639 script_input_list = script_input_list or torch.jit.script(input_list)
2651 script_out = fn(script_input_list)
2667 self.assertEqual(input_list, script_input_list)