experimental_experiment.torch_bench._bash_bench_set_torchbench¶
- class experimental_experiment.torch_bench._bash_bench_set_torchbench.TorchBenchRunner(device: str, partition_id: int = 0, total_partitions: int = 1, include_model_names: Set[str] | None = None, exclude_model_names: Set[str] | None = None, verbose: int = 0, warmup: int = 10, repeat: int = 30, fake_tensor: bool = False, no_grad: bool = True, target_opset: int = 18, dtype: Any | None = None, nvtx: bool = False, dump_ort: bool = False)[source]¶