neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fuse_conv_with_math

Fuse Conv with Math Graph Rewriter.

Module Contents

Classes

FuseConvWithMathOptimizer

Convert below subgraph to Conv2D + BiasAdd by eliminating math ops.
class neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fuse_conv_with_math.FuseConvWithMathOptimizer(model)[source]

Convert below subgraph to Conv2D + BiasAdd by eliminating math ops.

Conv2D Conv2D
|

Sub | | —-> |

RealDiv |
|

Mul | | |

BiasAdd BiasAdd