Spaces:
Sleeping
Sleeping
File size: 1,929 Bytes
dc2106c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 |
# Copyright (c) ONNX Project Contributors
#
# SPDX-License-Identifier: Apache-2.0
import onnx
import onnx.onnx_cpp2py_export.parser as C # noqa: N812
class ParseError(Exception):
pass
def parse_model(model_text: str) -> onnx.ModelProto:
"""Parse a string to build a ModelProto.
Arguments:
model_text (string): formatted string
Returns:
ModelProto
"""
(success, msg, model_proto_str) = C.parse_model(model_text)
if success:
return onnx.load_from_string(model_proto_str)
raise ParseError(msg)
def parse_graph(graph_text: str) -> onnx.GraphProto:
"""Parse a string to build a GraphProto.
Arguments:
graph_text (string): formatted string
Returns:
GraphProto
"""
(success, msg, graph_proto_str) = C.parse_graph(graph_text)
if success:
graph_proto = onnx.GraphProto()
graph_proto.ParseFromString(graph_proto_str)
return graph_proto
raise ParseError(msg)
def parse_function(function_text: str) -> onnx.FunctionProto:
"""Parse a string to build a FunctionProto.
Arguments:
function_text (string): formatted string
Returns:
FunctionProto
"""
(success, msg, function_proto_str) = C.parse_function(function_text)
if success:
function_proto = onnx.FunctionProto()
function_proto.ParseFromString(function_proto_str)
return function_proto
raise ParseError(msg)
def parse_node(node_text: str) -> onnx.NodeProto:
"""Parse a string to build a NodeProto.
Arguments:
node_text: formatted string
Returns:
NodeProto
"""
(success, msg, node_proto_str) = C.parse_node(node_text)
if success:
node_proto = onnx.NodeProto()
node_proto.ParseFromString(node_proto_str)
return node_proto
raise ParseError(msg)
|