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)
|