Searched refs:_check_jacobian_vectorize_correctness (Results 1 – 1 of 1) sorted by relevance
/aosp_15_r20/external/pytorch/test/autograd/ |
H A D | test_functional.py | 893 def _check_jacobian_vectorize_correctness(self, f, inputs, test_forward_ad=True): member in TestAutogradFunctional 910 self._check_jacobian_vectorize_correctness(f, x) 919 self._check_jacobian_vectorize_correctness(f, (x, y)) 928 self._check_jacobian_vectorize_correctness(f, (x, y)) 937 self._check_jacobian_vectorize_correctness(f, (x, y)) 947 self._check_jacobian_vectorize_correctness(f, (x, y)) 954 self._check_jacobian_vectorize_correctness(g, x) 962 self._check_jacobian_vectorize_correctness(h, (x, y)) 972 self._check_jacobian_vectorize_correctness(f, (x, y)) 983 self._check_jacobian_vectorize_correctness(f, (x, y), test_forward_ad=False)
|