Searched defs:CompilerFn (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/_dynamo/ |
| H A D | convert_frame.py | 416 compiler_fn: CompilerFn,
546 compiler_fn: CompilerFn,
582 compiler_fn: CompilerFn,
1043 def __init__(self, compiler_fn: CompilerFn, hooks: Hooks) -> None:
1135 def convert_frame(compiler_fn: CompilerFn, hooks: Hooks) -> ConvertFrame:
|
| H A D | output_graph.py | 194 def __init__(self, backend: CompilerFn):
245 compiler_fn: Optional[CompilerFn],
|
| /aosp_15_r20/external/pytorch/torch/_dynamo/backends/ |
| H A D | registry.py | 21 CompilerFn = Callable[[fx.GraphModule, List[torch.Tensor]], CompiledFn] variable
28 compiler_fn: Optional[CompilerFn] = None,
|