Home
last modified time | relevance | path

Searched defs:model_iter_fn (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/benchmarks/dynamo/
H A Dcommon.py465 model_iter_fn, argument
561 def coverage_experiment(args, model_iter_fn, model, example_inputs): argument
597 def speedup_experiment_fx2trt(args, model_iter_fn, model, example_inputs): argument
606 def recompile_profiler_experiment(args, model_iter_fn, model, example_inputs): argument
649 def speedup_experiment(args, model_iter_fn, model, example_inputs, **kwargs): argument
807 def speedup_experiment_ds(args, model_iter_fn, model, example_inputs): argument
901 model_iter_fn, argument
1046 def overhead_experiment(*args, model_iter_fn): argument
1071 def baselines(models, model_iter_fn, example_inputs, args): argument
1121 def xla(args, model_iter_fn, model, example_inputs): argument
[all …]
H A Ddist_util.py83 def model_iter_fn(model, example_inputs, collect_outputs=False): function