Lines Matching refs:scripted
61 scripted = self.checkScript(func, (a,))
62 self.assertAllFused(scripted.graph_for(a))
186 scripted = self.checkScript(f, (x, y))
187 self.assertAllFused(scripted.graph_for(x, y))
569 scripted = self.checkScript(f, (x,))
570 self.assertAllFused(scripted.graph_for(x))
579 scripted = self.checkScript(fn_test_scalar_arg, (x, p))
580 self.assertAllFused(scripted.graph_for(x, p))
589 scripted = torch.jit.script(fn_test_scalar_arg_requires_grad)
590 out = scripted(x, p)
591 … self.assertAllFused(scripted.graph_for(x, p), except_for=("aten::size", "prim::BroadcastSizes",