Home
last modified time | relevance | path

Searched refs:keep_inference_input_mutations (Results 1 – 16 of 16) sorted by relevance

/aosp_15_r20/external/pytorch/torch/_functorch/
H A Daot_autograd.py626 keep_input_mutations=aot_config.keep_inference_input_mutations,
668 keep_input_mutations=aot_config.keep_inference_input_mutations,
678 keep_input_mutations=aot_config.keep_inference_input_mutations,
775 keep_inference_input_mutations: bool = False,
846 keep_inference_input_mutations=keep_inference_input_mutations,
943 keep_inference_input_mutations=False, argument
1042 keep_inference_input_mutations=keep_inference_input_mutations,
1470 keep_inference_input_mutations=False,
/aosp_15_r20/external/pytorch/torch/_dynamo/backends/
H A Ddebugging.py104 keep_inference_input_mutations=True,
109 fw_compiler=boxed_nop, keep_inference_input_mutations=True
H A Dcudagraphs.py196 keep_inference_input_mutations=torch._dynamo.config.cudagraph_backend_keep_input_mutation,
/aosp_15_r20/external/pytorch/test/functorch/
H A Dtest_aotdispatch.py332 keep_inference_input_mutations=keep_input_mutations,
343 keep_inference_input_mutations=keep_input_mutations,
822 f_compiled = aot_function(f, nop, keep_inference_input_mutations=True)
1160 keep_inference_input_mutations=True,
1195 keep_inference_input_mutations=True,
1292 keep_inference_input_mutations=True,
1326 keep_inference_input_mutations=True,
1531 keep_inference_input_mutations=True,
2452 compiled_f = aot_function(fn, nop, keep_inference_input_mutations=True)
3094 keep_inference_input_mutations=False,
[all …]
/aosp_15_r20/external/pytorch/torch/_functorch/_aot_autograd/
H A Ddispatch_and_compile_graph.py81 keep_data_input_mutations=aot_config.keep_inference_input_mutations,
H A Dautograd_cache.py197 aot_config.keep_inference_input_mutations,
H A Dschemas.py811 keep_inference_input_mutations: bool
H A Dtraced_function_transforms.py486 if aot_config.keep_inference_input_mutations:
H A Druntime_wrappers.py146 keep_input_mutations=aot_config.keep_inference_input_mutations,
/aosp_15_r20/external/pytorch/torch/_higher_order_ops/
H A Dmap.py62 keep_inference_input_mutations=False,
H A Dutils.py301 keep_inference_input_mutations=False,
H A Dflex_attention.py498 keep_inference_input_mutations=False,
/aosp_15_r20/external/pytorch/test/dynamo/
H A Dtest_aot_autograd_cache.py566 keep_inference_input_mutations=False,
H A Dtest_repros.py190 keep_inference_input_mutations=True,
/aosp_15_r20/external/pytorch/torch/_inductor/
H A Dcompile_fx.py1519 keep_inference_input_mutations=True,
H A Dpattern_matcher.py1865 keep_inference_input_mutations=True,