Lines Matching refs:checkScript

219         scripted = self.checkScript(fused_kernel, (a, a))
241             scripted = self.checkScript(func, (a,))
257             scripted = self.checkScript(func, (a,))
259             scripted = self.checkScript(func_neg, (a,))
270             self.checkScript(func, (a,))
280             scripted = self.checkScript(func, (a,))
297             scripted = self.checkScript(foo, inputs)
310             ge = self.checkScript(decode, inputs)
342             self.checkScript(scaleshift, inputs)
401                 scripted = self.checkScript(f, (x, y))
417             self.checkScript(fn, inputs)
452                     self.checkScript(fn, [tensor])
498                 self.checkScript(func, inputs)
521             ge = self.checkScript(fn, inputs)
541                 s = self.checkScript(f, inputs)
569                 s = self.checkScript(f, (inp1, inp2), profiling=ProfilingMode.PROFILING)
593             s = self.checkScript(
610             s = self.checkScript(clamp_int, (x, eta), profiling=ProfilingMode.PROFILING)
892             scripted = self.checkScript(f, (x,))
903             scripted = self.checkScript(fn_test_scalar_arg, (x, p))
942         ge = self.checkScript(fn, inputs_cpu)
971         ge = self.checkScript(fn, inputs)
993             module = self.checkScript(LSTMCellS, inputs)
1064             module = self.checkScript(MiLSTMCell, inputs)
1124             ge = self.checkScript(fn_test_erf, (x,), profiling=ProfilingMode.PROFILING)
1127             ge = self.checkScript(fn_test_erf, (x,), profiling=ProfilingMode.PROFILING)
1197         ge = self.checkScript(fn, (x, y))
1256             ge = self.checkScript(should_fuse, inputs)
1266             ge = self.checkScript(should_fuse_scalar, inputs)
1290             script_f = self.checkScript(f, (x, y))
1303         s = self.checkScript(fn, (x,))
2102             scripted = self.checkScript(eager, (t, t, t, t, 0.1))
2117             script = self.checkScript(eager, (x,))
2130             script = self.checkScript(eager, (a, b))
2145         script = self.checkScript(eager_tt, (a, b))
2148         script = self.checkScript(eager_ts, (a, s))
2150         script = self.checkScript(eager_st, (s, b))
2165         script = self.checkScript(eager, (input, weight, bias))
2179         script = self.checkScript(eager, (input, weight, bias))
2209         script = self.checkScript(eager, (x,))
2220         script = self.checkScript(eager, (x, y))
2274         script = self.checkScript(eager, (x, y, z))
2317         self.checkScript(
2349         script = self.checkScript(eager, (1.0,))
2597         scripted = self.checkScript(success, (torch.rand([4]),))