neural_compressor.tensorflow.quantization.utils.quantize_graph.qat.quantize_wrapper

QAT Quantize Wrapper Class.

Module Contents

Classes

QuantizeWrapperBase	Base class for quantize wrapper.
QuantizeWrapper	General QuantizeWrapper for quantizable layers.

class neural_compressor.tensorflow.quantization.utils.quantize_graph.qat.quantize_wrapper.QuantizeWrapperBase(layer, **kwargs)

Base class for quantize wrapper.

class neural_compressor.tensorflow.quantization.utils.quantize_graph.qat.quantize_wrapper.QuantizeWrapper(layer, **kwargs)

General QuantizeWrapper for quantizable layers.

Weights and inputs will be quantized according to the layer type and quantize config.