neural_compressor.ux.components.optimization.pruning.optimize_model

Pruning optimization script.

Module Contents

Functions

parse_args(→ Any)

Parse input arguments.

optimize_model(→ None)

Execute pruning optimization.

neural_compressor.ux.components.optimization.pruning.optimize_model.parse_args() Any

Parse input arguments.

neural_compressor.ux.components.optimization.pruning.optimize_model.optimize_model(input_graph: str, output_graph: str, framework: str, config: str) None

Execute pruning optimization.