Searched defs:grad_saved (Results 1 – 2 of 2) 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 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
|