Searched refs:PythonFunctionalizeAPI (Results 1 – 6 of 6) sorted by relevance
333 from torch._subclasses.functional_tensor import PythonFunctionalizeAPI335 ctx = PythonFunctionalizeAPI()415 from torch._subclasses.functional_tensor import PythonFunctionalizeAPI417 ctx = PythonFunctionalizeAPI()
261 from torch._subclasses.functional_tensor import PythonFunctionalizeAPI263 ctx = PythonFunctionalizeAPI()
672 class PythonFunctionalizeAPI(BaseFunctionalizeAPI): class
148 PythonFunctionalizeAPI as _PythonFunctionalizeAPI,
123 PythonFunctionalizeAPI,146 gm = make_fx(PythonFunctionalizeAPI().functionalize(f))(t1, t2)
16 PythonFunctionalizeAPI,1676 fn = PythonFunctionalizeAPI().functionalize(fn)