Searched refs:grad_saved (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/testing/_internal/ |
| H A D | autograd_function_db.py | 42 def backward(ctx, grad_output, grad_saved): argument
44 return NumpyMul.apply(grad_output, dinput) + 6 * NumpyMul.apply(grad_saved, input)
70 def backward(ctx, grad_output, grad_saved): argument
100 def backward(ctx, grad_output, grad_saved): argument
|
| /aosp_15_r20/external/pytorch/test/functorch/ |
| H A D | test_eager_transforms.py | 1278 def backward(ctx, grad_output, grad_saved): argument
1359 def backward(ctx, grad_output, grad_saved): argument
1381 def backward(ctx, grad_output, grad_saved): argument
1405 def backward(ctx, grad_output, grad_saved): argument
1430 def backward(ctx, grad_output, grad_saved): argument
1454 def backward(ctx, grad_output, grad_saved): argument
1478 def backward(ctx, grad_output, grad_saved): argument
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_autograd.py | 831 grad_saved = params.grad
834 id(grad_saved) == id(params.grad), should_preserve_reference
|