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
|