Lines Matching refs:scripted

219         scripted = self.checkScript(fused_kernel, (a, a))
220         graph = scripted.graph_for(a, a)
231         self.assertEqual(scripted(a, a), fused_kernel(a, a))
241             scripted = self.checkScript(func, (a,))
257             scripted = self.checkScript(func, (a,))
259             scripted = self.checkScript(func_neg, (a,))
280             scripted = self.checkScript(func, (a,))
297             scripted = self.checkScript(foo, inputs)
298             self.assertAllFused(scripted.graph_for(*inputs))
401                 scripted = self.checkScript(f, (x, y))
402                 self.assertEqual(scripted(x, y).shape, (3, 4))
403                 self.assertAllFused(scripted.graph_for(x, y))
892             scripted = self.checkScript(f, (x,))
893             self.assertAllFused(scripted.graph_for(x))
903             scripted = self.checkScript(fn_test_scalar_arg, (x, p))
904             self.assertAllFused(scripted.graph_for(x, p))
915             scripted = torch.jit.script(fn_test_scalar_arg_requires_grad)
916             out = scripted(x, p)
917             out = scripted(x, p)
918             out = scripted(x, p)
920                 scripted.graph_for(x, p),
2102             scripted = self.checkScript(eager, (t, t, t, t, 0.1))
2597         scripted = self.checkScript(success, (torch.rand([4]),))