Searched defs:cotangents (Results 1 – 3 of 3) sorted by relevance
| /external/pytorch/test/functorch/ |
| D | test_ops.py | 204 def wrapped(cotangents): argument
232 def get_vjp_fn_and_args_with_cotangents(f, sample, cotangents): argument
1742 def get_vjp(cotangents, *primals): argument
1867 def push_vjp(primals, cotangents): argument
1883 def reference(primals, cotangents, primals_tangents, cotangents_tangents): argument
2114 def push_vjp(primals, cotangents): argument
2471 def compute_grad(cotangents): argument
2669 def inner(primals, cotangents): argument
|
| /external/pytorch/torch/_functorch/ |
| D | eager_transforms.py | 65 def _vjp_treespec_compare(primals_out, cotangents): argument
425 def wrapper(cotangents, retain_graph=True, create_graph=None): argument
|
| /external/pytorch/test/ |
| D | test_decomp.py | 148 def wrapped(cotangents): argument
|