neural_compressor.tensorflow.quantization.utils.quantize_graph.qat.fake_quantize

QAT Fake Quantize Graph Class.

Module Contents

Classes

FakeQuantizeBase

ABC interface class for applying fake quantization by insert qdq.

FakeQuantize

The class that applies fake quantization.

class neural_compressor.tensorflow.quantization.utils.quantize_graph.qat.fake_quantize.FakeQuantizeBase[source]

ABC interface class for applying fake quantization by insert qdq.

class neural_compressor.tensorflow.quantization.utils.quantize_graph.qat.fake_quantize.FakeQuantize(per_channel=False, num_bits=8, channel_axis=-1, symmetric=True, narrow_range=True)[source]

The class that applies fake quantization.