Searched refs:CachedMethod (Results 1 – 2 of 2) sorted by relevance
| /external/pytorch/torch/_inductor/codegen/ |
| D | memory_planning.py | 15 from ..utils import _align, align, cache_on_self, CachedMethod, IndentedBuffer
207 get_live_ranges: CachedMethod[[], LiveRanges]
208 get_size_hint: CachedMethod[[], int]
209 get_symbolic_size: CachedMethod[[], sympy.Expr]
|
| /external/pytorch/torch/_inductor/ |
| D | utils.py | 456 class CachedMethod(Protocol, Generic[P, RV]): class
466 def cache_on_self(fn: Callable[Concatenate[Any, P], RV]) -> CachedMethod[P, RV]:
|