Introduce OpSignature to IR

[justinchuby]

Introduce OpSignature accessible from the .signature property of all OpLike objects (traced function, onnx function and op). The OpSignature class leverages the IR to represent the signature of an operator, preserving ordering of all inputs and provides easy to work with type representations.

The PR also deprecates the ParamSchema class and properties.

Fixes #1697

The next PR will replace param_schemas usage.

[codecov]

❌ 1 Tests Failed:

Tests completed	Failed	Passed	Skipped
14189	1	14188	2656

View the top 1 failed tests by shortest run time

onnxscript.backend.onnx_export_test.TestOnnxBackEnd test_export2python_produces_correct_onnx_script_model_0043_test_argmax_keepdims_random_select_last_index

Stack Traces | 0.004s run time

onnxscript\backend\onnx_export_test.py:137: in extract_functions
    mod = importlib.import_module(import_name)
C:\hostedtoolcache\windows\Python\3.11.9\x64\Lib\importlib\__init__.py:126: in import_module
    return _bootstrap._gcd_import(name[level:], package, level)
E   ModuleNotFoundError: No module named 'tests.onnx_backend_test_code.test_argmax_keepdims_random_select_last_index'

The above exception was the direct cause of the following exception:
.nox\test\Lib\site-packages\parameterized\parameterized.py:620: in standalone_func
    return func(*(a + p.args), **p.kwargs, **kw)
onnxscript\backend\onnx_export_test.py:271: in test_export2python_produces_correct_onnx_script_model
    functions = extract_functions(backend_test.name, code, self.test_folder)
onnxscript\backend\onnx_export_test.py:139: in extract_functions
    raise AssertionError(
E   AssertionError: Unable to import 'tests.onnx_backend_test_code.test_argmax_keepdims_random_select_last_index' (e=No module named 'tests.onnx_backend_test_code.test_argmax_keepdims_random_select_last_index') (file: 'D:\\a\\onnxscript\\onnxscript\\tests\\onnx_backend_test_code\\test_argmax_keepdims_random_select_last_index.py', absolute path: 'D:\\a\\onnxscript\\onnxscript\\tests\\onnx_backend_test_code\\test_argmax_keepdims_random_select_last_index.py', current folder: D:\a\onnxscript\onnxscript
E   ---- CONTENT --
E   import numpy
E   from onnx import TensorProto
E   from onnx.helper import make_tensor
E   from onnxscript import script, external_tensor
E   from onnxscript.values import Opset
E   from onnxscript.onnx_types import FLOAT, INT64
E   from onnxscript.onnx_opset import opset13
E   
E   @script()
E   def bck_test_argmax_keepdims_random_select_last_index(data: FLOAT[2,3,4]) -> (INT64[2,1,4]):
E       result = opset13.ArgMax(data, axis=1, keepdims=1, select_last_index=1)
E       return result

To view individual test run time comparison to the main branch, go to the Test Analytics Dashboard

[github-actions]

Test Results

     24 files  ±  0      24 suites  ±0   2h 44m 6s ⏱️ + 4m 1s
 14 152 tests + 34  11 778 ✅ + 36    2 336 💤 ±0   32 ❌  - 2   6 🔥 ±0 
355 111 runs  +238  81 240 ✅ +240  273 617 💤 ±0  212 ❌  - 2  42 🔥 ±0 

For more details on these failures and errors, see this check.

Results for commit ff1aaa5. ± Comparison against base commit fac4825.

♻️ This comment has been updated with latest results.

[titaiwangms]

Should we put a comment saying this should align to torch _schema.py until we move it to here?

[titaiwangms]

Actually, we only have the test for this file, which means that torch's _schema.py should be following this one.

[justinchuby]

When the signature property is defined, PyTorch onnx exporter will not generate a new signature, and instead use what is here. However, since it is doing isinstance checks on the signature the exporter itself defines, we get isinstance errors there.