onnx.checker

CheckerContext

onnx.checker.DEFAULT_CONTEXT

别名

onnx.checker 模块

用于检查ONNX协议消息是否合法的图形工具。

exception onnx.checker.ValidationError

基础类: Exception

onnx.checker.check_attribute(attr: ~onnx.onnx_ml_pb2.AttributeProto, ctx: ~onnx.onnx_cpp2py_export.checker.CheckerContext = <onnx.onnx_cpp2py_export.checker.CheckerContext object>, lexical_scope_ctx: ~onnx.onnx_cpp2py_export.checker.LexicalScopeContext = <onnx.onnx_cpp2py_export.checker.LexicalScopeContext object>) → None[源代码]
onnx.checker.check_function(function: ~onnx.onnx_ml_pb2.FunctionProto, ctx: ~onnx.onnx_cpp2py_export.checker.CheckerContext | None = None, lexical_scope_ctx: ~onnx.onnx_cpp2py_export.checker.LexicalScopeContext = <onnx.onnx_cpp2py_export.checker.LexicalScopeContext object>) → None[source]
onnx.checker.check_graph(graph: ~onnx.onnx_ml_pb2.GraphProto, ctx: ~onnx.onnx_cpp2py_export.checker.CheckerContext = <onnx.onnx_cpp2py_export.checker.CheckerContext object>, lexical_scope_ctx: ~onnx.onnx_cpp2py_export.checker.LexicalScopeContext = <onnx.onnx_cpp2py_export.checker.LexicalScopeContext object>) → None[source]
onnx.checker.check_model(model: ModelProto | str | bytes | PathLike, full_check: bool = False, skip_opset_compatibility_check: bool = False, check_custom_domain: bool = False) → None[source]

检查模型的一致性。

如果模型的ir_version未正确设置或高于检查器的ir_version，或者如果模型在metadata_props中有重复的键，则会引发异常。

如果IR版本大于等于3，模型必须指定opset_import。 如果IR版本小于3，模型不能有任何opset_import指定。

Parameters:

  • model – 要检查的模型。如果模型是一个路径，函数会先检查模型路径。如果模型的字节大小超过2GB，应该使用模型路径调用函数。

  • full_check – 如果为True，函数还会运行形状推断检查。

  • skip_opset_compatibility_check – 如果为True，函数将跳过操作集兼容性检查。

  • check_custom_domain – 如果为True，函数将检查所有域。否则仅检查内置域。

onnx.checker.check_node(node: ~onnx.onnx_ml_pb2.NodeProto, ctx: ~onnx.onnx_cpp2py_export.checker.CheckerContext = <onnx.onnx_cpp2py_export.checker.CheckerContext object>, lexical_scope_ctx: ~onnx.onnx_cpp2py_export.checker.LexicalScopeContext = <onnx.onnx_cpp2py_export.checker.LexicalScopeContext object>) → None[source]
onnx.checker.check_sparse_tensor(sparse: ~onnx.onnx_ml_pb2.SparseTensorProto, ctx: ~onnx.onnx_cpp2py_export.checker.CheckerContext = <onnx.onnx_cpp2py_export.checker.CheckerContext object>) → None[source]
onnx.checker.check_tensor(tensor: ~onnx.onnx_ml_pb2.TensorProto, ctx: ~onnx.onnx_cpp2py_export.checker.CheckerContext = <onnx.onnx_cpp2py_export.checker.CheckerContext object>) → None[source]
onnx.checker.check_value_info(value_info: ~onnx.onnx_ml_pb2.ValueInfoProto, ctx: ~onnx.onnx_cpp2py_export.checker.CheckerContext = <onnx.onnx_cpp2py_export.checker.CheckerContext object>) → None[source]