Skip to content

Commit 00480d1

Browse files
xaduprexadupre
authored
Handles sequences in TorchOnnxEvaluator (#125)
* Handles sequences * refactoring * changes * mypy * docuemntation * rename * doc
1 parent a37a6d5 commit 00480d1

32 files changed

+739
-186
lines changed

‎CHANGELOGS.rst‎

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,7 @@ Change Logs
44
0.6.1
55
+++++
66

7+
* :pr:`125`: handles sequences in TorchOnnxEvaluator
78
* :pr:`123`: add subgraphs to TorchOnnxEvaluator
89
* :pr:`122`: add local functions to TorchOnnxEvaluator
910
* :pr:`120`: enables TorchOnnxEvaluator in command line ``python -m onnx_diagnostic validate ...``

‎_doc/api/api.rst‎

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,7 @@
1+
2+
onnx_diagnostic.api
3+
===================
4+
5+
.. automodule:: onnx_diagnostic.api
6+
:members:
7+
:no-undoc-members:

‎_doc/api/index.rst‎

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -19,6 +19,7 @@ API of onnx_diagnostic
1919
:maxdepth: 1
2020
:caption: modules
2121

22+
api
2223
ext_test_case
2324

2425
.. automodule:: onnx_diagnostic

‎_doc/api/reference/torch_ops/controlflow_ops.rst‎

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,6 @@
1+
2+
onnx_diagnostic.reference.torch_ops.controlflow_ops
3+
===================================================
4+
5+
.. automodule:: onnx_diagnostic.reference.torch_ops.controlflow_ops
6+
:members:

‎_doc/api/reference/torch_ops/generator_ops.rst‎

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,6 @@
1+
2+
onnx_diagnostic.reference.torch_ops.generator_ops
3+
=================================================
4+
5+
.. automodule:: onnx_diagnostic.reference.torch_ops.generator_ops
6+
:members:

‎_doc/api/reference/torch_ops/index.rst‎

Lines changed: 20 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -9,19 +9,39 @@ onnx_diagnostic.reference.torch_ops
99

1010
access_ops
1111
binary_ops
12+
controlflow_ops
13+
generator_ops
14+
nn_ops
15+
other_ops
16+
reduce_ops
17+
sequence_ops
18+
shape_ops
19+
unary_ops
1220

1321
OpRun
1422
+++++
1523

1624
.. autoclass:: onnx_diagnostic.reference.torch_ops.OpRun
1725
:members:
1826

27+
OpRunTensor
28+
+++++++++++
29+
30+
.. autoclass:: onnx_diagnostic.reference.torch_ops.OpRunTensor
31+
:members:
32+
1933
OpRunValue
2034
++++++++++
2135

2236
.. autoclass:: onnx_diagnostic.reference.torch_ops.OpRunValue
2337
:members:
2438

39+
OpRunSequence
40+
+++++++++++++
41+
42+
.. autoclass:: onnx_diagnostic.reference.torch_ops.OpRunSequence
43+
:members:
44+
2545
OpRunFunction
2646
+++++++++++++
2747

‎_doc/api/reference/torch_ops/nn_ops.rst‎

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,6 @@
1+
2+
onnx_diagnostic.reference.torch_ops.nn_ops
3+
==========================================
4+
5+
.. automodule:: onnx_diagnostic.reference.torch_ops.nn_ops
6+
:members:

‎_doc/api/reference/torch_ops/other_ops.rst‎

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,6 @@
1+
2+
onnx_diagnostic.reference.torch_ops.other_ops
3+
=============================================
4+
5+
.. automodule:: onnx_diagnostic.reference.torch_ops.other_ops
6+
:members:

‎_doc/api/reference/torch_ops/reduce_ops.rst‎

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,6 @@
1+
2+
onnx_diagnostic.reference.torch_ops.reduce_ops
3+
==============================================
4+
5+
.. automodule:: onnx_diagnostic.reference.torch_ops.reduce_ops
6+
:members:

‎_doc/api/reference/torch_ops/sequence_ops.rst‎

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,6 @@
1+
2+
onnx_diagnostic.reference.torch_ops.sequence_ops
3+
================================================
4+
5+
.. automodule:: onnx_diagnostic.reference.torch_ops.sequence_ops
6+
:members:

0 commit comments

Comments
 (0)