LICENSE
MANIFEST.in
README.md
VERSION
pyproject.toml
setup.py
onnxscript/__init__.py
onnxscript/converter.py
onnxscript/evaluator.py
onnxscript/irbuilder.py
onnxscript/main.py
onnxscript/onnx_types.py
onnxscript/py.typed
onnxscript/sourceinfo.py
onnxscript/tensor.py
onnxscript/type_annotation.py
onnxscript/values.py
onnxscript.egg-info/PKG-INFO
onnxscript.egg-info/SOURCES.txt
onnxscript.egg-info/dependency_links.txt
onnxscript.egg-info/requires.txt
onnxscript.egg-info/top_level.txt
onnxscript/_framework_apis/__init__.py
onnxscript/_framework_apis/torch_2_5.py
onnxscript/_framework_apis/torch_2_6.py
onnxscript/_framework_apis/torch_2_7.py
onnxscript/_framework_apis/torch_2_8.py
onnxscript/_framework_apis/torch_2_9.py
onnxscript/_internal/__init__.py
onnxscript/_internal/analysis.py
onnxscript/_internal/ast_utils.py
onnxscript/_internal/autocast.py
onnxscript/_internal/deprecation.py
onnxscript/_internal/param_manipulation.py
onnxscript/_internal/utils.py
onnxscript/_internal/version_utils.py
onnxscript/_thirdparty/asciichartpy.py
onnxscript/backend/__init__.py
onnxscript/backend/onnx_backend.py
onnxscript/backend/onnx_export.py
onnxscript/function_libs/tools/torch_lib/deduce_type_constraints.py
onnxscript/function_libs/tools/torch_lib/generate_aten_signatures.py
onnxscript/function_libs/tools/torch_lib/generate_prims_signatures.py
onnxscript/function_libs/torch_lib/__init__.py
onnxscript/function_libs/torch_lib/_constants.py
onnxscript/function_libs/torch_lib/_flags.py
onnxscript/function_libs/torch_lib/registration.py
onnxscript/function_libs/torch_lib/tensor_typing.py
onnxscript/function_libs/torch_lib/graph_building/__init__.py
onnxscript/function_libs/torch_lib/ops/__init__.py
onnxscript/function_libs/torch_lib/ops/common.py
onnxscript/function_libs/torch_lib/ops/core.py
onnxscript/function_libs/torch_lib/ops/fft.py
onnxscript/function_libs/torch_lib/ops/linalg.py
onnxscript/function_libs/torch_lib/ops/nested.py
onnxscript/function_libs/torch_lib/ops/nn.py
onnxscript/function_libs/torch_lib/ops/prims.py
onnxscript/function_libs/torch_lib/ops/quantized_decomposed.py
onnxscript/function_libs/torch_lib/ops/sparse.py
onnxscript/function_libs/torch_lib/ops/special.py
onnxscript/function_libs/torch_lib/ops/vision.py
onnxscript/ir/__init__.py
onnxscript/ir/_schemas.py
onnxscript/ir/_tape.py
onnxscript/ir/convenience.py
onnxscript/ir/passes/__init__.py
onnxscript/ir/passes/common/__init__.py
onnxscript/onnx_opset/__init__.py
onnxscript/onnx_opset/_impl/opset1.py
onnxscript/onnx_opset/_impl/opset10.py
onnxscript/onnx_opset/_impl/opset11.py
onnxscript/onnx_opset/_impl/opset12.py
onnxscript/onnx_opset/_impl/opset13.py
onnxscript/onnx_opset/_impl/opset14.py
onnxscript/onnx_opset/_impl/opset15.py
onnxscript/onnx_opset/_impl/opset16.py
onnxscript/onnx_opset/_impl/opset17.py
onnxscript/onnx_opset/_impl/opset18.py
onnxscript/onnx_opset/_impl/opset19.py
onnxscript/onnx_opset/_impl/opset2.py
onnxscript/onnx_opset/_impl/opset20.py
onnxscript/onnx_opset/_impl/opset21.py
onnxscript/onnx_opset/_impl/opset22.py
onnxscript/onnx_opset/_impl/opset23.py
onnxscript/onnx_opset/_impl/opset24.py
onnxscript/onnx_opset/_impl/opset3.py
onnxscript/onnx_opset/_impl/opset4.py
onnxscript/onnx_opset/_impl/opset5.py
onnxscript/onnx_opset/_impl/opset6.py
onnxscript/onnx_opset/_impl/opset7.py
onnxscript/onnx_opset/_impl/opset8.py
onnxscript/onnx_opset/_impl/opset9.py
onnxscript/onnx_opset/_impl/opset_ai_onnx_ml1.py
onnxscript/onnx_opset/_impl/opset_ai_onnx_ml2.py
onnxscript/onnx_opset/_impl/opset_ai_onnx_ml3.py
onnxscript/onnx_opset/_impl/opset_ai_onnx_ml4.py
onnxscript/onnx_opset/_impl/opset_ai_onnx_ml5.py
onnxscript/optimizer/__init__.py
onnxscript/optimizer/_constant_folding.py
onnxscript/optimizer/_optimizer.py
onnxscript/rewriter/__init__.py
onnxscript/rewriter/_basics.py
onnxscript/rewriter/_fusion_utils.py
onnxscript/rewriter/_ir_utils.py
onnxscript/rewriter/_matcher.py
onnxscript/rewriter/_pattern_ir.py
onnxscript/rewriter/_rewrite_rule.py
onnxscript/rewriter/pattern.py
onnxscript/rewriter/testing.py
onnxscript/rewriter/models/_bart_encoder.py
onnxscript/rewriter/models/_phi2lm.py
onnxscript/rewriter/models/_phi4lm.py
onnxscript/rewriter/models/_rotary_embedding_models.py
onnxscript/rewriter/models/_smollm_1.py
onnxscript/rewriter/models/_smollm_2.py
onnxscript/rewriter/models/_test_models.py
onnxscript/rewriter/models/_whisper_decoder.py
onnxscript/rewriter/models/_whisper_encoder.py
onnxscript/rewriter/onnx_fusions/__init__.py
onnxscript/rewriter/onnx_fusions/_onnx_fusions.py
onnxscript/rewriter/onnxruntime/__init__.py
onnxscript/rewriter/onnxruntime/bfloat16_utils/bfloat16_converter.py
onnxscript/rewriter/ort_fusions/__init__.py
onnxscript/rewriter/ort_fusions/_core.py
onnxscript/rewriter/ort_fusions/_test_utils.py
onnxscript/rewriter/ort_fusions/attention.py
onnxscript/rewriter/ort_fusions/bias_gelu.py
onnxscript/rewriter/ort_fusions/cos_sin_cache.py
onnxscript/rewriter/ort_fusions/erfgelu.py
onnxscript/rewriter/ort_fusions/fused_matmul_rule_sets.py
onnxscript/rewriter/ort_fusions/gelu.py
onnxscript/rewriter/ort_fusions/gqa.py
onnxscript/rewriter/ort_fusions/gqa_packed_qkv.py
onnxscript/rewriter/ort_fusions/group_normalization_merge_silu.py
onnxscript/rewriter/ort_fusions/instance_to_group_normalization.py
onnxscript/rewriter/ort_fusions/mha.py
onnxscript/rewriter/ort_fusions/mha_bias.py
onnxscript/rewriter/ort_fusions/mha_scale.py
onnxscript/rewriter/ort_fusions/rms_normalization.py
onnxscript/rewriter/ort_fusions/rotary_embedding.py
onnxscript/rewriter/ort_fusions/sdpa.py
onnxscript/rewriter/ort_fusions/sdpa_via_mha.py
onnxscript/rewriter/ort_fusions/shape_optimization.py
onnxscript/rewriter/ort_fusions/skip_normalization.py
onnxscript/rewriter/ort_fusions/softmax.py
onnxscript/rewriter/rules/__init__.py
onnxscript/rewriter/rules/common/__init__.py
onnxscript/rewriter/rules/common/_basic_rules.py
onnxscript/rewriter/rules/common/_broadcast_to_matmul.py
onnxscript/rewriter/rules/common/_cast_constant_of_shape.py
onnxscript/rewriter/rules/common/_collapse_slices.py
onnxscript/rewriter/rules/common/_fuse_batchnorm.py
onnxscript/rewriter/rules/common/_fuse_conv_affine.py
onnxscript/rewriter/rules/common/_fuse_hardswish.py
onnxscript/rewriter/rules/common/_fuse_pad_into_conv.py
onnxscript/rewriter/rules/common/_fuse_relus_clips.py
onnxscript/rewriter/rules/common/_gemm_to_matmul_add.py
onnxscript/rewriter/rules/common/_matmul_add_to_gemm.py
onnxscript/rewriter/rules/common/_min_max_to_clip.py
onnxscript/rewriter/rules/common/_no_op.py
onnxscript/rewriter/rules/common/_redundant_scatter_nd.py
onnxscript/rewriter/rules/fusion/__init__.py
onnxscript/rewriter/rules/fusion/_gqa.py
onnxscript/rewriter/rules/fusion/_layer_norm.py
onnxscript/rewriter/rules/fusion/_rms_normalization.py
onnxscript/rewriter/rules/fusion/_rotary_embedding.py
onnxscript/testing/__init__.py
onnxscript/tools/__init__.py
onnxscript/tools/memory_peak.py
onnxscript/tools/transformers_models/__init__.py
onnxscript/tools/transformers_models/llama.py
onnxscript/tools/transformers_models/mistral.py
onnxscript/tools/transformers_models/phi.py
onnxscript/tools/transformers_models/phi3.py
onnxscript/utils/__init__.py
onnxscript/utils/evaluation_utils.py
onnxscript/utils/timing_utils.py
onnxscript/utils/utils.py
onnxscript/version_converter/__init__.py
onnxscript/version_converter/_c_api_utils.py
onnxscript/version_converter/_version_converter.py