Home
last modified time | relevance | path

Searched refs:BaseFunctionalizeAPI (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/torch/_subclasses/
H A Dfunctional_tensor.py636 class BaseFunctionalizeAPI(ABC): class
672 class PythonFunctionalizeAPI(BaseFunctionalizeAPI):
722 class CppFunctionalizeAPI(BaseFunctionalizeAPI):
758 class FunctorchFunctionalizeAPI(BaseFunctionalizeAPI):
/aosp_15_r20/external/pytorch/torch/_higher_order_ops/
H A Dflex_attention.py388 ctx: torch._subclasses.functional_tensor.BaseFunctionalizeAPI, argument
954 ctx: torch._subclasses.functional_tensor.BaseFunctionalizeAPI, argument