Lines Matching refs:scripted

252         scripted = torch.jit.script(m)
256         ).check_next("prim::EnumValue").check_next("return").run(str(scripted.graph))
258         self.assertEqual(scripted(), Color.RED.value)
275         scripted = torch.jit.script(m)
277         self.assertEqual(scripted(), (Color.RED.name, Color.RED.value))
311         scripted = torch.jit.script(m)
315         ).check_next("return").run(str(scripted.graph))
317         self.assertEqual(scripted(), Color.RED)
333         scripted = torch.jit.script(iterate_enum)
337         ).check_same("Color.GREEN").check_same("Color.BLUE").run(str(scripted.graph))
340         self.assertEqual(scripted(Color.RED), [Color.GREEN.value, Color.BLUE.value])
341         self.assertEqual(scripted(Color.GREEN), [Color.RED.value, Color.BLUE.value])