neural_compressor.tensorflow.quantization.utils.quantize_graph.quantize_graph_pooling

Quantize the AvgPool/MaxPool.

Module Contents

Classes

FuseNodeStartWithPooling

Quantize the AvgPool and MaxPool.

class neural_compressor.tensorflow.quantization.utils.quantize_graph.quantize_graph_pooling.FuseNodeStartWithPooling(**kwargs)[source]

Quantize the AvgPool and MaxPool.