neural_compressor.adaptor.tf_utils.graph_rewriter.generic.strip_equivalent_nodes

Strip Equivalent Nodes Graph Rewriter.

Module Contents

Classes

StripEquivalentNodesOptimizer

Remove the equivalent nodes which have the same inputs and attributes.

class neural_compressor.adaptor.tf_utils.graph_rewriter.generic.strip_equivalent_nodes.StripEquivalentNodesOptimizer(model, output_node_names)[source]

Remove the equivalent nodes which have the same inputs and attributes.