Searched refs:_test_binary_op_shape (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_jit.py | 8573 def _test_binary_op_shape(self, ops, input_dims=1): member in TestScript
8637 self._test_binary_op_shape(['mul', 'div', 'add', 'sub'], 0)
8638 self._test_binary_op_shape(['mul', 'div', 'add', 'sub'], 3)
|