Searched defs:save_for_backward (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/autograd/ | ||
| H A D | function.py | 35 def save_for_backward(self, *tensors: torch.Tensor): member in FunctionCtx |
| /aosp_15_r20/external/pytorch/torch/csrc/autograd/ | ||
| H A D | custom_function.cpp | 506 void AutogradContext::save_for_backward(variable_list to_save) { in save_for_backward() function in torch::autograd::AutogradContext |
| /aosp_15_r20/external/pytorch/torch/_dynamo/ | ||
| H A D | side_effects.py | 85 save_for_backward=None, argument |
| /aosp_15_r20/external/pytorch/torch/_functorch/ | ||
| H A D | autograd_function.py | 608 def save_for_backward(self, *tensors): member in CtxCustomSave |