`kedro_onnx.typing`#

Type aliases for Kedro ONNX plugin.

Module Contents#

Classes#

Protocol for ONNX model.

Attributes#

`OnnxFrameworks`	Literal for supported ONNX frameworks.
`IT`
`OT`

kedro_onnx.typing.OnnxFrameworks[source]#: Literal for supported ONNX frameworks.

kedro_onnx.typing.IT[source]#

kedro_onnx.typing.OT[source]#

class kedro_onnx.typing.ModelProto[source]#

Bases: Protocol

Protocol for ONNX model.

doc_string :str[source]#

domain :str[source]#

ir_version :int[source]#

model_version :int[source]#

producer_name :str[source]#

producer_version :str[source]#

graph :onnx.onnx_ml_pb2.GraphProto[source]#

ParseFromString(s: bytes) → None[source]#

Parse serialized model.

Parameters:: s (bytes) – Serialized model.

SerializeToString() → bytes[source]#

Serialize model.

Returns:: Serialized model.
Return type:: bytes