Home
last modified time | relevance | path

Searched defs:mark_mutation_hidden_from_autograd (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/pytorch/torch/_subclasses/
H A Dfunctional_tensor.py269 def mark_mutation_hidden_from_autograd(self) -> None: member in FunctionalTensor
668 def mark_mutation_hidden_from_autograd(self, tensor) -> None: member in BaseFunctionalizeAPI
717 def mark_mutation_hidden_from_autograd(self, tensor) -> None: member in PythonFunctionalizeAPI
754 def mark_mutation_hidden_from_autograd(self, tensor) -> None: member in CppFunctionalizeAPI
800 def mark_mutation_hidden_from_autograd(self, tensor) -> None: member in FunctorchFunctionalizeAPI
/aosp_15_r20/external/pytorch/aten/src/ATen/
H A DFunctionalTensorWrapper.h83 void mark_mutation_hidden_from_autograd() { in mark_mutation_hidden_from_autograd() function
H A DFunctionalStorageImpl.h131 void mark_mutation_hidden_from_autograd() { in mark_mutation_hidden_from_autograd() function
H A DFunctionalTensorWrapper.cpp694 void mark_mutation_hidden_from_autograd(const Tensor& functional_tensor) { in mark_mutation_hidden_from_autograd() function