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

Rename FusedBatchNorm op to FusedBatchNormV2 Graph Rewriter.

Module Contents

Classes

RenameBatchNormOptimizer

Rename FusedBatchNorm op to FusedBatchNormV2.

class neural_compressor.adaptor.tf_utils.graph_rewriter.generic.rename_batch_norm.RenameBatchNormOptimizer(model)[source]: Rename FusedBatchNorm op to FusedBatchNormV2.