Searched defs:model_iter_fn (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/benchmarks/dynamo/ |
| H A D | common.py | 465 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 D | dist_util.py | 83 def model_iter_fn(model, example_inputs, collect_outputs=False): function
|