neural_compressor.ux.utils.workload.graph_optimization

Configuration graph_optimization module.

Module Contents

Classes

GraphOptimization

Configuration Graph Optimization class.

class neural_compressor.ux.utils.workload.graph_optimization.GraphOptimization(data: Dict[str, Any] = {})

Bases: neural_compressor.ux.utils.json_serializer.JsonSerializer

Configuration Graph Optimization class.

set_precisions(precisions: str | List[str]) → None

Update graph_optimization precisions in config.