Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/test/
H A Dtest_jit.py8573 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)