Lines Matching defs:jvp
6311 def jvp(ctx, x_t, y_t): member in TestAutograd.test_gradcheck_forward_ad_runs_with_no_requires_grad.UserFn
6370 def jvp(ctx, x_t, y_t): member in TestAutograd.test_gradcheck_forward_ad_respects_requires_grad.UserFn
6456 def jvp(ctx, gI, _1, _2): member in TestAutograd.test_gradcheck_check_forward_or_backward_only.UserFn
6542 def jvp(ctx, gI): member in TestAutograd.test_gradcheck_forward_ad_batched_grad.Fn
8390 def jvp(ctx, x_t): member in TestAutograd.test_custom_function_mark_dirty_not_differentiable.get_custom_fn.InplaceMul
8598 def jvp(ctx, gI, _): member in TestAutograd.test_custom_function_forward_mode_view_checks.ViewFn
8637 def jvp(ctx, gI, _): member in TestAutograd.test_custom_function_forward_mode_inplace_checks.InplaceFn
8671 def jvp(ctx, gI, _): member in TestAutograd.test_custom_function_forward_mode_wrong_formula.UserFn
8693 def jvp(ctx, nt, x_t, nt2, y_t): member in TestAutograd.test_custom_function_forward_mode_non_tensor_before_tensor_args.MyFn
8742 def jvp(ctx, x_t, y_t): member in TestAutograd.test_custom_function_forward_mode_forward_is_no_op.MyFn
8787 def jvp(ctx, x_t, y_t, _): member in TestAutograd.test_custom_function_save_for_forward.Func
8818 def jvp(ctx, x_t): member in TestAutograd.test_custom_function_save_for_forward.Func
8846 def jvp(ctx, x_tangent, y_tangent): member in TestAutograd.test_custom_function_forward_mode_non_differentiable.Func
8867 def jvp(ctx, x_tangent, y_tangent): member in TestAutograd.test_custom_function_forward_mode_non_differentiable.Func
8884 def jvp(ctx, x_tangent, y_tangent): member in TestAutograd.test_custom_function_forward_mode_non_differentiable.FuncWrong
8901 def jvp(ctx, x_tangent): member in TestAutograd.test_custom_function_forward_mode_non_differentiable.Func
10332 def jvp(tangent): function
10347 def jvp(tangent): function
10372 def jvp(tangent): function
10396 def jvp(tangent): function