neural_compressor.ux.components.benchmark.benchmark_model

Generic benchmark script.

Module Contents

Functions

parse_args(→ Any)	Parse input arguments.
benchmark_model(→ None)	Execute benchmark.
set_eager_execution(→ None)	Set eager execution as required by model.

neural_compressor.ux.components.benchmark.benchmark_model.parse_args() → Any

Parse input arguments.

neural_compressor.ux.components.benchmark.benchmark_model.benchmark_model(input_graph: str, config: str, benchmark_mode: str, framework: str) → None

Execute benchmark.

neural_compressor.ux.components.benchmark.benchmark_model.set_eager_execution(input_graph: str) → None

Set eager execution as required by model.