Lines Matching defs:ctors
164 def test_vjp_err_check(self, ctors): argument
203 def test_vjp_err_check_strict(self, ctors): argument
246 def test_vjp_no_grad(self, ctors): argument
267 def test_vjp_output(self, ctors): argument
302 def test_vjp_scalar(self, ctors): argument
326 def test_vjp_create_graph(self, ctors): argument
394 def test_jvp_err_check(self, ctors): argument
433 def test_jvp_err_check_strict(self, ctors): argument
476 def test_jvp_no_grad(self, ctors): argument
497 def test_jvp_output(self, ctors): argument
532 def test_jvp_scalar(self, ctors): argument
556 def test_jvp_create_graph(self, ctors): argument
636 def test_construct_standard_basis_for(self, ctors): argument
653 def test_construct_standard_basis_for_cuda(self, ctors): argument
662 def _test_vectorize_raises_no_warnings(self, api, ctors): argument
677 def test_jacobian_vectorize_raises_no_warnings(self, ctors): argument
681 def test_hessian_vectorize_raises_no_warnings(self, ctors): argument
686 def test_jacobian_err_check(self, vectorize, ctors): argument
717 def test_jacobian_err_check_strict(self, ctors): argument
759 def test_jacobian_err_check_strict_vectorize(self, ctors): argument
768 def test_jacobian_no_grad(self, ctors): argument
784 def test_jacobian_output(self, vectorize, ctors): argument
812 def test_jacobian_scalar(self, vectorize, ctors): argument
829 def test_jacobian_create_graph(self, vectorize, ctors): argument
905 def test_jacobian_vectorize_correctness_simple(self, ctors): argument
913 def test_jacobian_vectorize_correctness_multi_input(self, ctors): argument
922 def test_jacobian_vectorize_correctness_multi_input_multi_output(self, ctors): argument
931 def test_jacobian_vectorize_correctness_unrelated_outputs(self, ctors): argument
940 def test_jacobian_vectorize_correctness_zero_dim(self, ctors): argument
966 def test_jacobian_vectorize_correctness_different_devices(self, ctors): argument
975 def test_jacobian_vectorize_correctness_different_dtype(self, ctors): argument
996 def test_hessian_vectorize_correctness_simple(self, ctors): argument
1004 def test_hessian_vectorize_correctness_multi_input(self, ctors): argument
1014 def test_hessian_vectorize_correctness_unrelated_outputs(self, ctors): argument
1033 def test_hessian_err_check(self, vectorize, ctors): argument
1078 def test_hessian_err_check_strict(self, ctors): argument
1119 def test_hessian_err_check_strict_vectorize(self, ctors): argument
1128 def test_hessian_no_grad(self, ctors): argument
1150 def test_hessian_output(self, vectorize, ctors): argument
1172 def test_hessian_scalar(self, vectorize, ctors): argument
1193 def test_hessian_create_graph(self, vectorize, ctors): argument
1267 def test_vhp_err_check(self, ctors): argument
1315 def test_vhp_err_check_strict(self, ctors): argument
1357 def test_vhp_no_grad(self, ctors): argument
1376 def test_vhp_output(self, ctors): argument
1399 def test_vhp_scalar(self, ctors): argument
1425 def test_vhp_create_graph(self, ctors): argument
1488 def test_hvp_err_check(self, ctors): argument
1537 def test_hvp_err_check_strict(self, ctors): argument
1579 def test_hvp_no_grad(self, ctors): argument
1598 def test_hvp_output(self, ctors): argument
1621 def test_hvp_scalar(self, ctors): argument
1647 def test_hvp_create_graph(self, ctors): argument
1710 def test_jacobian_match_vjp_jvp(self, ctors): argument
1725 def test_hessian_match_vhp_hvp(self, ctors): argument