| /external/pytorch/torch/autograd/ |
| D | functional.py | 264 def vjp(func, inputs, v=None, create_graph=False, strict=False): function
737 def vjp(grad_output): function
|
| D | __init__.py | 481 def vjp(gO): function
|
| D | function.py | 438 vjp = backward variable in _SingleLevelFunction
|
| D | gradcheck.py | 1132 def vjp(v): function
|
| /external/pytorch/functorch/ |
| D | README.md | 199 ### vjp subsection
|
| /external/pytorch/torch/_functorch/ |
| D | deprecated.py | 82 def vjp(func: Callable, *primals, has_aux: bool = False): function
|
| D | eager_transforms.py | 242 def vjp(func: Callable, *primals, has_aux: bool = False): function
|
| /external/pytorch/benchmarks/functional_autograd_benchmark/ |
| D | functional_autograd_benchmark.py | 60 def vjp(model, inp, v=None, strict=None): function
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | backprop.py | 583 def vjp(dy=None): function
|
| /external/pytorch/test/ |
| D | test_legacy_vmap.py | 2811 def vjp(v): function
2825 def vjp(v): function
|
| D | test_autograd.py | 6449 def vjp(ctx, gO): member in TestAutograd.test_gradcheck_check_forward_or_backward_only.UserFn
6538 def vjp(ctx, gO): member in TestAutograd.test_gradcheck_forward_ad_batched_grad.Fn
8551 def vjp(ctx, foo): member in TestAutograd.test_custom_function_error.BadBw2
8592 def vjp(ctx, gO): member in TestAutograd.test_custom_function_forward_mode_view_checks.ViewFn
8633 def vjp(ctx, gO): member in TestAutograd.test_custom_function_forward_mode_inplace_checks.InplaceFn
8667 def vjp(ctx, gO): member in TestAutograd.test_custom_function_forward_mode_wrong_formula.UserFn
8738 def vjp(ctx, gO1, gO2): member in TestAutograd.test_custom_function_forward_mode_forward_is_no_op.MyFn
8793 def vjp(ctx, grad_out): member in TestAutograd.test_custom_function_save_for_forward.Func
8823 def vjp(ctx, grad_out): member in TestAutograd.test_custom_function_save_for_forward.Func
|
| /external/pytorch/test/dynamo/ |
| D | test_autograd_function.py | 807 def vjp(ctx, gO1, gO2): member in AutogradFunctionTests.test_smoke_from_test_autograd.MyFn2
|
| /external/pytorch/test/functorch/ |
| D | test_vmap.py | 4009 def vjp(v): function
4023 def vjp(v): function
|