`neural_compressor.adaptor.tf_utils.graph_rewriter.generic.fuse_gelu`

Fuse small ops to Gelu Graph Rewriter.

Module Contents

Classes

FuseGeluOptimizer

Fuse Sqrt + RealDiv + Erf + AddV2 + Mul + Mul into Gelu op.

class neural_compressor.adaptor.tf_utils.graph_rewriter.generic.fuse_gelu.FuseGeluOptimizer(model)[source]: Fuse Sqrt + RealDiv + Erf + AddV2 + Mul + Mul into Gelu op.