`neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.grappler_pass`

Tensorflow Grappler Graph Rewriter.

Module Contents

Classes

GrapplerOptimizer

A python wrapper that leverages the built-in tensorflow grappler API to optimize the graph.

class neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.grappler_pass.GrapplerOptimizer(model, input_output_names, opt_cfg)[source]: A python wrapper that leverages the built-in tensorflow grappler API to optimize the graph.