neural_compressor.adaptor.tf_utils.graph_rewriter.generic.strip_unused_nodes

Strip unused nodes Graph Rewriter.

Module Contents

Classes

StripUnusedNodesOptimizer

Remove the unused nodes in the graph.

class neural_compressor.adaptor.tf_utils.graph_rewriter.generic.strip_unused_nodes.StripUnusedNodesOptimizer(model, input_node_names, output_node_names)[source]

Remove the unused nodes in the graph.