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