`neural_compressor.adaptor.tf_utils.graph_rewriter.onnx.onnx_graph`

ONNX Graph wrapper for Tensorflow model converting to ONNX model.

Module Contents

Classes

Class that provides graph manipulation and matching.

class neural_compressor.adaptor.tf_utils.graph_rewriter.onnx.onnx_graph.OnnxGraph(nodes, output_shapes=None, dtypes=None, target=None, opset=None, extra_opset=None, input_names=None, output_names=None, is_subgraph=False, graph_name=None)[source]: Class that provides graph manipulation and matching.