Searched refs:inference_compiler (Results 1 – 7 of 7) sorted by relevance
| /external/pytorch/torch/_functorch/ |
| D | aot_autograd.py | 776 inference_compiler: Optional[Callable] = None,
836 if inference_compiler is None:
837 inference_compiler = fw_compiler
841 inference_compiler=inference_compiler,
944 inference_compiler: Optional[Callable] = None,
970 if inference_compiler is None:
971 inference_compiler = fw_compiler
1037 inference_compiler=inference_compiler,
1462 inference_compiler=None,
|
| /external/pytorch/torch/_inductor/ |
| D | compile_fx.py | 1432 inference_compiler = functools.partial(
1442 inference_compiler = functools.partial(fw_compiler_base, is_inference=True)
1508 return inference_compiler(unlifted_gm, example_inputs_)
1516 inference_compiler=inference_compiler,
|
| /external/pytorch/torch/_functorch/_aot_autograd/ |
| D | jit_compile_runtime_wrappers.py | 166 aot_config.inference_compiler
167 if aot_config.inference_compiler is not None
|
| D | schemas.py | 817 inference_compiler: Optional[Callable] = None variable in AOTConfig
|
| /external/pytorch/torch/_dynamo/backends/ |
| D | cudagraphs.py | 195 inference_compiler=functools.partial(forward_cudagraphs, is_inference=True),
|
| /external/pytorch/test/dynamo/ |
| D | test_aot_autograd_cache.py | 561 inference_compiler=None,
|
| /external/pytorch/test/functorch/ |
| D | test_aotdispatch.py | 5326 inference_compiler = make_boxed_compiler(
5329 aot_fn = aot_function(generate, nop, inference_compiler=inference_compiler)
5782 inference_compiler = make_boxed_compiler(
5785 compiled_fn = aot_function(fn, nop, inference_compiler=inference_compiler)
|