Home
last modified time | relevance | path

Searched defs:ChoiceCaller (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/pytorch/torch/_inductor/autoheuristic/
H A Dautoheuristic.py225 fallback: Callable[[], Optional[ChoiceCaller]],
226 choices: List[ChoiceCaller],
269 self, input_nodes: List[Any], choices: List[ChoiceCaller]
284 timings: Dict[ChoiceCaller, float],
287 choices: List[ChoiceCaller],
/aosp_15_r20/external/pytorch/torch/_inductor/
H A Dselect_algorithm.py1120 def __init__(self, msg, choice: ChoiceCaller, inputs_str) -> None:
1142 name: str, inputs_key: str, choices: List[ChoiceCaller]
1173 choices: List[ChoiceCaller],
1552 timings: Dict[ChoiceCaller, float],
1705 [Dict[ChoiceCaller, float], str, List[Any], List[ChoiceCaller]], None
1728 fn: Callable[[Dict[ChoiceCaller, float], str, List[Any], List[ChoiceCaller]], None]
H A Ddebug.py531 timings: Dict["ChoiceCaller", float], # type: ignore[name-defined] # noqa: F821
H A Dcodecache.py245 choices: List[ChoiceCaller],
248 benchmark: Optional[Callable[[Any], Dict[ChoiceCaller, float]]],
H A Dir.py3608 class ChoiceCaller: class
3662 choice_timings: Callable[[], Dict[ChoiceCaller, float]],
/aosp_15_r20/external/pytorch/torch/_inductor/codegen/cuda/
H A Dgemm_template.py411 choices: List[ChoiceCaller],
483 choices: List[ChoiceCaller],
1010 choices: List[ChoiceCaller],
1317 choices: List[ChoiceCaller],
/aosp_15_r20/external/pytorch/test/inductor/
H A Dtest_max_autotune.py257 choices: List[ChoiceCaller],
260 benchmark: Callable[[Any], Dict[ChoiceCaller, float]],