Home
last modified time | relevance | path

Searched refs:disable_saved_tensors_hooks (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/pytorch/test/dynamo/
H A Dtest_ctx_manager.py1162 @torch.autograd.graph.disable_saved_tensors_hooks("This is not supported")
1199 @torch.autograd.graph.disable_saved_tensors_hooks("This is not supported")
1211 with torch.autograd.graph.disable_saved_tensors_hooks(
1239 @torch.autograd.graph.disable_saved_tensors_hooks("This is not supported")
1241 @torch.autograd.graph.disable_saved_tensors_hooks(
1257 with torch.autograd.graph.disable_saved_tensors_hooks(
1291 with torch.autograd.graph.disable_saved_tensors_hooks(
H A Dtest_repros.py1121 with torch.autograd.graph.disable_saved_tensors_hooks("hooks are disabled"):
1128 with torch.autograd.graph.disable_saved_tensors_hooks("hooks are disabled"):
/aosp_15_r20/external/pytorch/torch/_dynamo/variables/
H A Dtorch.py89 torch.autograd.graph.disable_saved_tensors_hooks,
314 elif self.value is torch.autograd.graph.disable_saved_tensors_hooks:
/aosp_15_r20/external/pytorch/torch/_functorch/
H A Dvmap.py47 with torch.autograd.graph.disable_saved_tensors_hooks(message):
/aosp_15_r20/external/pytorch/torch/autograd/
H A Dgraph.py352 def disable_saved_tensors_hooks(error_message): function
/aosp_15_r20/external/pytorch/docs/source/
H A Dautograd.rst357 .. autoclass:: torch.autograd.graph.disable_saved_tensors_hooks
/aosp_15_r20/external/pytorch/test/
H A Dtest_autograd.py9483 with torch.autograd.graph.disable_saved_tensors_hooks("error message"):
9492 with torch.autograd.graph.disable_saved_tensors_hooks("error message"):
9498 with torch.autograd.graph.disable_saved_tensors_hooks("outer"):
9499 with torch.autograd.graph.disable_saved_tensors_hooks("inner"):