neural_compressor.tensorflow.algorithms.static_quant.tensorflow
Tensorflow Adaptor Classes.
Module Contents
Classes
Adaptor Layer for stock tensorflow and spr-base. |
|
Tensorflow ITEX Adaptor Class. |
|
Base config class for TensorFlow. |
|
Tensorflow Query Capability Class. |
|
Convert StaticQuantConfig to the format used by static quant algo. |
- class neural_compressor.tensorflow.algorithms.static_quant.tensorflow.TensorFlowAdaptor(framework_specific_info)[source]
Adaptor Layer for stock tensorflow and spr-base.
- class neural_compressor.tensorflow.algorithms.static_quant.tensorflow.Tensorflow_ITEXAdaptor(framework_specific_info)[source]
Tensorflow ITEX Adaptor Class.
- class neural_compressor.tensorflow.algorithms.static_quant.tensorflow.TensorFlowConfig(precisions=None)[source]
Base config class for TensorFlow.
- class neural_compressor.tensorflow.algorithms.static_quant.tensorflow.TensorflowQuery(local_config_file=None, performance_only=False, itex_mode=False, quant_mode='static')[source]
Tensorflow Query Capability Class.
- class neural_compressor.tensorflow.algorithms.static_quant.tensorflow.TensorflowConfigConverter(quant_config: neural_compressor.tensorflow.quantization.config.StaticQuantConfig, capability: Dict)[source]
Convert StaticQuantConfig to the format used by static quant algo.