neural_compressor.adaptor.tf_utils.graph_rewriter.onnx.onnx_schema

ONNX Operator Schemas for Tensorflow model converting to ONNX model.

Module Contents

Classes

OnnxOpSchema

Wrapper for Onnx schema.

Functions

get_schema(name, max_inclusive_opset_version[, domain])

Get schema by name within specific version.

get_max_supported_opset_version([domain])

Get max supported opset version by current onnx package given a domain.

class neural_compressor.adaptor.tf_utils.graph_rewriter.onnx.onnx_schema.OnnxOpSchema(name, domain, since_version, attributes)[source]

Wrapper for Onnx schema.

neural_compressor.adaptor.tf_utils.graph_rewriter.onnx.onnx_schema.get_schema(name, max_inclusive_opset_version, domain=None)[source]

Get schema by name within specific version.

neural_compressor.adaptor.tf_utils.graph_rewriter.onnx.onnx_schema.get_max_supported_opset_version(domain=None)[source]

Get max supported opset version by current onnx package given a domain.