Home
last modified time | relevance | path

Searched refs:python_op (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/torch/testing/_internal/distributed/_shard/sharded_tensor/
H A D_test_ops_common.py124 def gen_binary_op_func(python_op, inplace=False): argument
126 if "torch" in python_op:
/aosp_15_r20/external/pytorch/test/jit/
H A Dtest_exception.py153 def python_op(x): function
158 return python_op(x)
/aosp_15_r20/external/pytorch/test/
H A Dtest_binary_ufuncs.py1857 def _scalar_helper(python_op, torch_op): argument
1867 expected = python_op(a, b)
2080 for python_op, torch_op in (
2114 expected = python_op(first_scalar, second_scalar)
2116 self.assertEqual(expected, python_op(first, second))