neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fuse_column_wise_mul

Fuse Columnwise Mul Graph Rewriter.

Module Contents

Classes

FuseColumnWiseMulOptimizer

Fuse Mul op into Conv2D/DepthwiseConv2dNative/MatMul.

class neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fuse_column_wise_mul.FuseColumnWiseMulOptimizer(model)[source]

Fuse Mul op into Conv2D/DepthwiseConv2dNative/MatMul.