Searched refs:global_setup (Results 1 – 8 of 8) sorted by relevance
| /external/pytorch/torch/utils/benchmark/utils/ |
| D | cpp_jit.py | 113 global_setup: str,
118 ("// GLOBAL_SETUP_TEMPLATE_LOCATION", global_setup, 0),
155 def compile_timeit_template(*, stmt: str, setup: str, global_setup: str) -> TimeitModuleType:
160 …module = _compile_template(stmt=stmt, setup=setup, global_setup=global_setup, src=src, is_standalo…
165 def compile_callgrind_template(*, stmt: str, setup: str, global_setup: str) -> str:
170 …target = _compile_template(stmt=stmt, setup=setup, global_setup=global_setup, src=src, is_standalo…
|
| D | timer.py | 42 global_setup: str,
61 self._global_setup: str = textwrap.dedent(global_setup)
69 global_setup=self._global_setup,
189 global_setup: str = "",
211 if global_setup:
222 timer_kwargs["global_setup"] = global_setup
253 global_setup=global_setup,
|
| D | common.py | 35 global_setup: str = "" variable in TaskSpec
|
| /external/pytorch/benchmarks/instruction_counts/core/ |
| D | expand.py | 249 global_setup: str = ""
251 global_setup = textwrap.dedent(
263 global_setup=global_setup,
|
| D | api.py | 63 global_setup: str = "" variable in GroupedSetup
274 global_setup=cpp_global_setup,
437 global_setup = "\n".join(lines_by_label.pop("GLOBAL_SETUP"))
439 return lines_by_label, setup, global_setup
|
| /external/pytorch/benchmarks/instruction_counts/worker/ |
| D | main.py | 75 global_setup: str = "" variable in WorkerTimerArgs
138 global_setup=timer_args.global_setup,
|
| /external/pytorch/torch/utils/benchmark/utils/valgrind_wrapper/ |
| D | timer_interface.py | 637 global_setup=task_spec.global_setup,
|
| /external/pytorch/test/benchmark_utils/ |
| D | test_benchmark_utils.py | 214 global_setup="#define TIMER_GLOBAL_CHECK",
|