Home
last modified time | relevance | path

Searched refs:allow_mutation_on_saved_tensors (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/torch/autograd/
H A Dgraph.py669 def allow_mutation_on_saved_tensors(): function
/aosp_15_r20/external/pytorch/test/
H A Dtest_autograd.py12133 with torch.autograd.graph.allow_mutation_on_saved_tensors() as ctx:
12156 with torch.autograd.graph.allow_mutation_on_saved_tensors() as ctx:
12163 with torch.autograd.graph.allow_mutation_on_saved_tensors() as ctx:
12178 with torch.autograd.graph.allow_mutation_on_saved_tensors() as ctx:
12188 with torch.autograd.graph.allow_mutation_on_saved_tensors() as ctx:
12203 with torch.autograd.graph.allow_mutation_on_saved_tensors() as ctx:
12219 with torch.autograd.graph.allow_mutation_on_saved_tensors() as ctx:
12229 with torch.autograd.graph.allow_mutation_on_saved_tensors() as ctx:
12242 with torch.autograd.graph.allow_mutation_on_saved_tensors() as ctx:
12257 with torch.autograd.graph.allow_mutation_on_saved_tensors() as ctx:
[all …]
/aosp_15_r20/external/pytorch/docs/source/
H A Dautograd.rst361 .. autoclass:: torch.autograd.graph.allow_mutation_on_saved_tensors