Searched refs:_test_reference_numerics (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_unary_ufuncs.py | 182 def _test_reference_numerics(self, dtype, op, tensors, equal_nan=True): member in TestUnaryUfuncs
277 self._test_reference_numerics(dtype, op, tensors)
288 self._test_reference_numerics(dtype, op, tensors)
299 self._test_reference_numerics(dtype, op, tensors)
310 self._test_reference_numerics(dtype, op, tensors)
|
| H A D | test_binary_ufuncs.py | 120 def _test_reference_numerics(self, dtype, op, gen, equal_nan=True): member in TestBinaryUfuncs
196 self._test_reference_numerics(dtype, op, gen, equal_nan=True)
206 self._test_reference_numerics(dtype, op, gen, equal_nan=True)
226 self._test_reference_numerics(dtype, op, gen, equal_nan=True)
243 self._test_reference_numerics(dtype, op, gen, equal_nan=True)
257 self._test_reference_numerics(dtype, op, gen, equal_nan=True)
267 self._test_reference_numerics(dtype, op, gen, equal_nan=True)
271 self._test_reference_numerics(dtype, op, gen, equal_nan=True)
|