Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/_subclasses/
H A Dfunctional_tensor.py638 def wrap_tensors(self, args: Tuple[Any]) -> Tuple[Any]: member in BaseFunctionalizeAPI
680 def wrap_tensors(self, args: Tuple[Any]) -> Tuple[Any]: member in PythonFunctionalizeAPI
723 def wrap_tensors(self, args: Tuple[Any]) -> Tuple[Any]: member in CppFunctionalizeAPI
762 def wrap_tensors(self, args: Tuple[Any]) -> Tuple[Any]: member in FunctorchFunctionalizeAPI
/aosp_15_r20/external/pytorch/torch/_numpy/
H A D_normalizations.py181 def wrap_tensors(result): function
/aosp_15_r20/external/pytorch/torch/_inductor/
H A Dlowering.py1701 def wrap_tensors(x): function