API¶
Versions
The documentation was generated with the following versions.
<<<
import onnx
import onnx_array_api
import onnxruntime
import torch
import transformers
import onnxscript
for pck in [onnx, onnx_array_api, onnxruntime, torch, transformers, onnxscript]:
if pck is None:
continue
try:
print(f"{pck.__name__}: {pck.__version__}")
except AttributeError as e:
print(f"{pck.__name__}: {e}")
>>>
onnx: 1.17.0
onnx_array_api: 0.2.0
onnxruntime: 1.18.0+cu118
torch: 2.4.0.dev20240425+cu118
transformers: 4.39.3
onnxscript: module 'onnxscript' has no attribute '__version__'
Statistics
<<<
import os
import pandas
import experimental_experiment
from experimental_experiment.ext_test_case import (
statistics_on_file,
statistics_on_folder,
)
root = os.path.dirname(experimental_experiment.__file__)
stat = statistics_on_folder(
[
root,
os.path.join(root, "..", "_doc"),
os.path.join(root, "..", "_unittests"),
],
aggregation=2,
)
df = pandas.DataFrame(stat)
gr = df.drop("name", axis=1).groupby(["ext", "dir"]).sum().reset_index()
gr = gr[gr["dir"] != "_doc/auto_examples"]
print(gr)
print("--------------------")
print("total in experimental_experiment/...")
total = (
gr[gr["dir"].str.contains("experimental_experiment/")]
.drop(["ext", "dir"], axis=1)
.sum(axis=0)
)
print(total)
>>>
ext dir lines chars
0 .py _doc/ 123 4813
2 .py _doc/examples 3091 97441
3 .py _unittests/ut_convert_tools 164 5927
4 .py _unittests/ut_gradient 336 10712
5 .py _unittests/ut_issues 64 1938
6 .py _unittests/ut_onnxscript 1984 55835
7 .py _unittests/ut_plotting 17 541
8 .py _unittests/ut_reference 502 16979
9 .py _unittests/ut_torch_bench 117 3202
10 .py _unittests/ut_torch_dynamo 658 17265
11 .py _unittests/ut_torch_interpreter 5379 158442
12 .py _unittests/ut_torch_models 663 20344
13 .py _unittests/ut_xbuilder 189 6742
14 .py _unittests/ut_xoptim 6149 175892
15 .py _unittests/ut_xrun_custom_ops 458 12916
16 .py _unittests/ut_xrun_doc 498 14373
17 .py _unittests/ut_xrun_models 718 24801
18 .py experimental_experiment/ 1265 34213
19 .py experimental_experiment/convert 160 5305
20 .py experimental_experiment/gradient 925 30006
21 .py experimental_experiment/plotting 433 8174
22 .py experimental_experiment/reference 1014 28339
23 .py experimental_experiment/torch_bench 1410 44625
24 .py experimental_experiment/torch_dynamo 1258 37851
25 .py experimental_experiment/torch_interpreter 5928 152134
26 .py experimental_experiment/torch_models 827 26004
27 .py experimental_experiment/xbuilder 4930 141640
28 .py experimental_experiment/xoptim 6367 192150
29 .rst _doc/ 393 17266
30 .rst _doc/api 339 16163
32 .rst _doc/bench 272 20649
33 .rst _doc/design 809 34127
34 .rst _doc/models 704 23564
35 .rst _doc/tutorial 49 1629
--------------------
total in experimental_experiment/...
lines 24517
chars 700441
dtype: int64