neural_compressor.tensorflow.algorithms.static_quant.tensorflow

Tensorflow Adaptor Classes.

Module Contents

Classes

TensorFlowAdaptor

Adaptor Layer for stock tensorflow and spr-base.

Tensorflow_ITEXAdaptor

Tensorflow ITEX Adaptor Class.

TensorFlowConfig

Base config class for TensorFlow.

TensorflowQuery

Tensorflow Query Capability Class.

TensorflowConfigConverter

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.