Searched defs:AOTConfig (Results 1 – 7 of 7) sorted by relevance
| /external/pytorch/torch/_functorch/_aot_autograd/ |
| D | dispatch_and_compile_graph.py | 45 def _create_graph(f, args, *, aot_config: AOTConfig) -> torch.fx.GraphModule:
67 aot_config: AOTConfig,
237 aot_config: AOTConfig,
|
| D | autograd_cache.py | 167 aot_config: AOTConfig,
189 def _reduce_aot_config(aot_config: AOTConfig):
231 config: AOTConfig,
320 aot_config: AOTConfig,
454 aot_config: AOTConfig,
|
| D | runtime_wrappers.py | 88 aot_config: AOTConfig,
137 aot_config: AOTConfig,
470 aot_config: AOTConfig,
564 aot_config: AOTConfig,
601 aot_config: AOTConfig,
618 _aot_config: AOTConfig,
780 aot_config: AOTConfig,
941 aot_config: AOTConfig,
1013 aot_config: AOTConfig,
1135 aot_config: AOTConfig,
[all …]
|
| D | jit_compile_runtime_wrappers.py | 88 aot_config: AOTConfig,
128 aot_config: AOTConfig,
315 aot_config: AOTConfig,
|
| D | traced_function_transforms.py | 192 def create_joint(fn: Callable, *, aot_config: AOTConfig) -> Any:
393 aot_config: AOTConfig,
|
| D | schemas.py | 800 class AOTConfig: class
|
| /external/pytorch/torch/_functorch/ |
| D | aot_autograd.py | 436 aot_config: AOTConfig,
503 flat_args: List[Any], aot_config: AOTConfig
517 aot_config: AOTConfig,
530 aot_config: AOTConfig,
|