Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/benchmarks/record_function_benchmark/
H A Drecord_function_bench.py12 def prepare_lstm_jit(bench_args): argument
25 def prepare_resnet50_jit(bench_args): argument
40 def run_bench(model_names, bench_args): argument
/aosp_15_r20/external/pytorch/benchmarks/fastrnns/
H A Dbench.py225 def bench_group(model_list, bench_name, bench_group, bench_args): argument
329 bench_args = copy.deepcopy(vars(args)) variable