Searched defs:x_tangent (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/testing/_internal/ |
| H A D | autograd_function_db.py | 136 def jvp(ctx, x_tangent, y_tangent): argument
173 def jvp(ctx, x_tangent, y_tangent): argument
203 def jvp(ctx, x_tangent): argument
246 def jvp(ctx, x_tangent, _): argument
276 def jvp(ctx, x_tangent, _): argument
335 def jvp(ctx, x_tangent, ind_tangent, ind_inv_tangent, _): argument
362 def jvp(ctx, x_tangent, ind_tangent, ind_inv_tangent, _): argument
390 def jvp(ctx, x_tangent, _): argument
413 def jvp(ctx, x_tangent, _): argument
438 def jvp(ctx, x_tangent): argument
[all …]
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_autograd.py | 8846 def jvp(ctx, x_tangent, y_tangent): argument
8867 def jvp(ctx, x_tangent, y_tangent): argument
8884 def jvp(ctx, x_tangent, y_tangent): argument
8901 def jvp(ctx, x_tangent): argument
|