neural_compressor.adaptor.tf_utils.graph_rewriter.graph_base

Graph Rewrite Base Class.

Module Contents

Classes

GraphRewriterBase

Graph Rewrite Base class.

class neural_compressor.adaptor.tf_utils.graph_rewriter.graph_base.GraphRewriterBase(model)[source]

Graph Rewrite Base class.

We abstract this base class and define the interface only.

Parameters:

object (model) – the input model to be converted.