• Home
  • Raw
  • Download

Lines Matching refs:vectorize

673             result = api(foo, x, vectorize=True)
686 def test_jacobian_err_check(self, vectorize, ctors): argument
697 res = autogradF.jacobian(foo, (inp, 2), vectorize=vectorize)
703 res = autogradF.jacobian(bar, inp, vectorize=vectorize)
705 res = autogradF.jacobian(foo, inp, vectorize=vectorize)
713 res = autogradF.jacobian(foo, inp, vectorize=vectorize)
765 res = autogradF.jacobian(foo, inp, strict=True, vectorize=True)
784 def test_jacobian_output(self, vectorize, ctors): argument
789 res = autogradF.jacobian(exp_reducer, inputs, vectorize=vectorize)
797 res = autogradF.jacobian(identity, inputs, vectorize=vectorize)
806 res = autogradF.jacobian(add_exp_reducer, inputs, vectorize=vectorize)
812 def test_jacobian_scalar(self, vectorize, ctors): argument
817 res = autogradF.jacobian(reducer, inputs, vectorize=vectorize)
824 res = autogradF.jacobian(expander, inputs, vectorize=vectorize)
829 def test_jacobian_create_graph(self, vectorize, ctors): argument
835 exp_reducer, inputs, create_graph=True, vectorize=vectorize
842 exp_reducer, inp, create_graph=True, vectorize=vectorize
848 exp_reducer, inp, create_graph=True, vectorize=vectorize
861 add_exp_reducer, inputs, create_graph=True, vectorize=vectorize
869 add_exp_reducer, inp, create_graph=True, vectorize=vectorize
875 add_exp_reducer, inp, create_graph=True, vectorize=vectorize
883 add_exp_reducer, (x, y), create_graph=True, vectorize=vectorize
894 expected = autogradF.jacobian(f, inputs, vectorize=False)
895 result_backward_mode = autogradF.jacobian(f, inputs, vectorize=True)
900 f, inputs, strategy="forward-mode", vectorize=True
986 expected = autogradF.hessian(f, inputs, vectorize=False)
987 result = autogradF.hessian(f, inputs, vectorize=True)
991 f, inputs, outer_jacobian_strategy="forward-mode", vectorize=True
1033 def test_hessian_err_check(self, vectorize, ctors): argument
1050 res = autogradF.hessian(foo, (inp, 2), vectorize=vectorize)
1055 res = autogradF.hessian(bar, inp, vectorize=vectorize)
1059 res = autogradF.hessian(bar2, inp, vectorize=vectorize)
1064 res = autogradF.hessian(bar3, inp, vectorize=vectorize)
1066 res = autogradF.hessian(foo, inp, vectorize=vectorize)
1074 res = autogradF.hessian(foo, inp, vectorize=vectorize)
1125 res = autogradF.hessian(foo, inp, strict=True, vectorize=True)
1150 def test_hessian_output(self, vectorize, ctors): argument
1155 res = autogradF.hessian(pow_reducer, inputs, vectorize=vectorize)
1163 res = autogradF.hessian(add_pow_reducer, inputs, vectorize=vectorize)
1172 def test_hessian_scalar(self, vectorize, ctors): argument
1177 res = autogradF.hessian(reducer, inputs, vectorize=vectorize)
1181 res = autogradF.hessian(reducer, inputs, vectorize=vectorize)
1188 res = autogradF.hessian(bad_reducer, inputs, vectorize=vectorize)
1193 def test_hessian_create_graph(self, vectorize, ctors): argument
1199 pow_reducer, inputs, create_graph=True, vectorize=vectorize
1206 pow_reducer, inp, create_graph=True, vectorize=vectorize
1212 pow_reducer, inp, create_graph=True, vectorize=vectorize
1225 add_pow_reducer, inputs, create_graph=True, vectorize=vectorize
1239 add_pow_reducer, inp, create_graph=True, vectorize=vectorize
1247 add_pow_reducer, inp, create_graph=True, vectorize=vectorize
1256 add_pow_reducer, (x, y), create_graph=True, vectorize=vectorize