neural_compressor.tensorflow.quantization.utils.graph_util

Tensorflow Graph Utils Helper Classes.

Module Contents

Classes

GraphAnalyzer

Tensorflow Graph Analyzer class which implemented under singleton mode.

GraphRewriterHelper

Encapsulates the graph operation into one class.

class neural_compressor.tensorflow.quantization.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.tensorflow.quantization.utils.graph_util.GraphRewriterHelper[source]

Encapsulates the graph operation into one class.