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/_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/runtime_typing.py
onnxscript/_internal/utils.py
onnxscript/_internal/version_utils.py
onnxscript/_legacy_ir/__init__.py
onnxscript/_legacy_ir/visitor.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/graph_building/_graph_building_torch.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/ir/passes/common/_c_api_utils.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/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/onnx_opset/_impl/opset_ai_onnx_preview_training1.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/broadcast_to_matmul.py
onnxscript/rewriter/cast_constant_of_shape.py
onnxscript/rewriter/collapse_slices.py
onnxscript/rewriter/fuse_batchnorm.py
onnxscript/rewriter/gemm_to_matmul_add.py
onnxscript/rewriter/generic_pattern.py
onnxscript/rewriter/llama_rule_sets.py
onnxscript/rewriter/no_op.py
onnxscript/rewriter/pattern.py
onnxscript/rewriter/testing.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/fuse_mha_bias.py
onnxscript/rewriter/ort_fusions/fuse_packed_qkv_gqa.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/group_normalization_merge_silu.py
onnxscript/rewriter/ort_fusions/instance_to_group_normalization.py
onnxscript/rewriter/ort_fusions/mha.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/ort_fusions/models/_rotary_embedding_models.py
onnxscript/rewriter/ort_fusions/models/_smollm_1.py
onnxscript/rewriter/ort_fusions/models/_smollm_2.py
onnxscript/rewriter/ort_fusions/models/_test_models.py
onnxscript/rewriter/ort_fusions/models/_whisper_decoder.py
onnxscript/rewriter/ort_fusions/models/_whisper_encoder.py
onnxscript/testing/__init__.py
onnxscript/tools/__init__.py
onnxscript/tools/memory_peak.py
onnxscript/tools/training_helper.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/_version_converter.py