`neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.expanddims_optimizer`

ExpandDims Graph Rewriter.

Module Contents

Classes

ExpandDimsOptimizer

Calculate ExpandDims and remove it if its input is weight and next node is Conv2D.

class neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.expanddims_optimizer.ExpandDimsOptimizer(model)[source]: Calculate ExpandDims and remove it if its input is weight and next node is Conv2D.