Home
last modified time | relevance | path

Searched refs:python_fn (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/pytorch/test/jit/
H A Dtest_autodiff.py100 def python_fn(x): function
105 z = python_fn(r)
124 def python_fn(x): function
129 y = python_fn(b)
/aosp_15_r20/external/pytorch/torch/testing/_internal/
H A Djit_utils.py414 python_fn = frame[name]
416 python_fn = script
418 python_fn(*inputs)
435 ge = torch.jit.script(python_fn)
478 python_fn = frame[name]
496 python_fn = script
509 python_outputs = python_fn(*inputs)
522 python_outputs = python_fn(*inputs)
/aosp_15_r20/external/pytorch/test/expect/
H A DTestJit.test_pretty_printer-python_op_name_test.expect2 return ^python_fn()(y)
/aosp_15_r20/external/pytorch/test/
H A Dtest_jit.py2613 def python_fn(x): function
2618 return python_fn(y)
12184 def python_fn(x): function
12189 return python_fn(x) + 1
12279 def python_fn(x): function
12288 return torch.mm(python_fn(x), self.param)
12376 def python_fn(x): function
12381 return python_fn(x) + 1
12435 def python_fn(x): function
12445 return python_fn(torch.mm(x, self.param))