Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/test/autograd/
H A Dtest_functional.py893 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)