Searched defs:_compiled_autograd_key (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/autograd/ | ||
| H A D | function.py | 316 def _compiled_autograd_key(self): member in BackwardCFunction |
| /aosp_15_r20/external/pytorch/torch/_functorch/_aot_autograd/ | ||
| H A D | runtime_wrappers.py | 1507 def _compiled_autograd_key(ctx): member in AOTDispatchAutograd.post_compile.CompiledFunction |