Searched defs:print_per_iter (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/benchmarks/operator_benchmark/ |
| H A D | benchmark_pytorch.py | 126 def run_jit_forward(self, num_runs, print_per_iter=False, cuda_sync=False): argument
148 def run_forward(self, num_runs, print_per_iter, cuda_sync): argument
174 def run_backward(self, num_runs, print_per_iter=False): argument
|
| H A D | benchmark_core.py | 270 def _launch_forward(self, test_case, iters, print_per_iter): argument
281 def _launch_backward(self, test_case, iters, print_per_iter=False): argument
292 def _measure_time(self, launch_test, test_case, iters, print_per_iter): argument
|