`neural_compressor.adaptor.tf_utils.tf2onnx_converter`

Tensorflow QDQ model convert to ONNX QDQ model.

Module Contents

Classes

TensorflowQDQToOnnxQDQConverter

Convert tensorflow QDQ graph to ONNX QDQ graph.

class neural_compressor.adaptor.tf_utils.tf2onnx_converter.TensorflowQDQToOnnxQDQConverter(model, input_names, output_names, shape_override, inputs_as_nchw=None, opset_version=utils.DEFAULT_OPSET_VERSION)[source]: Convert tensorflow QDQ graph to ONNX QDQ graph.