| /external/pytorch/torch/testing/_internal/ |
| D | autograd_function_db.py | 52 def jvp(ctx, input_tangent): member in NumpyCube
76 def jvp(ctx, input_tangent): member in CubeGenVmap
136 def jvp(ctx, x_tangent, y_tangent): member in NumpyMul
173 def jvp(ctx, x_tangent, y_tangent): member in MulGenVmap
203 def jvp(ctx, x_tangent): member in NumpyExp_
246 def jvp(ctx, x_tangent, _): member in NumpySort
276 def jvp(ctx, x_tangent, _): member in SortGenVmap
335 def jvp(ctx, x_tangent, ind_tangent, ind_inv_tangent, _): member in NumpyTake
362 def jvp(ctx, x_tangent, ind_tangent, ind_inv_tangent, _): member in TakeGenVmap
390 def jvp(ctx, x_tangent, _): member in Select
[all …]
|
| /external/sg3_utils/lib/ |
| D | sg_pr2serr.c | 318 json_value * jvp = json_object_new(0); in sgj_start_r() local
396 json_value * jvp = (json_value *)(jop ? jop : jsp->basep); in sgj_js2file() local
945 sgj_jtype_to_s(char * b, int blen_max, json_value * jvp) in sgj_jtype_to_s()
968 json_value * jvp, int64_t val_instead) in sgj_haj_helper()
1014 const char * name, enum sgj_separator_t sep, json_value * jvp, in sgj_haj_xx()
1116 json_value * jvp; in sgj_haj_vs() local
1128 json_value * jvp; in sgj_haj_vi() local
1139 json_value * jvp; in sgj_haj_vistr() local
1151 json_value * jvp; in sgj_haj_vi_nex() local
1163 json_value * jvp; in sgj_haj_vistr_nex() local
[all …]
|
| /external/sg3_utils/testing/ |
| D | sg_tst_json_builder.c | 42 json_value * jvp = NULL; in main() local
|
| /external/pytorch/torch/autograd/ |
| D | functional.py | 359 def jvp(func, inputs, v=None, create_graph=False, strict=False): function
526 def jvp(tangents): function
|
| D | function.py | 441 def jvp(ctx: Any, *grad_inputs: Any) -> Any: member in _SingleLevelFunction
|
| D | gradcheck.py | 1066 def jvp(tangent: torch.Tensor): function
|
| /external/pytorch/functorch/ |
| D | README.md | 208 ### jvp subsection
|
| /external/pytorch/torch/_functorch/ |
| D | autograd_function.py | 129 def jvp(ctx, *tangents): function
434 def jvp(ctx, *tangents): function
|
| D | deprecated.py | 87 def jvp( function
|
| D | eager_transforms.py | 1025 def jvp( function
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | forwardprop.py | 413 def jvp(self, primals, unconnected_gradients=UnconnectedGradients.NONE): member in ForwardAccumulator
|
| D | pywrap_tfe_src.cc | 2959 PyObject* jvp = reinterpret_cast<TFE_Py_ForwardAccumulator*>(accumulator) local
3027 PyObject* jvp = (*it)->accumulator->FetchJVP(existing_input); local
3055 PyObject* jvp = new_tensors[i]; local
|
| /external/pytorch/benchmarks/functional_autograd_benchmark/ |
| D | functional_autograd_benchmark.py | 66 def jvp(model, inp, v=None, strict=None): function
|
| /external/pytorch/test/ |
| D | test_autograd.py | 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
[all …]
|
| /external/pytorch/test/dynamo/ |
| D | test_autograd_function.py | 795 def jvp(ctx, x_t): member in AutogradFunctionTests.test_smoke_from_test_autograd.InplaceMul
811 def jvp(ctx, x_t, y_t): member in AutogradFunctionTests.test_smoke_from_test_autograd.MyFn2
|
| /external/pytorch/test/functorch/ |
| D | test_eager_transforms.py | 1192 def jvp(ctx, x_t): member in TestAutogradFunction.test_function_returns_input.A
|