neural_compressor.adaptor.tf_utils.graph_rewriter.generic.fuse_pad_with_conv

Fuse Pad into Conv Graph Rewriter.

Module Contents

Classes

FusePadWithConv2DOptimizer

Fuse Pad op into Conv2D/DepthwiseConv2dNative/Conv3D.

class neural_compressor.adaptor.tf_utils.graph_rewriter.generic.fuse_pad_with_conv.FusePadWithConv2DOptimizer(model, excluded_op_names, inputs, cfg, new_api, itex_qdq_mode=False)[source]

Fuse Pad op into Conv2D/DepthwiseConv2dNative/Conv3D.