-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy path__init__.py
More file actions
58 lines (45 loc) · 1.34 KB
/
__init__.py
File metadata and controls
58 lines (45 loc) · 1.34 KB
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
from typing import Dict, Optional
from onnx import ModelProto
from ..annotations import domain
from .model import OnnxGraph, ProtoType
from .var import Var, Vars
def start(
opset: Optional[int] = None,
opsets: Optional[Dict[str, int]] = None,
ir_version: Optional[int] = None,
) -> OnnxGraph:
"""
Starts an onnx model.
:param opset: main opset version
:param opsets: others opsets as a dictionary
:param ir_version: specify the ir_version as well
:return: an instance of :class:`onnx_array_api.light_api.OnnxGraph`
A very simple model:
.. runpython::
:showcode:
from onnx_array_api.light_api import start
onx = start().vin("X").Neg().rename("Y").vout().to_onnx()
print(onx)
Another with operator Add:
.. runpython::
:showcode:
from onnx_array_api.light_api import start
onx = (
start()
.vin("X")
.vin("Y")
.bring("X", "Y")
.Add()
.rename("Z")
.vout()
.to_onnx()
)
print(onx)
"""
return OnnxGraph(opset=opset, opsets=opsets, ir_version=ir_version)
def g() -> OnnxGraph:
"""
Starts a subgraph.
:return: an instance of :class:`onnx_array_api.light_api.OnnxGraph`
"""
return OnnxGraph(proto_type=ProtoType.GRAPH)