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

Folding BatchNorm Graph Rewriter.

Module Contents

Classes

FoldBatchNormNodesOptimizer

Folding BatchNorm nodes into Conv.

class neural_compressor.adaptor.tf_utils.graph_rewriter.generic.fold_batch_norm.FoldBatchNormNodesOptimizer(model)[source]: Folding BatchNorm nodes into Conv.