neural_compressor.adaptor.tf_utils.graph_util

Tensorflow Graph Utils Helper Classes.

Classes

GraphAnalyzer

Tensorflow Graph Analyzer class which implemented under singleton mode.

GraphRewriterHelper

Encapsulates the graph operation into one class.

Module Contents

class neural_compressor.adaptor.tf_utils.graph_util.GraphAnalyzer(extend_engine=None)[source]

Tensorflow Graph Analyzer class which implemented under singleton mode.

This class provides the following API: * Analyze the graph * Analyze the input/output node names of the specified graph

class neural_compressor.adaptor.tf_utils.graph_util.GraphRewriterHelper[source]

Encapsulates the graph operation into one class.