onnx.checker

CheckerContext

onnx.checker.DEFAULT_CONTEXT

別名為 <onnx.onnx_cpp2py_export.checker.CheckerContext 物件>

onnx.checker 模組

用於檢查 ONNX proto 訊息是否合法的圖形公用程式。

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 物件>, lexical_scope_ctx: ~onnx.onnx_cpp2py_export.checker.LexicalScopeContext = <onnx.onnx_cpp2py_export.checker.LexicalScopeContext 物件>) → 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 物件>) → None[原始碼]
onnx.checker.check_graph(graph: ~onnx.onnx_ml_pb2.GraphProto, ctx: ~onnx.onnx_cpp2py_export.checker.CheckerContext = <onnx.onnx_cpp2py_export.checker.CheckerContext 物件>, lexical_scope_ctx: ~onnx.onnx_cpp2py_export.checker.LexicalScopeContext = <onnx.onnx_cpp2py_export.checker.LexicalScopeContext 物件>) → 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 – 要檢查的模型。如果模型是路徑，則函數會先檢查模型路徑。如果模型位元組大小大於 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[原始碼]
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[原始碼]
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[原始碼]