Home
last modified time | relevance | path

Searched defs:functionalize (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/torch/_subclasses/
H A Dfunctional_tensor.py648 def functionalize(self, inner_f: Callable) -> Callable: member in BaseFunctionalizeAPI
693 def functionalize(self, inner_f: Callable) -> Callable: member in PythonFunctionalizeAPI
737 def functionalize(self, inner_f: Callable) -> Callable: member in CppFunctionalizeAPI
778 def functionalize(self, inner_f: Callable) -> Callable: member in FunctorchFunctionalizeAPI
/aosp_15_r20/external/pytorch/torch/_functorch/
H A Ddeprecated.py133 def functionalize(func: Callable, *, remove: str = "mutations") -> Callable: function
H A Deager_transforms.py1508 def functionalize(func: Callable, *, remove: str = "mutations") -> Callable: function