onnx.checker¶
CheckerContext¶
- onnx.checker.DEFAULT_CONTEXT¶
<onnx.onnx_cpp2py_export.checker.CheckerContext object> 的别名
onnx.checker
模块¶
用于检查 ONNX proto 消息是否合法的图工具。
- 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 [源码]¶
- 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 [源码]¶
- 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 [源码]¶
检查模型的一致性。
如果模型的 ir_version 未正确设置或高于检查器的 ir_version,或者模型的 metadata_props 中有重复的键,将引发异常。
如果 IR 版本 >= 3,模型必须指定 opset_import。如果 IR 版本 < 3,模型不能指定任何 opset_import。
- 参数:
model – 要检查的模型。如果 model 是一个路径,该函数会首先检查模型路径。如果模型的字节大小超过 2GB,应使用模型路径调用该函数。
full_check – 如果为 True,该函数还会运行形状推断检查。
skip_opset_compatibility_check – 如果为 True,该函数会跳过 opset 兼容性检查。
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 [源码]¶
- 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 [源码]¶