Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/test/jit/
H A Dtest_await.py44 script_out = sm(inp)
45 self.assertTrue(torch.allclose(torch.eye(2) + 102, script_out))
46 self.assertTrue(torch.allclose(script_out, out))
59 script_out = sm(inp)
60 self.assertTrue(torch.allclose(torch.eye(2) + 13, script_out))
61 self.assertTrue(torch.allclose(script_out, out))
83 script_out = sm(inp)
84 self.assertTrue(torch.allclose(torch.eye(2), script_out))
85 self.assertTrue(torch.allclose(script_out, out))
112 script_out = sm(inp)
[all …]
H A Dtest_list_dict.py1739 script_out = scripted_func(self.dict())
1741 self.assertEqual(len(eager_out), len(script_out))
1743 self.assertTrue(item in script_out)
1754 script_out = torch.jit.script(fn)(self.dict(), *args)
1755 self.assertEqual(eager_out, script_out)
1836 script_out = popitem(self.dict())
1839 self.assertEqual(len(eager_out[1]), len(script_out[1]))
1842 self.assertTrue(isinstance(script_out[0][0], str))
1843 self.assertTrue(isinstance(script_out[0][1], torch.Tensor))
2415 script_out = fn(script_input_dict)
[all …]
H A Dtest_recursive_script.py518 script_out = sm(t)
519 self.assertNotEqual(eager_out, script_out)
552 script_out = sm(t.clone())
553 self.assertNotEqual(eager_out, script_out)
/aosp_15_r20/external/harfbuzz_ng/src/
H A Dhb-ot-tag.cc575 hb_script_t script_out = hb_ot_tag_to_script (script_tag); in hb_ot_tags_to_script_and_language() local
577 *script = script_out; in hb_ot_tags_to_script_and_language()
582 hb_ot_tags_from_script_and_language (script_out, in hb_ot_tags_to_script_and_language()
/aosp_15_r20/external/pytorch/torch/testing/_internal/
H A Djit_utils.py638 script_out = sm(*args)
640 self.assertEqual(eager_out, script_out)
/aosp_15_r20/external/pytorch/test/
H A Dtest_jit.py12340 script_out = scripted_fn()
12341 for a, b in zip(eager_out, script_out):
14110 script_out = self.runAndSaveRNG(cu.test, args_fn())
14112 self.assertEqual(script_out, eager_out)
14137 script_out = cu.test(torch.ones(2, 2), g)
14147 self.assertEqual(script_out, eager_out)
14847 script_out = self.runAndSaveRNG(lambda: M()(*inputs), ())[0]
14849 self.assertEqual(eager_out, script_out)
14864 script_out = self.runAndSaveRNG(lambda x: S()(x), (input,))[0]
14866 self.assertEqual(eager_out, script_out)
H A Dtest_fx.py877 script_out = scripted_lowered(x)
878 torch.testing.assert_close(script_out, ref_out)
/aosp_15_r20/external/pytorch/test/quantization/fx/
H A Dtest_quantize_fx.py9310 script_out = script(input_value)