Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/jit/
H A D_decompositions.py70 def decomposition_decorator(f: Callable[_P, _T]) -> Callable[_P, _T]: function
93 return decomposition_decorator
/aosp_15_r20/external/pytorch/torch/_decomp/
H A D__init__.py178 def decomposition_decorator(fn: Callable[_P, _T]) -> Callable[_P, _T]: function
194 return decomposition_decorator