├── .azure-pipelines ├── publish-dev.yml ├── publish.yml ├── release-dev.yml ├── release.yml └── stages │ ├── jobs │ └── steps │ │ └── release-steps.yml │ └── release-stage.yml ├── .editorconfig ├── .gitattributes ├── .github ├── codeql │ └── codeql-config.yml ├── copilot-instructions.md ├── dependabot.yaml ├── release.yml └── workflows │ ├── codeql-analysis.yml │ ├── lint.yaml │ ├── main.yaml │ └── pages.yaml ├── .gitignore ├── .lintrunner.toml ├── .vscode └── extensions.json ├── CODE_OF_CONDUCT.md ├── CONTRIBUTING.md ├── LICENSE ├── MANIFEST.in ├── README.md ├── ROADMAP.md ├── SECURITY.md ├── SUPPORT.md ├── VERSION ├── codecov.yml ├── docs ├── _static │ ├── css │ │ └── custom.css │ ├── logo-dark.png │ ├── logo-light.png │ ├── logo.ai │ └── logo_main.png ├── _templates │ ├── classtemplate.rst │ ├── classtemplate_inherited.rst │ └── functiontemplate.rst ├── api │ ├── converter.md │ ├── decorator.md │ ├── index.md │ ├── opsets.md │ ├── optimizer.md │ ├── rewriter.md │ ├── rewriter_pattern.md │ ├── testing.md │ ├── values.md │ └── version_converter.md ├── articles │ ├── 2023-06-20-introducing-onnx-script │ │ ├── debugging_screenshot.png │ │ ├── index.md │ │ ├── intellisense_screenshot.png │ │ └── ten_chunks_model.png │ └── index.md ├── conf.py ├── examples │ ├── 01_plot_selu.py │ ├── 02_plot_square_loss.py │ ├── 03_export_lib.py │ ├── 04_plot_eager_mode_evaluation.py │ ├── 05_plot_model_props.py │ ├── 06_plot_model_local_funs.py │ └── README.txt ├── index.md ├── ir │ └── index.md ├── test │ └── test_documentation_examples.py ├── tutorial │ ├── examples │ │ ├── dropout.py │ │ ├── firstdim.py │ │ ├── forloop.py │ │ ├── forwhileloop.py │ │ ├── hardmax_end_to_end.py │ │ ├── leaky_relu.py │ │ ├── leaky_relu_attr_promoted.py │ │ ├── omitted_input.py │ │ ├── outerscope_redef_error.py │ │ ├── scanloop.py │ │ ├── softplus.py │ │ ├── tensor_attr.py │ │ ├── tensor_attr2.py │ │ ├── tensor_attr_short.py │ │ └── whileloop.py │ ├── index.md │ ├── optimizer │ │ ├── index.md │ │ └── optimize.md │ └── rewriter │ │ ├── allow_other_inputs.md │ │ ├── attributes.md │ │ ├── commute.md │ │ ├── conditional_rewrite.md │ │ ├── domain_option.md │ │ ├── examples │ │ ├── allow_other_attributes.py │ │ ├── allow_other_inputs.py │ │ ├── broadcast_matmul.py │ │ ├── domain_option.py │ │ ├── erfgelu.py │ │ ├── img │ │ │ ├── broadcast_01.png │ │ │ ├── broadcast_02.png │ │ │ ├── erfgelu_01.png │ │ │ ├── erfgelu_02.png │ │ │ ├── erfgelu_03_commute.png │ │ │ ├── erfgelu_04_commute.png │ │ │ ├── erfgelu_05_commute.png │ │ │ ├── erfgelu_06_commute.png │ │ │ └── erfgelu_07_commute.png │ │ ├── or_pattern.py │ │ └── outputs_option.py │ │ ├── index.md │ │ ├── node_value_checkers.md │ │ ├── or_pattern.md │ │ ├── outputs_option.md │ │ ├── rewrite_patterns.md │ │ └── simple_example.md └── update_readme.py ├── examples ├── custom_op_expansion.py ├── pattern_matching_example.py └── pattern_rewriting.py ├── noxfile.py ├── onnxscript ├── __init__.py ├── _framework_apis │ ├── __init__.py │ ├── torch_2_5.py │ ├── torch_2_6.py │ ├── torch_2_7.py │ ├── torch_2_8.py │ └── torch_2_9.py ├── _internal │ ├── __init__.py │ ├── analysis.py │ ├── analysis_test.py │ ├── ast_utils.py │ ├── autocast.py │ ├── deprecation.py │ ├── param_manipulation.py │ ├── param_manipulation_test.py │ ├── utils.py │ └── version_utils.py ├── _thirdparty │ ├── README.md │ └── asciichartpy.py ├── backend │ ├── __init__.py │ ├── onnx_backend.py │ ├── onnx_backend_test.py │ ├── onnx_export.py │ └── onnx_export_test.py ├── converter.py ├── converter_test.py ├── evaluator.py ├── evaluator_test.py ├── function_libs │ ├── tools │ │ └── torch_lib │ │ │ ├── deduce_type_constraints.py │ │ │ ├── deduce_type_constraints_test.py │ │ │ ├── generate_aten_signatures.py │ │ │ └── generate_prims_signatures.py │ └── torch_lib │ │ ├── __init__.py │ │ ├── _constants.py │ │ ├── _flags.py │ │ ├── graph_building │ │ └── __init__.py │ │ ├── ops │ │ ├── __init__.py │ │ ├── common.py │ │ ├── core.py │ │ ├── fft.py │ │ ├── linalg.py │ │ ├── nested.py │ │ ├── nn.py │ │ ├── prims.py │ │ ├── quantized_decomposed.py │ │ ├── sparse.py │ │ ├── special.py │ │ └── vision.py │ │ ├── registration.py │ │ └── tensor_typing.py ├── ir │ ├── README.md │ ├── __init__.py │ ├── _schemas.py │ ├── _schemas_test.py │ ├── _tape.py │ ├── _tape_test.py │ ├── convenience.py │ └── passes │ │ ├── __init__.py │ │ └── common │ │ └── __init__.py ├── irbuilder.py ├── main.py ├── onnx_opset │ ├── __init__.py │ └── _impl │ │ ├── opset1.py │ │ ├── opset10.py │ │ ├── opset11.py │ │ ├── opset12.py │ │ ├── opset13.py │ │ ├── opset14.py │ │ ├── opset15.py │ │ ├── opset16.py │ │ ├── opset17.py │ │ ├── opset18.py │ │ ├── opset19.py │ │ ├── opset2.py │ │ ├── opset20.py │ │ ├── opset21.py │ │ ├── opset22.py │ │ ├── opset23.py │ │ ├── opset24.py │ │ ├── opset3.py │ │ ├── opset4.py │ │ ├── opset5.py │ │ ├── opset6.py │ │ ├── opset7.py │ │ ├── opset8.py │ │ ├── opset9.py │ │ ├── opset_ai_onnx_ml1.py │ │ ├── opset_ai_onnx_ml2.py │ │ ├── opset_ai_onnx_ml3.py │ │ ├── opset_ai_onnx_ml4.py │ │ └── opset_ai_onnx_ml5.py ├── onnx_types.py ├── optimizer │ ├── __init__.py │ ├── _constant_folding.py │ ├── _constant_folding_test.py │ ├── _function_folding_test.py │ ├── _optimizer.py │ └── _optimizer_test.py ├── py.typed ├── rewriter │ ├── __init__.py │ ├── _basics.py │ ├── _fusion_utils.py │ ├── _ir_utils.py │ ├── _matcher.py │ ├── _pattern_ir.py │ ├── _pattern_ir_test.py │ ├── _rewrite_rule.py │ ├── match_context_test.py │ ├── models │ │ ├── _bart_encoder.py │ │ ├── _phi2lm.py │ │ ├── _phi4lm.py │ │ ├── _rotary_embedding_models.py │ │ ├── _smollm_1.py │ │ ├── _smollm_2.py │ │ ├── _test_models.py │ │ ├── _whisper_decoder.py │ │ └── _whisper_encoder.py │ ├── onnx_fusions │ │ ├── __init__.py │ │ └── _onnx_fusions.py │ ├── onnxruntime │ │ ├── README.md │ │ ├── __init__.py │ │ └── bfloat16_utils │ │ │ ├── bfloat16_converter.py │ │ │ └── bfloat16_converter_test.py │ ├── ort_fusions │ │ ├── __init__.py │ │ ├── _core.py │ │ ├── _test_utils.py │ │ ├── attention.py │ │ ├── attention_test.py │ │ ├── bias_gelu.py │ │ ├── bias_gelu_test.py │ │ ├── cos_sin_cache.py │ │ ├── cos_sin_cache_test.py │ │ ├── erfgelu.py │ │ ├── fuse_xformers_test.py │ │ ├── fused_matmul_rule_sets.py │ │ ├── fused_matmul_rule_sets_test.py │ │ ├── gelu.py │ │ ├── gelu_test.py │ │ ├── gqa.py │ │ ├── gqa_packed_qkv.py │ │ ├── gqa_packed_qkv_test.py │ │ ├── gqa_test.py │ │ ├── group_normalization_merge_silu.py │ │ ├── group_normalization_merge_silu_test.py │ │ ├── instance_to_group_normalization.py │ │ ├── instance_to_group_normalization_test.py │ │ ├── mha.py │ │ ├── mha_bias.py │ │ ├── mha_scale.py │ │ ├── mha_test.py │ │ ├── rms_normalization.py │ │ ├── rms_normalization_test.py │ │ ├── rotary_embedding.py │ │ ├── rotary_embedding_test.py │ │ ├── sdpa.py │ │ ├── sdpa_test.py │ │ ├── sdpa_via_mha.py │ │ ├── shape_optimization.py │ │ ├── shape_optimization_test.py │ │ ├── skip_normalization.py │ │ ├── skip_normalization_test.py │ │ ├── softmax.py │ │ └── softmax_test.py │ ├── pattern.py │ ├── pattern_base_test.py │ ├── pattern_test.py │ ├── rules │ │ ├── __init__.py │ │ ├── common │ │ │ ├── __init__.py │ │ │ ├── _basic_rules.py │ │ │ ├── _basic_rules_test.py │ │ │ ├── _broadcast_to_matmul.py │ │ │ ├── _broadcast_to_matmul_test.py │ │ │ ├── _cast_constant_of_shape.py │ │ │ ├── _cast_constant_of_shape_test.py │ │ │ ├── _collapse_slices.py │ │ │ ├── _collapse_slices_test.py │ │ │ ├── _fuse_batchnorm.py │ │ │ ├── _fuse_batchnorm_test.py │ │ │ ├── _fuse_conv_affine.py │ │ │ ├── _fuse_conv_affine_test.py │ │ │ ├── _fuse_hardswish.py │ │ │ ├── _fuse_hardswish_test.py │ │ │ ├── _fuse_pad_into_conv.py │ │ │ ├── _fuse_pad_into_conv_test.py │ │ │ ├── _fuse_relus_clips.py │ │ │ ├── _fuse_relus_clips_test.py │ │ │ ├── _gemm_to_matmul_add.py │ │ │ ├── _gemm_to_matmul_add_test.py │ │ │ ├── _matmul_add_to_gemm.py │ │ │ ├── _matmul_add_to_gemm_test.py │ │ │ ├── _min_max_to_clip.py │ │ │ ├── _min_max_to_clip_test.py │ │ │ ├── _no_op.py │ │ │ ├── _no_op_test.py │ │ │ ├── _redundant_scatter_nd.py │ │ │ ├── _redundant_scatter_nd_test.py │ │ │ ├── _remove_optional_bias.py │ │ │ └── _remove_optional_bias_test.py │ │ └── fusion │ │ │ ├── __init__.py │ │ │ ├── _gqa.py │ │ │ ├── _gqa_test.py │ │ │ ├── _layer_norm.py │ │ │ ├── _layer_norm_test.py │ │ │ ├── _rms_normalization.py │ │ │ ├── _rms_normalization_test.py │ │ │ ├── _rotary_embedding.py │ │ │ └── _rotary_embedding_test.py │ └── testing.py ├── sourceinfo.py ├── tensor.py ├── tensor_test.py ├── testing │ └── __init__.py ├── tools │ ├── __init__.py │ ├── memory_peak.py │ ├── memory_peak_test.py │ └── transformers_models │ │ ├── __init__.py │ │ ├── llama.py │ │ ├── llama_test.py │ │ ├── mistral.py │ │ ├── mistral_test.py │ │ ├── phi.py │ │ ├── phi3.py │ │ ├── phi3_test.py │ │ └── phi_test.py ├── type_annotation.py ├── type_annotation_test.py ├── utils │ ├── __init__.py │ ├── evaluation_utils.py │ ├── metadata_merger.py │ ├── replace.py │ ├── timing_utils.py │ └── utils.py ├── values.py ├── values_test.py └── version_converter │ ├── __init__.py │ ├── _c_api_utils.py │ ├── _version_converter.py │ └── _version_converter_test.py ├── opgen ├── README.md ├── __init__.py ├── __main__.py ├── onnx_opset_builder.py ├── pygen.py └── pygen_test.py ├── pyproject.toml ├── pyproject_pylint.toml ├── requirements-dev.txt ├── requirements ├── ci │ ├── requirements-onnx-weekly.txt │ ├── requirements-ort-nightly.txt │ └── requirements-pytorch-nightly.txt └── lintrunner │ └── requirements.txt ├── setup.py ├── testdata ├── dort_models │ └── llama_forward.onnx ├── e2e_models │ ├── Speech2Text2ForCausalLM │ │ └── dynamo │ │ │ ├── Speech2Text2ForCausalLM_dynamo.onnx │ │ │ └── test_data_set_0 │ │ │ ├── input_0.pb │ │ │ ├── input_1.pb │ │ │ ├── output_0.pb │ │ │ ├── output_1.pb │ │ │ ├── output_10.pb │ │ │ ├── output_11.pb │ │ │ ├── output_12.pb │ │ │ ├── output_13.pb │ │ │ ├── output_2.pb │ │ │ ├── output_3.pb │ │ │ ├── output_4.pb │ │ │ ├── output_5.pb │ │ │ ├── output_6.pb │ │ │ ├── output_7.pb │ │ │ ├── output_8.pb │ │ │ └── output_9.pb │ ├── mobilenetv2_100 │ │ └── dynamo │ │ │ ├── mobilenetv2_100_dynamo.onnx │ │ │ └── test_data_set_0 │ │ │ ├── input_0.pb │ │ │ └── output_0.pb │ ├── resnet18 │ │ └── dynamo │ │ │ ├── resnet18_dynamo.onnx │ │ │ └── test_data_set_0 │ │ │ ├── input_0.pb │ │ │ └── output_0.pb │ └── torchscript_model │ │ └── torchscript_model.onnx └── unittest_models │ ├── attn_llama2_4_34_0 │ ├── attn_llama2_4_34_0.onnx │ └── test_data_set_0 │ │ ├── input_0.pb │ │ ├── input_1.pb │ │ ├── input_2.pb │ │ ├── input_3.pb │ │ ├── input_4.pb │ │ ├── input_5.pb │ │ ├── input_6.pb │ │ ├── input_7.pb │ │ ├── input_8.pb │ │ ├── output_0.pb │ │ ├── output_1.pb │ │ └── output_2.pb │ ├── attn_llama2_4_34_1 │ ├── attn_llama2_4_34_1.onnx │ └── test_data_set_0 │ │ ├── input_0.pb │ │ ├── input_1.pb │ │ ├── input_2.pb │ │ ├── input_3.pb │ │ ├── input_4.pb │ │ ├── input_5.pb │ │ ├── input_6.pb │ │ ├── input_7.pb │ │ ├── input_8.pb │ │ ├── output_0.pb │ │ ├── output_1.pb │ │ └── output_2.pb │ ├── attn_llama2_4_36_0 │ ├── attn_llama2_4_36_0.onnx │ └── test_data_set_0 │ │ ├── input_0.pb │ │ ├── input_1.pb │ │ ├── input_2.pb │ │ ├── input_3.pb │ │ ├── input_4.pb │ │ ├── input_5.pb │ │ ├── input_6.pb │ │ ├── input_7.pb │ │ ├── input_8.pb │ │ ├── output_0.pb │ │ ├── output_1.pb │ │ └── output_2.pb │ ├── attn_phi_1_5_0 │ ├── attn_phi_1_5_0.onnx │ └── test_data_set_0 │ │ ├── input_0.pb │ │ ├── input_1.pb │ │ ├── input_10.pb │ │ ├── input_11.pb │ │ ├── input_12.pb │ │ ├── input_2.pb │ │ ├── input_3.pb │ │ ├── input_4.pb │ │ ├── input_5.pb │ │ ├── input_6.pb │ │ ├── input_7.pb │ │ ├── input_8.pb │ │ ├── input_9.pb │ │ ├── output_0.pb │ │ ├── output_1.pb │ │ └── output_2.pb │ ├── attn_phi_1_5_1 │ ├── attn_phi_1_5_1.onnx │ └── test_data_set_0 │ │ ├── input_0.pb │ │ ├── input_1.pb │ │ ├── input_10.pb │ │ ├── input_11.pb │ │ ├── input_12.pb │ │ ├── input_2.pb │ │ ├── input_3.pb │ │ ├── input_4.pb │ │ ├── input_5.pb │ │ ├── input_6.pb │ │ ├── input_7.pb │ │ ├── input_8.pb │ │ ├── input_9.pb │ │ ├── output_0.pb │ │ ├── output_1.pb │ │ └── output_2.pb │ ├── attn_phi_1_5_2 │ ├── attn_phi_1_5_2.onnx │ └── test_data_set_0 │ │ ├── input_0.pb │ │ ├── input_1.pb │ │ ├── input_10.pb │ │ ├── input_11.pb │ │ ├── input_12.pb │ │ ├── input_2.pb │ │ ├── input_3.pb │ │ ├── input_4.pb │ │ ├── input_5.pb │ │ ├── input_6.pb │ │ ├── input_7.pb │ │ ├── input_8.pb │ │ ├── input_9.pb │ │ ├── output_0.pb │ │ ├── output_1.pb │ │ └── output_2.pb │ ├── attn_phi_1_5_3 │ ├── attn_phi_1_5_3.onnx │ └── test_data_set_0 │ │ ├── input_0.pb │ │ ├── input_1.pb │ │ ├── input_10.pb │ │ ├── input_11.pb │ │ ├── input_12.pb │ │ ├── input_2.pb │ │ ├── input_3.pb │ │ ├── input_4.pb │ │ ├── input_5.pb │ │ ├── input_6.pb │ │ ├── input_7.pb │ │ ├── input_8.pb │ │ ├── input_9.pb │ │ ├── output_0.pb │ │ ├── output_1.pb │ │ └── output_2.pb │ ├── attn_stable_diffusion_unet_0 │ ├── attn_stable_diffusion_unet_0.onnx │ └── test_data_set_0 │ │ ├── input_0.pb │ │ ├── input_1.pb │ │ ├── input_2.pb │ │ ├── input_3.pb │ │ ├── input_4.pb │ │ ├── input_5.pb │ │ ├── input_6.pb │ │ └── output_0.pb │ ├── attn_stable_diffusion_unet_1 │ ├── attn_stable_diffusion_unet_1.onnx │ └── test_data_set_0 │ │ ├── input_0.pb │ │ ├── input_1.pb │ │ ├── input_2.pb │ │ ├── input_3.pb │ │ ├── input_4.pb │ │ ├── input_5.pb │ │ ├── input_6.pb │ │ └── output_0.pb │ ├── attn_stable_diffusion_unet_without_encoder_hidden_states_0 │ ├── attn_stable_diffusion_unet_without_encoder_hidden_states_0.onnx │ └── test_data_set_0 │ │ ├── input_0.pb │ │ ├── input_1.pb │ │ ├── input_2.pb │ │ ├── input_3.pb │ │ ├── input_4.pb │ │ ├── input_5.pb │ │ └── output_0.pb │ ├── attn_stable_diffusion_unet_without_encoder_hidden_states_1 │ ├── attn_stable_diffusion_unet_without_encoder_hidden_states_1.onnx │ └── test_data_set_0 │ │ ├── input_0.pb │ │ ├── input_1.pb │ │ ├── input_2.pb │ │ ├── input_3.pb │ │ ├── input_4.pb │ │ ├── input_5.pb │ │ └── output_0.pb │ ├── attn_yi_4_37_0 │ ├── attn_yi_4_37_0.onnx │ └── test_data_set_0 │ │ ├── input_0.pb │ │ ├── input_1.pb │ │ ├── input_2.pb │ │ ├── input_3.pb │ │ ├── input_4.pb │ │ ├── input_5.pb │ │ ├── input_6.pb │ │ ├── input_7.pb │ │ ├── input_8.pb │ │ ├── output_0.pb │ │ ├── output_1.pb │ │ └── output_2.pb │ ├── geglu_stable_diffusion_unet_0 │ ├── geglu_stable_diffusion_unet_0.onnx │ └── test_data_set_0 │ │ ├── input_0.pb │ │ ├── input_1.pb │ │ ├── input_2.pb │ │ └── output_0.pb │ ├── geglu_stable_diffusion_unet_1 │ ├── geglu_stable_diffusion_unet_1.onnx │ └── test_data_set_0 │ │ ├── input_0.pb │ │ ├── input_1.pb │ │ ├── input_2.pb │ │ └── output_0.pb │ ├── geglu_stable_diffusion_unet_2 │ ├── geglu_stable_diffusion_unet_2.onnx │ └── test_data_set_0 │ │ ├── input_0.pb │ │ ├── input_1.pb │ │ ├── input_2.pb │ │ └── output_0.pb │ ├── geglu_stable_diffusion_unet_3 │ ├── geglu_stable_diffusion_unet_3.onnx │ └── test_data_set_0 │ │ ├── input_0.pb │ │ ├── input_1.pb │ │ ├── input_2.pb │ │ └── output_0.pb │ ├── gelu_phi_1_5_0 │ ├── gelu_phi_1_5_0.onnx │ └── test_data_set_0 │ │ ├── input_0.pb │ │ └── output_0.pb │ ├── gelu_phi_1_5_1 │ ├── gelu_phi_1_5_1.onnx │ └── test_data_set_0 │ │ ├── input_0.pb │ │ └── output_0.pb │ ├── gelu_phi_1_5_2 │ ├── gelu_phi_1_5_2.onnx │ └── test_data_set_0 │ │ ├── input_0.pb │ │ └── output_0.pb │ ├── gelu_phi_1_5_3 │ ├── gelu_phi_1_5_3.onnx │ └── test_data_set_0 │ │ ├── input_0.pb │ │ └── output_0.pb │ ├── ln_llama2_0 │ ├── ln_llama2_0.onnx │ └── test_data_set_0 │ │ ├── input_0.pb │ │ ├── input_1.pb │ │ └── output_0.pb │ ├── ln_llama2_1 │ ├── ln_llama2_1.onnx │ └── test_data_set_0 │ │ ├── input_0.pb │ │ ├── input_1.pb │ │ └── output_0.pb │ ├── ln_llama2_2 │ ├── ln_llama2_2.onnx │ └── test_data_set_0 │ │ ├── input_0.pb │ │ ├── input_1.pb │ │ └── output_0.pb │ ├── ln_llama2_3 │ ├── ln_llama2_3.onnx │ └── test_data_set_0 │ │ ├── input_0.pb │ │ ├── input_1.pb │ │ └── output_0.pb │ ├── sdpa_llama2_0 │ ├── sdpa_llama2_0.onnx │ └── test_data_set_0 │ │ ├── input_0.pb │ │ ├── input_1.pb │ │ ├── input_2.pb │ │ ├── input_3.pb │ │ ├── input_4.pb │ │ ├── input_5.pb │ │ ├── input_6.pb │ │ ├── input_7.pb │ │ ├── output_0.pb │ │ ├── output_1.pb │ │ └── output_2.pb │ ├── sdpa_llama2_1 │ ├── sdpa_llama2_1.onnx │ └── test_data_set_0 │ │ ├── input_0.pb │ │ ├── input_1.pb │ │ ├── input_2.pb │ │ ├── input_3.pb │ │ ├── input_4.pb │ │ ├── input_5.pb │ │ ├── input_6.pb │ │ ├── input_7.pb │ │ ├── output_0.pb │ │ ├── output_1.pb │ │ └── output_2.pb │ ├── sdpa_llama2_2 │ ├── sdpa_llama2_2.onnx │ └── test_data_set_0 │ │ ├── input_0.pb │ │ ├── input_1.pb │ │ ├── input_2.pb │ │ ├── input_3.pb │ │ ├── input_4.pb │ │ ├── input_5.pb │ │ ├── input_6.pb │ │ ├── input_7.pb │ │ ├── output_0.pb │ │ ├── output_1.pb │ │ └── output_2.pb │ ├── sdpa_llama2_3 │ ├── sdpa_llama2_3.onnx │ └── test_data_set_0 │ │ ├── input_0.pb │ │ ├── input_1.pb │ │ ├── input_2.pb │ │ ├── input_3.pb │ │ ├── input_4.pb │ │ ├── input_5.pb │ │ ├── input_6.pb │ │ ├── input_7.pb │ │ ├── output_0.pb │ │ ├── output_1.pb │ │ └── output_2.pb │ ├── sdpa_llama2_4_38_0 │ ├── sdpa_llama2_4_38_0.onnx │ └── test_data_set_0 │ │ ├── input_0.pb │ │ ├── input_1.pb │ │ ├── input_2.pb │ │ ├── input_3.pb │ │ ├── input_4.pb │ │ ├── input_5.pb │ │ ├── input_6.pb │ │ ├── input_7.pb │ │ ├── input_8.pb │ │ ├── output_0.pb │ │ ├── output_1.pb │ │ └── output_2.pb │ ├── sdpa_yi_0 │ ├── sdpa_yi_0.onnx │ └── test_data_set_0 │ │ ├── input_0.pb │ │ ├── input_1.pb │ │ ├── input_2.pb │ │ ├── input_3.pb │ │ ├── input_4.pb │ │ ├── input_5.pb │ │ ├── input_6.pb │ │ ├── input_7.pb │ │ ├── output_0.pb │ │ ├── output_1.pb │ │ └── output_2.pb │ ├── sdpa_yi_1 │ ├── sdpa_yi_1.onnx │ └── test_data_set_0 │ │ ├── input_0.pb │ │ ├── input_1.pb │ │ ├── input_2.pb │ │ ├── input_3.pb │ │ ├── input_4.pb │ │ ├── input_5.pb │ │ ├── input_6.pb │ │ ├── input_7.pb │ │ ├── output_0.pb │ │ ├── output_1.pb │ │ └── output_2.pb │ └── sdpa_yi_4_38_0 │ ├── sdpa_yi_4_38_0.onnx │ └── test_data_set_0 │ ├── input_0.pb │ ├── input_1.pb │ ├── input_2.pb │ ├── input_3.pb │ ├── input_4.pb │ ├── input_5.pb │ ├── input_6.pb │ ├── input_7.pb │ ├── input_8.pb │ ├── output_0.pb │ ├── output_1.pb │ └── output_2.pb ├── tests ├── README.md ├── __init__.py ├── common │ ├── __init__.py │ ├── onnx_script_test_case.py │ └── testutils.py ├── eager_mode_test.py ├── eager_test.py ├── external_tensor_test.py ├── function_libs │ ├── __init__.py │ └── torch_lib │ │ ├── README.md │ │ ├── __init__.py │ │ ├── e2e_ops_tests.py │ │ ├── error_reproduction.py │ │ ├── extra_opinfo.py │ │ ├── ops_test.py │ │ ├── ops_test_common.py │ │ └── ops_test_data.py ├── functions │ ├── attr_test.py │ ├── gemmgelu.py │ ├── gemmgelu_test.py │ ├── if_test.py │ ├── onnxfns1A_test.py │ ├── onnxfns2_test.py │ ├── onnxfns_test.py │ └── ort_custom_ops.py ├── if_test.py ├── ir │ ├── graph_view_test.py │ └── serde_roundtrip_test.py ├── loop_test.py ├── models │ ├── __init__.py │ ├── attrref.py │ ├── cast_like.py │ ├── different_opset.py │ ├── dropout.py │ ├── eager_op.py │ ├── eg1.py │ ├── getitem.py │ ├── graph_attr.py │ ├── identity.py │ ├── if_statement.py │ ├── loops_break.py │ ├── loops_while.py │ ├── m1.py │ ├── multi.py │ ├── onnxfns1.py │ ├── onnxfns1A.py │ ├── onnxfns2.py │ ├── opt_input.py │ ├── opt_output.py │ ├── renaming.py │ ├── sequences.py │ ├── signal_dft.py │ ├── subfunction.py │ └── type_double.py ├── onnx_types_test.py ├── operator_test.py ├── optimizer │ └── test_models.py └── version_converter │ └── version_conversion_test.py └── tools ├── ir └── model_zoo_test │ └── model_zoo_test.py ├── onnx2external.py ├── onnx2script.py ├── optimize.py └── ort_rewriter_profiling ├── README.md ├── bench_model.py ├── nsys_profile.py ├── ort_rewrite.py └── profile_analysis.py /.azure-pipelines/publish-dev.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/.azure-pipelines/publish-dev.yml -------------------------------------------------------------------------------- /.azure-pipelines/publish.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/.azure-pipelines/publish.yml -------------------------------------------------------------------------------- /.azure-pipelines/release-dev.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/.azure-pipelines/release-dev.yml -------------------------------------------------------------------------------- /.azure-pipelines/release.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/.azure-pipelines/release.yml -------------------------------------------------------------------------------- /.azure-pipelines/stages/jobs/steps/release-steps.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/.azure-pipelines/stages/jobs/steps/release-steps.yml -------------------------------------------------------------------------------- /.azure-pipelines/stages/release-stage.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/.azure-pipelines/stages/release-stage.yml -------------------------------------------------------------------------------- /.editorconfig: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/.editorconfig -------------------------------------------------------------------------------- /.gitattributes: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/.gitattributes -------------------------------------------------------------------------------- /.github/codeql/codeql-config.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/.github/codeql/codeql-config.yml -------------------------------------------------------------------------------- /.github/copilot-instructions.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/.github/copilot-instructions.md -------------------------------------------------------------------------------- /.github/dependabot.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/.github/dependabot.yaml -------------------------------------------------------------------------------- /.github/release.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/.github/release.yml -------------------------------------------------------------------------------- /.github/workflows/codeql-analysis.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/.github/workflows/codeql-analysis.yml -------------------------------------------------------------------------------- /.github/workflows/lint.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/.github/workflows/lint.yaml -------------------------------------------------------------------------------- /.github/workflows/main.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/.github/workflows/main.yaml -------------------------------------------------------------------------------- /.github/workflows/pages.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/.github/workflows/pages.yaml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/.gitignore -------------------------------------------------------------------------------- /.lintrunner.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/.lintrunner.toml -------------------------------------------------------------------------------- /.vscode/extensions.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/.vscode/extensions.json -------------------------------------------------------------------------------- /CODE_OF_CONDUCT.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/CODE_OF_CONDUCT.md -------------------------------------------------------------------------------- /CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/CONTRIBUTING.md -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/LICENSE -------------------------------------------------------------------------------- /MANIFEST.in: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/MANIFEST.in -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/README.md -------------------------------------------------------------------------------- /ROADMAP.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/ROADMAP.md -------------------------------------------------------------------------------- /SECURITY.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/SECURITY.md -------------------------------------------------------------------------------- /SUPPORT.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/SUPPORT.md -------------------------------------------------------------------------------- /VERSION: -------------------------------------------------------------------------------- 1 | 0.5.7 2 | -------------------------------------------------------------------------------- /codecov.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/codecov.yml -------------------------------------------------------------------------------- /docs/_static/css/custom.css: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/docs/_static/css/custom.css -------------------------------------------------------------------------------- /docs/_static/logo-dark.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/docs/_static/logo-dark.png -------------------------------------------------------------------------------- /docs/_static/logo-light.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/docs/_static/logo-light.png -------------------------------------------------------------------------------- /docs/_static/logo.ai: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/docs/_static/logo.ai -------------------------------------------------------------------------------- /docs/_static/logo_main.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/docs/_static/logo_main.png -------------------------------------------------------------------------------- /docs/_templates/classtemplate.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/docs/_templates/classtemplate.rst -------------------------------------------------------------------------------- /docs/_templates/classtemplate_inherited.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/docs/_templates/classtemplate_inherited.rst -------------------------------------------------------------------------------- /docs/_templates/functiontemplate.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/docs/_templates/functiontemplate.rst -------------------------------------------------------------------------------- /docs/api/converter.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/docs/api/converter.md -------------------------------------------------------------------------------- /docs/api/decorator.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/docs/api/decorator.md -------------------------------------------------------------------------------- /docs/api/index.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/docs/api/index.md -------------------------------------------------------------------------------- /docs/api/opsets.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/docs/api/opsets.md -------------------------------------------------------------------------------- /docs/api/optimizer.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/docs/api/optimizer.md -------------------------------------------------------------------------------- /docs/api/rewriter.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/docs/api/rewriter.md -------------------------------------------------------------------------------- /docs/api/rewriter_pattern.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/docs/api/rewriter_pattern.md -------------------------------------------------------------------------------- /docs/api/testing.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/docs/api/testing.md -------------------------------------------------------------------------------- /docs/api/values.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/docs/api/values.md -------------------------------------------------------------------------------- /docs/api/version_converter.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/docs/api/version_converter.md -------------------------------------------------------------------------------- /docs/articles/2023-06-20-introducing-onnx-script/debugging_screenshot.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/docs/articles/2023-06-20-introducing-onnx-script/debugging_screenshot.png -------------------------------------------------------------------------------- /docs/articles/2023-06-20-introducing-onnx-script/index.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/docs/articles/2023-06-20-introducing-onnx-script/index.md -------------------------------------------------------------------------------- /docs/articles/2023-06-20-introducing-onnx-script/intellisense_screenshot.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/docs/articles/2023-06-20-introducing-onnx-script/intellisense_screenshot.png -------------------------------------------------------------------------------- /docs/articles/2023-06-20-introducing-onnx-script/ten_chunks_model.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/docs/articles/2023-06-20-introducing-onnx-script/ten_chunks_model.png -------------------------------------------------------------------------------- /docs/articles/index.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/docs/articles/index.md -------------------------------------------------------------------------------- /docs/conf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/docs/conf.py -------------------------------------------------------------------------------- /docs/examples/01_plot_selu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/docs/examples/01_plot_selu.py -------------------------------------------------------------------------------- /docs/examples/02_plot_square_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/docs/examples/02_plot_square_loss.py -------------------------------------------------------------------------------- /docs/examples/03_export_lib.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/docs/examples/03_export_lib.py -------------------------------------------------------------------------------- /docs/examples/04_plot_eager_mode_evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/docs/examples/04_plot_eager_mode_evaluation.py -------------------------------------------------------------------------------- /docs/examples/05_plot_model_props.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/docs/examples/05_plot_model_props.py -------------------------------------------------------------------------------- /docs/examples/06_plot_model_local_funs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/docs/examples/06_plot_model_local_funs.py -------------------------------------------------------------------------------- /docs/examples/README.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/docs/examples/README.txt -------------------------------------------------------------------------------- /docs/index.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/docs/index.md -------------------------------------------------------------------------------- /docs/ir/index.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/docs/ir/index.md -------------------------------------------------------------------------------- /docs/test/test_documentation_examples.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/docs/test/test_documentation_examples.py -------------------------------------------------------------------------------- /docs/tutorial/examples/dropout.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/docs/tutorial/examples/dropout.py -------------------------------------------------------------------------------- /docs/tutorial/examples/firstdim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/docs/tutorial/examples/firstdim.py -------------------------------------------------------------------------------- /docs/tutorial/examples/forloop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/docs/tutorial/examples/forloop.py -------------------------------------------------------------------------------- /docs/tutorial/examples/forwhileloop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/docs/tutorial/examples/forwhileloop.py -------------------------------------------------------------------------------- /docs/tutorial/examples/hardmax_end_to_end.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/docs/tutorial/examples/hardmax_end_to_end.py -------------------------------------------------------------------------------- /docs/tutorial/examples/leaky_relu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/docs/tutorial/examples/leaky_relu.py -------------------------------------------------------------------------------- /docs/tutorial/examples/leaky_relu_attr_promoted.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/docs/tutorial/examples/leaky_relu_attr_promoted.py -------------------------------------------------------------------------------- /docs/tutorial/examples/omitted_input.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/docs/tutorial/examples/omitted_input.py -------------------------------------------------------------------------------- /docs/tutorial/examples/outerscope_redef_error.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/docs/tutorial/examples/outerscope_redef_error.py -------------------------------------------------------------------------------- /docs/tutorial/examples/scanloop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/docs/tutorial/examples/scanloop.py -------------------------------------------------------------------------------- /docs/tutorial/examples/softplus.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/docs/tutorial/examples/softplus.py -------------------------------------------------------------------------------- /docs/tutorial/examples/tensor_attr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/docs/tutorial/examples/tensor_attr.py -------------------------------------------------------------------------------- /docs/tutorial/examples/tensor_attr2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/docs/tutorial/examples/tensor_attr2.py -------------------------------------------------------------------------------- /docs/tutorial/examples/tensor_attr_short.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/docs/tutorial/examples/tensor_attr_short.py -------------------------------------------------------------------------------- /docs/tutorial/examples/whileloop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/docs/tutorial/examples/whileloop.py -------------------------------------------------------------------------------- /docs/tutorial/index.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/docs/tutorial/index.md -------------------------------------------------------------------------------- /docs/tutorial/optimizer/index.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/docs/tutorial/optimizer/index.md -------------------------------------------------------------------------------- /docs/tutorial/optimizer/optimize.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/docs/tutorial/optimizer/optimize.md -------------------------------------------------------------------------------- /docs/tutorial/rewriter/allow_other_inputs.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/docs/tutorial/rewriter/allow_other_inputs.md -------------------------------------------------------------------------------- /docs/tutorial/rewriter/attributes.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/docs/tutorial/rewriter/attributes.md -------------------------------------------------------------------------------- /docs/tutorial/rewriter/commute.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/docs/tutorial/rewriter/commute.md -------------------------------------------------------------------------------- /docs/tutorial/rewriter/conditional_rewrite.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/docs/tutorial/rewriter/conditional_rewrite.md -------------------------------------------------------------------------------- /docs/tutorial/rewriter/domain_option.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/docs/tutorial/rewriter/domain_option.md -------------------------------------------------------------------------------- /docs/tutorial/rewriter/examples/allow_other_attributes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/docs/tutorial/rewriter/examples/allow_other_attributes.py -------------------------------------------------------------------------------- /docs/tutorial/rewriter/examples/allow_other_inputs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/docs/tutorial/rewriter/examples/allow_other_inputs.py -------------------------------------------------------------------------------- /docs/tutorial/rewriter/examples/broadcast_matmul.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/docs/tutorial/rewriter/examples/broadcast_matmul.py -------------------------------------------------------------------------------- /docs/tutorial/rewriter/examples/domain_option.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/docs/tutorial/rewriter/examples/domain_option.py -------------------------------------------------------------------------------- /docs/tutorial/rewriter/examples/erfgelu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/docs/tutorial/rewriter/examples/erfgelu.py -------------------------------------------------------------------------------- /docs/tutorial/rewriter/examples/img/broadcast_01.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/docs/tutorial/rewriter/examples/img/broadcast_01.png -------------------------------------------------------------------------------- /docs/tutorial/rewriter/examples/img/broadcast_02.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/docs/tutorial/rewriter/examples/img/broadcast_02.png -------------------------------------------------------------------------------- /docs/tutorial/rewriter/examples/img/erfgelu_01.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/docs/tutorial/rewriter/examples/img/erfgelu_01.png -------------------------------------------------------------------------------- /docs/tutorial/rewriter/examples/img/erfgelu_02.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/docs/tutorial/rewriter/examples/img/erfgelu_02.png -------------------------------------------------------------------------------- /docs/tutorial/rewriter/examples/img/erfgelu_03_commute.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/docs/tutorial/rewriter/examples/img/erfgelu_03_commute.png -------------------------------------------------------------------------------- /docs/tutorial/rewriter/examples/img/erfgelu_04_commute.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/docs/tutorial/rewriter/examples/img/erfgelu_04_commute.png -------------------------------------------------------------------------------- /docs/tutorial/rewriter/examples/img/erfgelu_05_commute.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/docs/tutorial/rewriter/examples/img/erfgelu_05_commute.png -------------------------------------------------------------------------------- /docs/tutorial/rewriter/examples/img/erfgelu_06_commute.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/docs/tutorial/rewriter/examples/img/erfgelu_06_commute.png -------------------------------------------------------------------------------- /docs/tutorial/rewriter/examples/img/erfgelu_07_commute.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/docs/tutorial/rewriter/examples/img/erfgelu_07_commute.png -------------------------------------------------------------------------------- /docs/tutorial/rewriter/examples/or_pattern.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/docs/tutorial/rewriter/examples/or_pattern.py -------------------------------------------------------------------------------- /docs/tutorial/rewriter/examples/outputs_option.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/docs/tutorial/rewriter/examples/outputs_option.py -------------------------------------------------------------------------------- /docs/tutorial/rewriter/index.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/docs/tutorial/rewriter/index.md -------------------------------------------------------------------------------- /docs/tutorial/rewriter/node_value_checkers.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/docs/tutorial/rewriter/node_value_checkers.md -------------------------------------------------------------------------------- /docs/tutorial/rewriter/or_pattern.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/docs/tutorial/rewriter/or_pattern.md -------------------------------------------------------------------------------- /docs/tutorial/rewriter/outputs_option.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/docs/tutorial/rewriter/outputs_option.md -------------------------------------------------------------------------------- /docs/tutorial/rewriter/rewrite_patterns.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/docs/tutorial/rewriter/rewrite_patterns.md -------------------------------------------------------------------------------- /docs/tutorial/rewriter/simple_example.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/docs/tutorial/rewriter/simple_example.md -------------------------------------------------------------------------------- /docs/update_readme.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/docs/update_readme.py -------------------------------------------------------------------------------- /examples/custom_op_expansion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/examples/custom_op_expansion.py -------------------------------------------------------------------------------- /examples/pattern_matching_example.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/examples/pattern_matching_example.py -------------------------------------------------------------------------------- /examples/pattern_rewriting.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/examples/pattern_rewriting.py -------------------------------------------------------------------------------- /noxfile.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/noxfile.py -------------------------------------------------------------------------------- /onnxscript/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/__init__.py -------------------------------------------------------------------------------- /onnxscript/_framework_apis/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/_framework_apis/__init__.py -------------------------------------------------------------------------------- /onnxscript/_framework_apis/torch_2_5.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/_framework_apis/torch_2_5.py -------------------------------------------------------------------------------- /onnxscript/_framework_apis/torch_2_6.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/_framework_apis/torch_2_6.py -------------------------------------------------------------------------------- /onnxscript/_framework_apis/torch_2_7.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/_framework_apis/torch_2_7.py -------------------------------------------------------------------------------- /onnxscript/_framework_apis/torch_2_8.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/_framework_apis/torch_2_8.py -------------------------------------------------------------------------------- /onnxscript/_framework_apis/torch_2_9.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/_framework_apis/torch_2_9.py -------------------------------------------------------------------------------- /onnxscript/_internal/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /onnxscript/_internal/analysis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/_internal/analysis.py -------------------------------------------------------------------------------- /onnxscript/_internal/analysis_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/_internal/analysis_test.py -------------------------------------------------------------------------------- /onnxscript/_internal/ast_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/_internal/ast_utils.py -------------------------------------------------------------------------------- /onnxscript/_internal/autocast.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/_internal/autocast.py -------------------------------------------------------------------------------- /onnxscript/_internal/deprecation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/_internal/deprecation.py -------------------------------------------------------------------------------- /onnxscript/_internal/param_manipulation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/_internal/param_manipulation.py -------------------------------------------------------------------------------- /onnxscript/_internal/param_manipulation_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/_internal/param_manipulation_test.py -------------------------------------------------------------------------------- /onnxscript/_internal/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/_internal/utils.py -------------------------------------------------------------------------------- /onnxscript/_internal/version_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/_internal/version_utils.py -------------------------------------------------------------------------------- /onnxscript/_thirdparty/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/_thirdparty/README.md -------------------------------------------------------------------------------- /onnxscript/_thirdparty/asciichartpy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/_thirdparty/asciichartpy.py -------------------------------------------------------------------------------- /onnxscript/backend/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/backend/__init__.py -------------------------------------------------------------------------------- /onnxscript/backend/onnx_backend.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/backend/onnx_backend.py -------------------------------------------------------------------------------- /onnxscript/backend/onnx_backend_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/backend/onnx_backend_test.py -------------------------------------------------------------------------------- /onnxscript/backend/onnx_export.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/backend/onnx_export.py -------------------------------------------------------------------------------- /onnxscript/backend/onnx_export_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/backend/onnx_export_test.py -------------------------------------------------------------------------------- /onnxscript/converter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/converter.py -------------------------------------------------------------------------------- /onnxscript/converter_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/converter_test.py -------------------------------------------------------------------------------- /onnxscript/evaluator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/evaluator.py -------------------------------------------------------------------------------- /onnxscript/evaluator_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/evaluator_test.py -------------------------------------------------------------------------------- /onnxscript/function_libs/tools/torch_lib/deduce_type_constraints.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/function_libs/tools/torch_lib/deduce_type_constraints.py -------------------------------------------------------------------------------- /onnxscript/function_libs/tools/torch_lib/deduce_type_constraints_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/function_libs/tools/torch_lib/deduce_type_constraints_test.py -------------------------------------------------------------------------------- /onnxscript/function_libs/tools/torch_lib/generate_aten_signatures.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/function_libs/tools/torch_lib/generate_aten_signatures.py -------------------------------------------------------------------------------- /onnxscript/function_libs/tools/torch_lib/generate_prims_signatures.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/function_libs/tools/torch_lib/generate_prims_signatures.py -------------------------------------------------------------------------------- /onnxscript/function_libs/torch_lib/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/function_libs/torch_lib/__init__.py -------------------------------------------------------------------------------- /onnxscript/function_libs/torch_lib/_constants.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/function_libs/torch_lib/_constants.py -------------------------------------------------------------------------------- /onnxscript/function_libs/torch_lib/_flags.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/function_libs/torch_lib/_flags.py -------------------------------------------------------------------------------- /onnxscript/function_libs/torch_lib/graph_building/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/function_libs/torch_lib/graph_building/__init__.py -------------------------------------------------------------------------------- /onnxscript/function_libs/torch_lib/ops/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/function_libs/torch_lib/ops/__init__.py -------------------------------------------------------------------------------- /onnxscript/function_libs/torch_lib/ops/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/function_libs/torch_lib/ops/common.py -------------------------------------------------------------------------------- /onnxscript/function_libs/torch_lib/ops/core.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/function_libs/torch_lib/ops/core.py -------------------------------------------------------------------------------- /onnxscript/function_libs/torch_lib/ops/fft.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/function_libs/torch_lib/ops/fft.py -------------------------------------------------------------------------------- /onnxscript/function_libs/torch_lib/ops/linalg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/function_libs/torch_lib/ops/linalg.py -------------------------------------------------------------------------------- /onnxscript/function_libs/torch_lib/ops/nested.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/function_libs/torch_lib/ops/nested.py -------------------------------------------------------------------------------- /onnxscript/function_libs/torch_lib/ops/nn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/function_libs/torch_lib/ops/nn.py -------------------------------------------------------------------------------- /onnxscript/function_libs/torch_lib/ops/prims.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/function_libs/torch_lib/ops/prims.py -------------------------------------------------------------------------------- /onnxscript/function_libs/torch_lib/ops/quantized_decomposed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/function_libs/torch_lib/ops/quantized_decomposed.py -------------------------------------------------------------------------------- /onnxscript/function_libs/torch_lib/ops/sparse.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/function_libs/torch_lib/ops/sparse.py -------------------------------------------------------------------------------- /onnxscript/function_libs/torch_lib/ops/special.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/function_libs/torch_lib/ops/special.py -------------------------------------------------------------------------------- /onnxscript/function_libs/torch_lib/ops/vision.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/function_libs/torch_lib/ops/vision.py -------------------------------------------------------------------------------- /onnxscript/function_libs/torch_lib/registration.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/function_libs/torch_lib/registration.py -------------------------------------------------------------------------------- /onnxscript/function_libs/torch_lib/tensor_typing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/function_libs/torch_lib/tensor_typing.py -------------------------------------------------------------------------------- /onnxscript/ir/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/ir/README.md -------------------------------------------------------------------------------- /onnxscript/ir/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/ir/__init__.py -------------------------------------------------------------------------------- /onnxscript/ir/_schemas.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/ir/_schemas.py -------------------------------------------------------------------------------- /onnxscript/ir/_schemas_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/ir/_schemas_test.py -------------------------------------------------------------------------------- /onnxscript/ir/_tape.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/ir/_tape.py -------------------------------------------------------------------------------- /onnxscript/ir/_tape_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/ir/_tape_test.py -------------------------------------------------------------------------------- /onnxscript/ir/convenience.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/ir/convenience.py -------------------------------------------------------------------------------- /onnxscript/ir/passes/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/ir/passes/__init__.py -------------------------------------------------------------------------------- /onnxscript/ir/passes/common/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/ir/passes/common/__init__.py -------------------------------------------------------------------------------- /onnxscript/irbuilder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/irbuilder.py -------------------------------------------------------------------------------- /onnxscript/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/main.py -------------------------------------------------------------------------------- /onnxscript/onnx_opset/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/onnx_opset/__init__.py -------------------------------------------------------------------------------- /onnxscript/onnx_opset/_impl/opset1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/onnx_opset/_impl/opset1.py -------------------------------------------------------------------------------- /onnxscript/onnx_opset/_impl/opset10.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/onnx_opset/_impl/opset10.py -------------------------------------------------------------------------------- /onnxscript/onnx_opset/_impl/opset11.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/onnx_opset/_impl/opset11.py -------------------------------------------------------------------------------- /onnxscript/onnx_opset/_impl/opset12.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/onnx_opset/_impl/opset12.py -------------------------------------------------------------------------------- /onnxscript/onnx_opset/_impl/opset13.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/onnx_opset/_impl/opset13.py -------------------------------------------------------------------------------- /onnxscript/onnx_opset/_impl/opset14.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/onnx_opset/_impl/opset14.py -------------------------------------------------------------------------------- /onnxscript/onnx_opset/_impl/opset15.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/onnx_opset/_impl/opset15.py -------------------------------------------------------------------------------- /onnxscript/onnx_opset/_impl/opset16.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/onnx_opset/_impl/opset16.py -------------------------------------------------------------------------------- /onnxscript/onnx_opset/_impl/opset17.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/onnx_opset/_impl/opset17.py -------------------------------------------------------------------------------- /onnxscript/onnx_opset/_impl/opset18.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/onnx_opset/_impl/opset18.py -------------------------------------------------------------------------------- /onnxscript/onnx_opset/_impl/opset19.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/onnx_opset/_impl/opset19.py -------------------------------------------------------------------------------- /onnxscript/onnx_opset/_impl/opset2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/onnx_opset/_impl/opset2.py -------------------------------------------------------------------------------- /onnxscript/onnx_opset/_impl/opset20.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/onnx_opset/_impl/opset20.py -------------------------------------------------------------------------------- /onnxscript/onnx_opset/_impl/opset21.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/onnx_opset/_impl/opset21.py -------------------------------------------------------------------------------- /onnxscript/onnx_opset/_impl/opset22.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/onnx_opset/_impl/opset22.py -------------------------------------------------------------------------------- /onnxscript/onnx_opset/_impl/opset23.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/onnx_opset/_impl/opset23.py -------------------------------------------------------------------------------- /onnxscript/onnx_opset/_impl/opset24.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/onnx_opset/_impl/opset24.py -------------------------------------------------------------------------------- /onnxscript/onnx_opset/_impl/opset3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/onnx_opset/_impl/opset3.py -------------------------------------------------------------------------------- /onnxscript/onnx_opset/_impl/opset4.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/onnx_opset/_impl/opset4.py -------------------------------------------------------------------------------- /onnxscript/onnx_opset/_impl/opset5.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/onnx_opset/_impl/opset5.py -------------------------------------------------------------------------------- /onnxscript/onnx_opset/_impl/opset6.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/onnx_opset/_impl/opset6.py -------------------------------------------------------------------------------- /onnxscript/onnx_opset/_impl/opset7.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/onnx_opset/_impl/opset7.py -------------------------------------------------------------------------------- /onnxscript/onnx_opset/_impl/opset8.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/onnx_opset/_impl/opset8.py -------------------------------------------------------------------------------- /onnxscript/onnx_opset/_impl/opset9.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/onnx_opset/_impl/opset9.py -------------------------------------------------------------------------------- /onnxscript/onnx_opset/_impl/opset_ai_onnx_ml1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/onnx_opset/_impl/opset_ai_onnx_ml1.py -------------------------------------------------------------------------------- /onnxscript/onnx_opset/_impl/opset_ai_onnx_ml2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/onnx_opset/_impl/opset_ai_onnx_ml2.py -------------------------------------------------------------------------------- /onnxscript/onnx_opset/_impl/opset_ai_onnx_ml3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/onnx_opset/_impl/opset_ai_onnx_ml3.py -------------------------------------------------------------------------------- /onnxscript/onnx_opset/_impl/opset_ai_onnx_ml4.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/onnx_opset/_impl/opset_ai_onnx_ml4.py -------------------------------------------------------------------------------- /onnxscript/onnx_opset/_impl/opset_ai_onnx_ml5.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/onnx_opset/_impl/opset_ai_onnx_ml5.py -------------------------------------------------------------------------------- /onnxscript/onnx_types.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/onnx_types.py -------------------------------------------------------------------------------- /onnxscript/optimizer/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/optimizer/__init__.py -------------------------------------------------------------------------------- /onnxscript/optimizer/_constant_folding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/optimizer/_constant_folding.py -------------------------------------------------------------------------------- /onnxscript/optimizer/_constant_folding_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/optimizer/_constant_folding_test.py -------------------------------------------------------------------------------- /onnxscript/optimizer/_function_folding_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/optimizer/_function_folding_test.py -------------------------------------------------------------------------------- /onnxscript/optimizer/_optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/optimizer/_optimizer.py -------------------------------------------------------------------------------- /onnxscript/optimizer/_optimizer_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/optimizer/_optimizer_test.py -------------------------------------------------------------------------------- /onnxscript/py.typed: -------------------------------------------------------------------------------- 1 | # Marker file for PEP-561 (inline types) 2 | -------------------------------------------------------------------------------- /onnxscript/rewriter/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/rewriter/__init__.py -------------------------------------------------------------------------------- /onnxscript/rewriter/_basics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/rewriter/_basics.py -------------------------------------------------------------------------------- /onnxscript/rewriter/_fusion_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/rewriter/_fusion_utils.py -------------------------------------------------------------------------------- /onnxscript/rewriter/_ir_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/rewriter/_ir_utils.py -------------------------------------------------------------------------------- /onnxscript/rewriter/_matcher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/rewriter/_matcher.py -------------------------------------------------------------------------------- /onnxscript/rewriter/_pattern_ir.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/rewriter/_pattern_ir.py -------------------------------------------------------------------------------- /onnxscript/rewriter/_pattern_ir_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/rewriter/_pattern_ir_test.py -------------------------------------------------------------------------------- /onnxscript/rewriter/_rewrite_rule.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/rewriter/_rewrite_rule.py -------------------------------------------------------------------------------- /onnxscript/rewriter/match_context_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/rewriter/match_context_test.py -------------------------------------------------------------------------------- /onnxscript/rewriter/models/_bart_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/rewriter/models/_bart_encoder.py -------------------------------------------------------------------------------- /onnxscript/rewriter/models/_phi2lm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/rewriter/models/_phi2lm.py -------------------------------------------------------------------------------- /onnxscript/rewriter/models/_phi4lm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/rewriter/models/_phi4lm.py -------------------------------------------------------------------------------- /onnxscript/rewriter/models/_rotary_embedding_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/rewriter/models/_rotary_embedding_models.py -------------------------------------------------------------------------------- /onnxscript/rewriter/models/_smollm_1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/rewriter/models/_smollm_1.py -------------------------------------------------------------------------------- /onnxscript/rewriter/models/_smollm_2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/rewriter/models/_smollm_2.py -------------------------------------------------------------------------------- /onnxscript/rewriter/models/_test_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/rewriter/models/_test_models.py -------------------------------------------------------------------------------- /onnxscript/rewriter/models/_whisper_decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/rewriter/models/_whisper_decoder.py -------------------------------------------------------------------------------- /onnxscript/rewriter/models/_whisper_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/rewriter/models/_whisper_encoder.py -------------------------------------------------------------------------------- /onnxscript/rewriter/onnx_fusions/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/rewriter/onnx_fusions/__init__.py -------------------------------------------------------------------------------- /onnxscript/rewriter/onnx_fusions/_onnx_fusions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/rewriter/onnx_fusions/_onnx_fusions.py -------------------------------------------------------------------------------- /onnxscript/rewriter/onnxruntime/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/rewriter/onnxruntime/README.md -------------------------------------------------------------------------------- /onnxscript/rewriter/onnxruntime/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/rewriter/onnxruntime/__init__.py -------------------------------------------------------------------------------- /onnxscript/rewriter/onnxruntime/bfloat16_utils/bfloat16_converter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/rewriter/onnxruntime/bfloat16_utils/bfloat16_converter.py -------------------------------------------------------------------------------- /onnxscript/rewriter/onnxruntime/bfloat16_utils/bfloat16_converter_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/rewriter/onnxruntime/bfloat16_utils/bfloat16_converter_test.py -------------------------------------------------------------------------------- /onnxscript/rewriter/ort_fusions/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/rewriter/ort_fusions/__init__.py -------------------------------------------------------------------------------- /onnxscript/rewriter/ort_fusions/_core.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/rewriter/ort_fusions/_core.py -------------------------------------------------------------------------------- /onnxscript/rewriter/ort_fusions/_test_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/rewriter/ort_fusions/_test_utils.py -------------------------------------------------------------------------------- /onnxscript/rewriter/ort_fusions/attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/rewriter/ort_fusions/attention.py -------------------------------------------------------------------------------- /onnxscript/rewriter/ort_fusions/attention_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/rewriter/ort_fusions/attention_test.py -------------------------------------------------------------------------------- /onnxscript/rewriter/ort_fusions/bias_gelu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/rewriter/ort_fusions/bias_gelu.py -------------------------------------------------------------------------------- /onnxscript/rewriter/ort_fusions/bias_gelu_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/rewriter/ort_fusions/bias_gelu_test.py -------------------------------------------------------------------------------- /onnxscript/rewriter/ort_fusions/cos_sin_cache.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/rewriter/ort_fusions/cos_sin_cache.py -------------------------------------------------------------------------------- /onnxscript/rewriter/ort_fusions/cos_sin_cache_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/rewriter/ort_fusions/cos_sin_cache_test.py -------------------------------------------------------------------------------- /onnxscript/rewriter/ort_fusions/erfgelu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/rewriter/ort_fusions/erfgelu.py -------------------------------------------------------------------------------- /onnxscript/rewriter/ort_fusions/fuse_xformers_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/rewriter/ort_fusions/fuse_xformers_test.py -------------------------------------------------------------------------------- /onnxscript/rewriter/ort_fusions/fused_matmul_rule_sets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/rewriter/ort_fusions/fused_matmul_rule_sets.py -------------------------------------------------------------------------------- /onnxscript/rewriter/ort_fusions/fused_matmul_rule_sets_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/rewriter/ort_fusions/fused_matmul_rule_sets_test.py -------------------------------------------------------------------------------- /onnxscript/rewriter/ort_fusions/gelu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/rewriter/ort_fusions/gelu.py -------------------------------------------------------------------------------- /onnxscript/rewriter/ort_fusions/gelu_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/rewriter/ort_fusions/gelu_test.py -------------------------------------------------------------------------------- /onnxscript/rewriter/ort_fusions/gqa.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/rewriter/ort_fusions/gqa.py -------------------------------------------------------------------------------- /onnxscript/rewriter/ort_fusions/gqa_packed_qkv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/rewriter/ort_fusions/gqa_packed_qkv.py -------------------------------------------------------------------------------- /onnxscript/rewriter/ort_fusions/gqa_packed_qkv_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/rewriter/ort_fusions/gqa_packed_qkv_test.py -------------------------------------------------------------------------------- /onnxscript/rewriter/ort_fusions/gqa_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/rewriter/ort_fusions/gqa_test.py -------------------------------------------------------------------------------- /onnxscript/rewriter/ort_fusions/group_normalization_merge_silu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/rewriter/ort_fusions/group_normalization_merge_silu.py -------------------------------------------------------------------------------- /onnxscript/rewriter/ort_fusions/group_normalization_merge_silu_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/rewriter/ort_fusions/group_normalization_merge_silu_test.py -------------------------------------------------------------------------------- /onnxscript/rewriter/ort_fusions/instance_to_group_normalization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/rewriter/ort_fusions/instance_to_group_normalization.py -------------------------------------------------------------------------------- /onnxscript/rewriter/ort_fusions/instance_to_group_normalization_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/rewriter/ort_fusions/instance_to_group_normalization_test.py -------------------------------------------------------------------------------- /onnxscript/rewriter/ort_fusions/mha.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/rewriter/ort_fusions/mha.py -------------------------------------------------------------------------------- /onnxscript/rewriter/ort_fusions/mha_bias.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/rewriter/ort_fusions/mha_bias.py -------------------------------------------------------------------------------- /onnxscript/rewriter/ort_fusions/mha_scale.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/rewriter/ort_fusions/mha_scale.py -------------------------------------------------------------------------------- /onnxscript/rewriter/ort_fusions/mha_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/rewriter/ort_fusions/mha_test.py -------------------------------------------------------------------------------- /onnxscript/rewriter/ort_fusions/rms_normalization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/rewriter/ort_fusions/rms_normalization.py -------------------------------------------------------------------------------- /onnxscript/rewriter/ort_fusions/rms_normalization_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/rewriter/ort_fusions/rms_normalization_test.py -------------------------------------------------------------------------------- /onnxscript/rewriter/ort_fusions/rotary_embedding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/rewriter/ort_fusions/rotary_embedding.py -------------------------------------------------------------------------------- /onnxscript/rewriter/ort_fusions/rotary_embedding_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/rewriter/ort_fusions/rotary_embedding_test.py -------------------------------------------------------------------------------- /onnxscript/rewriter/ort_fusions/sdpa.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/rewriter/ort_fusions/sdpa.py -------------------------------------------------------------------------------- /onnxscript/rewriter/ort_fusions/sdpa_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/rewriter/ort_fusions/sdpa_test.py -------------------------------------------------------------------------------- /onnxscript/rewriter/ort_fusions/sdpa_via_mha.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/rewriter/ort_fusions/sdpa_via_mha.py -------------------------------------------------------------------------------- /onnxscript/rewriter/ort_fusions/shape_optimization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/rewriter/ort_fusions/shape_optimization.py -------------------------------------------------------------------------------- /onnxscript/rewriter/ort_fusions/shape_optimization_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/rewriter/ort_fusions/shape_optimization_test.py -------------------------------------------------------------------------------- /onnxscript/rewriter/ort_fusions/skip_normalization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/rewriter/ort_fusions/skip_normalization.py -------------------------------------------------------------------------------- /onnxscript/rewriter/ort_fusions/skip_normalization_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/rewriter/ort_fusions/skip_normalization_test.py -------------------------------------------------------------------------------- /onnxscript/rewriter/ort_fusions/softmax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/rewriter/ort_fusions/softmax.py -------------------------------------------------------------------------------- /onnxscript/rewriter/ort_fusions/softmax_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/rewriter/ort_fusions/softmax_test.py -------------------------------------------------------------------------------- /onnxscript/rewriter/pattern.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/rewriter/pattern.py -------------------------------------------------------------------------------- /onnxscript/rewriter/pattern_base_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/rewriter/pattern_base_test.py -------------------------------------------------------------------------------- /onnxscript/rewriter/pattern_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/rewriter/pattern_test.py -------------------------------------------------------------------------------- /onnxscript/rewriter/rules/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/rewriter/rules/__init__.py -------------------------------------------------------------------------------- /onnxscript/rewriter/rules/common/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/rewriter/rules/common/__init__.py -------------------------------------------------------------------------------- /onnxscript/rewriter/rules/common/_basic_rules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/rewriter/rules/common/_basic_rules.py -------------------------------------------------------------------------------- /onnxscript/rewriter/rules/common/_basic_rules_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/rewriter/rules/common/_basic_rules_test.py -------------------------------------------------------------------------------- /onnxscript/rewriter/rules/common/_broadcast_to_matmul.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/rewriter/rules/common/_broadcast_to_matmul.py -------------------------------------------------------------------------------- /onnxscript/rewriter/rules/common/_broadcast_to_matmul_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/rewriter/rules/common/_broadcast_to_matmul_test.py -------------------------------------------------------------------------------- /onnxscript/rewriter/rules/common/_cast_constant_of_shape.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/rewriter/rules/common/_cast_constant_of_shape.py -------------------------------------------------------------------------------- /onnxscript/rewriter/rules/common/_cast_constant_of_shape_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/rewriter/rules/common/_cast_constant_of_shape_test.py -------------------------------------------------------------------------------- /onnxscript/rewriter/rules/common/_collapse_slices.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/rewriter/rules/common/_collapse_slices.py -------------------------------------------------------------------------------- /onnxscript/rewriter/rules/common/_collapse_slices_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/rewriter/rules/common/_collapse_slices_test.py -------------------------------------------------------------------------------- /onnxscript/rewriter/rules/common/_fuse_batchnorm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/rewriter/rules/common/_fuse_batchnorm.py -------------------------------------------------------------------------------- /onnxscript/rewriter/rules/common/_fuse_batchnorm_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/rewriter/rules/common/_fuse_batchnorm_test.py -------------------------------------------------------------------------------- /onnxscript/rewriter/rules/common/_fuse_conv_affine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/rewriter/rules/common/_fuse_conv_affine.py -------------------------------------------------------------------------------- /onnxscript/rewriter/rules/common/_fuse_conv_affine_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/rewriter/rules/common/_fuse_conv_affine_test.py -------------------------------------------------------------------------------- /onnxscript/rewriter/rules/common/_fuse_hardswish.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/rewriter/rules/common/_fuse_hardswish.py -------------------------------------------------------------------------------- /onnxscript/rewriter/rules/common/_fuse_hardswish_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/rewriter/rules/common/_fuse_hardswish_test.py -------------------------------------------------------------------------------- /onnxscript/rewriter/rules/common/_fuse_pad_into_conv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/rewriter/rules/common/_fuse_pad_into_conv.py -------------------------------------------------------------------------------- /onnxscript/rewriter/rules/common/_fuse_pad_into_conv_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/rewriter/rules/common/_fuse_pad_into_conv_test.py -------------------------------------------------------------------------------- /onnxscript/rewriter/rules/common/_fuse_relus_clips.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/rewriter/rules/common/_fuse_relus_clips.py -------------------------------------------------------------------------------- /onnxscript/rewriter/rules/common/_fuse_relus_clips_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/rewriter/rules/common/_fuse_relus_clips_test.py -------------------------------------------------------------------------------- /onnxscript/rewriter/rules/common/_gemm_to_matmul_add.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/rewriter/rules/common/_gemm_to_matmul_add.py -------------------------------------------------------------------------------- /onnxscript/rewriter/rules/common/_gemm_to_matmul_add_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/rewriter/rules/common/_gemm_to_matmul_add_test.py -------------------------------------------------------------------------------- /onnxscript/rewriter/rules/common/_matmul_add_to_gemm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/rewriter/rules/common/_matmul_add_to_gemm.py -------------------------------------------------------------------------------- /onnxscript/rewriter/rules/common/_matmul_add_to_gemm_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/rewriter/rules/common/_matmul_add_to_gemm_test.py -------------------------------------------------------------------------------- /onnxscript/rewriter/rules/common/_min_max_to_clip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/rewriter/rules/common/_min_max_to_clip.py -------------------------------------------------------------------------------- /onnxscript/rewriter/rules/common/_min_max_to_clip_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/rewriter/rules/common/_min_max_to_clip_test.py -------------------------------------------------------------------------------- /onnxscript/rewriter/rules/common/_no_op.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/rewriter/rules/common/_no_op.py -------------------------------------------------------------------------------- /onnxscript/rewriter/rules/common/_no_op_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/rewriter/rules/common/_no_op_test.py -------------------------------------------------------------------------------- /onnxscript/rewriter/rules/common/_redundant_scatter_nd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/rewriter/rules/common/_redundant_scatter_nd.py -------------------------------------------------------------------------------- /onnxscript/rewriter/rules/common/_redundant_scatter_nd_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/rewriter/rules/common/_redundant_scatter_nd_test.py -------------------------------------------------------------------------------- /onnxscript/rewriter/rules/common/_remove_optional_bias.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/rewriter/rules/common/_remove_optional_bias.py -------------------------------------------------------------------------------- /onnxscript/rewriter/rules/common/_remove_optional_bias_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/rewriter/rules/common/_remove_optional_bias_test.py -------------------------------------------------------------------------------- /onnxscript/rewriter/rules/fusion/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/rewriter/rules/fusion/__init__.py -------------------------------------------------------------------------------- /onnxscript/rewriter/rules/fusion/_gqa.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/rewriter/rules/fusion/_gqa.py -------------------------------------------------------------------------------- /onnxscript/rewriter/rules/fusion/_gqa_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/rewriter/rules/fusion/_gqa_test.py -------------------------------------------------------------------------------- /onnxscript/rewriter/rules/fusion/_layer_norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/rewriter/rules/fusion/_layer_norm.py -------------------------------------------------------------------------------- /onnxscript/rewriter/rules/fusion/_layer_norm_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/rewriter/rules/fusion/_layer_norm_test.py -------------------------------------------------------------------------------- /onnxscript/rewriter/rules/fusion/_rms_normalization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/rewriter/rules/fusion/_rms_normalization.py -------------------------------------------------------------------------------- /onnxscript/rewriter/rules/fusion/_rms_normalization_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/rewriter/rules/fusion/_rms_normalization_test.py -------------------------------------------------------------------------------- /onnxscript/rewriter/rules/fusion/_rotary_embedding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/rewriter/rules/fusion/_rotary_embedding.py -------------------------------------------------------------------------------- /onnxscript/rewriter/rules/fusion/_rotary_embedding_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/rewriter/rules/fusion/_rotary_embedding_test.py -------------------------------------------------------------------------------- /onnxscript/rewriter/testing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/rewriter/testing.py -------------------------------------------------------------------------------- /onnxscript/sourceinfo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/sourceinfo.py -------------------------------------------------------------------------------- /onnxscript/tensor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/tensor.py -------------------------------------------------------------------------------- /onnxscript/tensor_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/tensor_test.py -------------------------------------------------------------------------------- /onnxscript/testing/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/testing/__init__.py -------------------------------------------------------------------------------- /onnxscript/tools/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/tools/__init__.py -------------------------------------------------------------------------------- /onnxscript/tools/memory_peak.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/tools/memory_peak.py -------------------------------------------------------------------------------- /onnxscript/tools/memory_peak_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/tools/memory_peak_test.py -------------------------------------------------------------------------------- /onnxscript/tools/transformers_models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/tools/transformers_models/__init__.py -------------------------------------------------------------------------------- /onnxscript/tools/transformers_models/llama.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/tools/transformers_models/llama.py -------------------------------------------------------------------------------- /onnxscript/tools/transformers_models/llama_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/tools/transformers_models/llama_test.py -------------------------------------------------------------------------------- /onnxscript/tools/transformers_models/mistral.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/tools/transformers_models/mistral.py -------------------------------------------------------------------------------- /onnxscript/tools/transformers_models/mistral_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/tools/transformers_models/mistral_test.py -------------------------------------------------------------------------------- /onnxscript/tools/transformers_models/phi.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/tools/transformers_models/phi.py -------------------------------------------------------------------------------- /onnxscript/tools/transformers_models/phi3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/tools/transformers_models/phi3.py -------------------------------------------------------------------------------- /onnxscript/tools/transformers_models/phi3_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/tools/transformers_models/phi3_test.py -------------------------------------------------------------------------------- /onnxscript/tools/transformers_models/phi_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/tools/transformers_models/phi_test.py -------------------------------------------------------------------------------- /onnxscript/type_annotation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/type_annotation.py -------------------------------------------------------------------------------- /onnxscript/type_annotation_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/type_annotation_test.py -------------------------------------------------------------------------------- /onnxscript/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /onnxscript/utils/evaluation_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/utils/evaluation_utils.py -------------------------------------------------------------------------------- /onnxscript/utils/metadata_merger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/utils/metadata_merger.py -------------------------------------------------------------------------------- /onnxscript/utils/replace.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/utils/replace.py -------------------------------------------------------------------------------- /onnxscript/utils/timing_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/utils/timing_utils.py -------------------------------------------------------------------------------- /onnxscript/utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/utils/utils.py -------------------------------------------------------------------------------- /onnxscript/values.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/values.py -------------------------------------------------------------------------------- /onnxscript/values_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/values_test.py -------------------------------------------------------------------------------- /onnxscript/version_converter/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/version_converter/__init__.py -------------------------------------------------------------------------------- /onnxscript/version_converter/_c_api_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/version_converter/_c_api_utils.py -------------------------------------------------------------------------------- /onnxscript/version_converter/_version_converter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/version_converter/_version_converter.py -------------------------------------------------------------------------------- /onnxscript/version_converter/_version_converter_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/onnxscript/version_converter/_version_converter_test.py -------------------------------------------------------------------------------- /opgen/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/opgen/README.md -------------------------------------------------------------------------------- /opgen/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /opgen/__main__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/opgen/__main__.py -------------------------------------------------------------------------------- /opgen/onnx_opset_builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/opgen/onnx_opset_builder.py -------------------------------------------------------------------------------- /opgen/pygen.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/opgen/pygen.py -------------------------------------------------------------------------------- /opgen/pygen_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/opgen/pygen_test.py -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/pyproject.toml -------------------------------------------------------------------------------- /pyproject_pylint.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/pyproject_pylint.toml -------------------------------------------------------------------------------- /requirements-dev.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/requirements-dev.txt -------------------------------------------------------------------------------- /requirements/ci/requirements-onnx-weekly.txt: -------------------------------------------------------------------------------- 1 | onnx-weekly==1.21.0.dev20251103 2 | -------------------------------------------------------------------------------- /requirements/ci/requirements-ort-nightly.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/requirements/ci/requirements-ort-nightly.txt -------------------------------------------------------------------------------- /requirements/ci/requirements-pytorch-nightly.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/requirements/ci/requirements-pytorch-nightly.txt -------------------------------------------------------------------------------- /requirements/lintrunner/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/requirements/lintrunner/requirements.txt -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/setup.py -------------------------------------------------------------------------------- /testdata/dort_models/llama_forward.onnx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/dort_models/llama_forward.onnx -------------------------------------------------------------------------------- /testdata/e2e_models/Speech2Text2ForCausalLM/dynamo/Speech2Text2ForCausalLM_dynamo.onnx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/e2e_models/Speech2Text2ForCausalLM/dynamo/Speech2Text2ForCausalLM_dynamo.onnx -------------------------------------------------------------------------------- /testdata/e2e_models/Speech2Text2ForCausalLM/dynamo/test_data_set_0/input_0.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/e2e_models/Speech2Text2ForCausalLM/dynamo/test_data_set_0/input_0.pb -------------------------------------------------------------------------------- /testdata/e2e_models/Speech2Text2ForCausalLM/dynamo/test_data_set_0/input_1.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/e2e_models/Speech2Text2ForCausalLM/dynamo/test_data_set_0/input_1.pb -------------------------------------------------------------------------------- /testdata/e2e_models/Speech2Text2ForCausalLM/dynamo/test_data_set_0/output_0.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/e2e_models/Speech2Text2ForCausalLM/dynamo/test_data_set_0/output_0.pb -------------------------------------------------------------------------------- /testdata/e2e_models/Speech2Text2ForCausalLM/dynamo/test_data_set_0/output_1.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/e2e_models/Speech2Text2ForCausalLM/dynamo/test_data_set_0/output_1.pb -------------------------------------------------------------------------------- /testdata/e2e_models/Speech2Text2ForCausalLM/dynamo/test_data_set_0/output_10.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/e2e_models/Speech2Text2ForCausalLM/dynamo/test_data_set_0/output_10.pb -------------------------------------------------------------------------------- /testdata/e2e_models/Speech2Text2ForCausalLM/dynamo/test_data_set_0/output_11.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/e2e_models/Speech2Text2ForCausalLM/dynamo/test_data_set_0/output_11.pb -------------------------------------------------------------------------------- /testdata/e2e_models/Speech2Text2ForCausalLM/dynamo/test_data_set_0/output_12.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/e2e_models/Speech2Text2ForCausalLM/dynamo/test_data_set_0/output_12.pb -------------------------------------------------------------------------------- /testdata/e2e_models/Speech2Text2ForCausalLM/dynamo/test_data_set_0/output_13.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/e2e_models/Speech2Text2ForCausalLM/dynamo/test_data_set_0/output_13.pb -------------------------------------------------------------------------------- /testdata/e2e_models/Speech2Text2ForCausalLM/dynamo/test_data_set_0/output_2.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/e2e_models/Speech2Text2ForCausalLM/dynamo/test_data_set_0/output_2.pb -------------------------------------------------------------------------------- /testdata/e2e_models/Speech2Text2ForCausalLM/dynamo/test_data_set_0/output_3.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/e2e_models/Speech2Text2ForCausalLM/dynamo/test_data_set_0/output_3.pb -------------------------------------------------------------------------------- /testdata/e2e_models/Speech2Text2ForCausalLM/dynamo/test_data_set_0/output_4.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/e2e_models/Speech2Text2ForCausalLM/dynamo/test_data_set_0/output_4.pb -------------------------------------------------------------------------------- /testdata/e2e_models/Speech2Text2ForCausalLM/dynamo/test_data_set_0/output_5.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/e2e_models/Speech2Text2ForCausalLM/dynamo/test_data_set_0/output_5.pb -------------------------------------------------------------------------------- /testdata/e2e_models/Speech2Text2ForCausalLM/dynamo/test_data_set_0/output_6.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/e2e_models/Speech2Text2ForCausalLM/dynamo/test_data_set_0/output_6.pb -------------------------------------------------------------------------------- /testdata/e2e_models/Speech2Text2ForCausalLM/dynamo/test_data_set_0/output_7.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/e2e_models/Speech2Text2ForCausalLM/dynamo/test_data_set_0/output_7.pb -------------------------------------------------------------------------------- /testdata/e2e_models/Speech2Text2ForCausalLM/dynamo/test_data_set_0/output_8.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/e2e_models/Speech2Text2ForCausalLM/dynamo/test_data_set_0/output_8.pb -------------------------------------------------------------------------------- /testdata/e2e_models/Speech2Text2ForCausalLM/dynamo/test_data_set_0/output_9.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/e2e_models/Speech2Text2ForCausalLM/dynamo/test_data_set_0/output_9.pb -------------------------------------------------------------------------------- /testdata/e2e_models/mobilenetv2_100/dynamo/mobilenetv2_100_dynamo.onnx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/e2e_models/mobilenetv2_100/dynamo/mobilenetv2_100_dynamo.onnx -------------------------------------------------------------------------------- /testdata/e2e_models/mobilenetv2_100/dynamo/test_data_set_0/input_0.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/e2e_models/mobilenetv2_100/dynamo/test_data_set_0/input_0.pb -------------------------------------------------------------------------------- /testdata/e2e_models/mobilenetv2_100/dynamo/test_data_set_0/output_0.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/e2e_models/mobilenetv2_100/dynamo/test_data_set_0/output_0.pb -------------------------------------------------------------------------------- /testdata/e2e_models/resnet18/dynamo/resnet18_dynamo.onnx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/e2e_models/resnet18/dynamo/resnet18_dynamo.onnx -------------------------------------------------------------------------------- /testdata/e2e_models/resnet18/dynamo/test_data_set_0/input_0.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/e2e_models/resnet18/dynamo/test_data_set_0/input_0.pb -------------------------------------------------------------------------------- /testdata/e2e_models/resnet18/dynamo/test_data_set_0/output_0.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/e2e_models/resnet18/dynamo/test_data_set_0/output_0.pb -------------------------------------------------------------------------------- /testdata/e2e_models/torchscript_model/torchscript_model.onnx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/e2e_models/torchscript_model/torchscript_model.onnx -------------------------------------------------------------------------------- /testdata/unittest_models/attn_llama2_4_34_0/attn_llama2_4_34_0.onnx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_llama2_4_34_0/attn_llama2_4_34_0.onnx -------------------------------------------------------------------------------- /testdata/unittest_models/attn_llama2_4_34_0/test_data_set_0/input_0.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_llama2_4_34_0/test_data_set_0/input_0.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_llama2_4_34_0/test_data_set_0/input_1.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_llama2_4_34_0/test_data_set_0/input_1.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_llama2_4_34_0/test_data_set_0/input_2.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_llama2_4_34_0/test_data_set_0/input_2.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_llama2_4_34_0/test_data_set_0/input_3.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_llama2_4_34_0/test_data_set_0/input_3.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_llama2_4_34_0/test_data_set_0/input_4.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_llama2_4_34_0/test_data_set_0/input_4.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_llama2_4_34_0/test_data_set_0/input_5.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_llama2_4_34_0/test_data_set_0/input_5.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_llama2_4_34_0/test_data_set_0/input_6.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_llama2_4_34_0/test_data_set_0/input_6.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_llama2_4_34_0/test_data_set_0/input_7.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_llama2_4_34_0/test_data_set_0/input_7.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_llama2_4_34_0/test_data_set_0/input_8.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_llama2_4_34_0/test_data_set_0/input_8.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_llama2_4_34_0/test_data_set_0/output_0.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_llama2_4_34_0/test_data_set_0/output_0.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_llama2_4_34_0/test_data_set_0/output_1.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_llama2_4_34_0/test_data_set_0/output_1.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_llama2_4_34_0/test_data_set_0/output_2.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_llama2_4_34_0/test_data_set_0/output_2.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_llama2_4_34_1/attn_llama2_4_34_1.onnx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_llama2_4_34_1/attn_llama2_4_34_1.onnx -------------------------------------------------------------------------------- /testdata/unittest_models/attn_llama2_4_34_1/test_data_set_0/input_0.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_llama2_4_34_1/test_data_set_0/input_0.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_llama2_4_34_1/test_data_set_0/input_1.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_llama2_4_34_1/test_data_set_0/input_1.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_llama2_4_34_1/test_data_set_0/input_2.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_llama2_4_34_1/test_data_set_0/input_2.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_llama2_4_34_1/test_data_set_0/input_3.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_llama2_4_34_1/test_data_set_0/input_3.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_llama2_4_34_1/test_data_set_0/input_4.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_llama2_4_34_1/test_data_set_0/input_4.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_llama2_4_34_1/test_data_set_0/input_5.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_llama2_4_34_1/test_data_set_0/input_5.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_llama2_4_34_1/test_data_set_0/input_6.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_llama2_4_34_1/test_data_set_0/input_6.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_llama2_4_34_1/test_data_set_0/input_7.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_llama2_4_34_1/test_data_set_0/input_7.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_llama2_4_34_1/test_data_set_0/input_8.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_llama2_4_34_1/test_data_set_0/input_8.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_llama2_4_34_1/test_data_set_0/output_0.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_llama2_4_34_1/test_data_set_0/output_0.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_llama2_4_34_1/test_data_set_0/output_1.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_llama2_4_34_1/test_data_set_0/output_1.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_llama2_4_34_1/test_data_set_0/output_2.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_llama2_4_34_1/test_data_set_0/output_2.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_llama2_4_36_0/attn_llama2_4_36_0.onnx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_llama2_4_36_0/attn_llama2_4_36_0.onnx -------------------------------------------------------------------------------- /testdata/unittest_models/attn_llama2_4_36_0/test_data_set_0/input_0.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_llama2_4_36_0/test_data_set_0/input_0.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_llama2_4_36_0/test_data_set_0/input_1.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_llama2_4_36_0/test_data_set_0/input_1.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_llama2_4_36_0/test_data_set_0/input_2.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_llama2_4_36_0/test_data_set_0/input_2.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_llama2_4_36_0/test_data_set_0/input_3.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_llama2_4_36_0/test_data_set_0/input_3.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_llama2_4_36_0/test_data_set_0/input_4.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_llama2_4_36_0/test_data_set_0/input_4.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_llama2_4_36_0/test_data_set_0/input_5.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_llama2_4_36_0/test_data_set_0/input_5.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_llama2_4_36_0/test_data_set_0/input_6.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_llama2_4_36_0/test_data_set_0/input_6.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_llama2_4_36_0/test_data_set_0/input_7.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_llama2_4_36_0/test_data_set_0/input_7.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_llama2_4_36_0/test_data_set_0/input_8.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_llama2_4_36_0/test_data_set_0/input_8.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_llama2_4_36_0/test_data_set_0/output_0.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_llama2_4_36_0/test_data_set_0/output_0.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_llama2_4_36_0/test_data_set_0/output_1.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_llama2_4_36_0/test_data_set_0/output_1.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_llama2_4_36_0/test_data_set_0/output_2.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_llama2_4_36_0/test_data_set_0/output_2.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_phi_1_5_0/attn_phi_1_5_0.onnx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_phi_1_5_0/attn_phi_1_5_0.onnx -------------------------------------------------------------------------------- /testdata/unittest_models/attn_phi_1_5_0/test_data_set_0/input_0.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_phi_1_5_0/test_data_set_0/input_0.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_phi_1_5_0/test_data_set_0/input_1.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_phi_1_5_0/test_data_set_0/input_1.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_phi_1_5_0/test_data_set_0/input_10.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_phi_1_5_0/test_data_set_0/input_10.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_phi_1_5_0/test_data_set_0/input_11.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_phi_1_5_0/test_data_set_0/input_11.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_phi_1_5_0/test_data_set_0/input_12.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_phi_1_5_0/test_data_set_0/input_12.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_phi_1_5_0/test_data_set_0/input_2.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_phi_1_5_0/test_data_set_0/input_2.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_phi_1_5_0/test_data_set_0/input_3.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_phi_1_5_0/test_data_set_0/input_3.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_phi_1_5_0/test_data_set_0/input_4.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_phi_1_5_0/test_data_set_0/input_4.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_phi_1_5_0/test_data_set_0/input_5.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_phi_1_5_0/test_data_set_0/input_5.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_phi_1_5_0/test_data_set_0/input_6.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_phi_1_5_0/test_data_set_0/input_6.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_phi_1_5_0/test_data_set_0/input_7.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_phi_1_5_0/test_data_set_0/input_7.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_phi_1_5_0/test_data_set_0/input_8.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_phi_1_5_0/test_data_set_0/input_8.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_phi_1_5_0/test_data_set_0/input_9.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_phi_1_5_0/test_data_set_0/input_9.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_phi_1_5_0/test_data_set_0/output_0.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_phi_1_5_0/test_data_set_0/output_0.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_phi_1_5_0/test_data_set_0/output_1.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_phi_1_5_0/test_data_set_0/output_1.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_phi_1_5_0/test_data_set_0/output_2.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_phi_1_5_0/test_data_set_0/output_2.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_phi_1_5_1/attn_phi_1_5_1.onnx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_phi_1_5_1/attn_phi_1_5_1.onnx -------------------------------------------------------------------------------- /testdata/unittest_models/attn_phi_1_5_1/test_data_set_0/input_0.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_phi_1_5_1/test_data_set_0/input_0.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_phi_1_5_1/test_data_set_0/input_1.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_phi_1_5_1/test_data_set_0/input_1.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_phi_1_5_1/test_data_set_0/input_10.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_phi_1_5_1/test_data_set_0/input_10.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_phi_1_5_1/test_data_set_0/input_11.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_phi_1_5_1/test_data_set_0/input_11.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_phi_1_5_1/test_data_set_0/input_12.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_phi_1_5_1/test_data_set_0/input_12.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_phi_1_5_1/test_data_set_0/input_2.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_phi_1_5_1/test_data_set_0/input_2.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_phi_1_5_1/test_data_set_0/input_3.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_phi_1_5_1/test_data_set_0/input_3.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_phi_1_5_1/test_data_set_0/input_4.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_phi_1_5_1/test_data_set_0/input_4.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_phi_1_5_1/test_data_set_0/input_5.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_phi_1_5_1/test_data_set_0/input_5.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_phi_1_5_1/test_data_set_0/input_6.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_phi_1_5_1/test_data_set_0/input_6.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_phi_1_5_1/test_data_set_0/input_7.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_phi_1_5_1/test_data_set_0/input_7.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_phi_1_5_1/test_data_set_0/input_8.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_phi_1_5_1/test_data_set_0/input_8.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_phi_1_5_1/test_data_set_0/input_9.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_phi_1_5_1/test_data_set_0/input_9.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_phi_1_5_1/test_data_set_0/output_0.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_phi_1_5_1/test_data_set_0/output_0.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_phi_1_5_1/test_data_set_0/output_1.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_phi_1_5_1/test_data_set_0/output_1.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_phi_1_5_1/test_data_set_0/output_2.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_phi_1_5_1/test_data_set_0/output_2.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_phi_1_5_2/attn_phi_1_5_2.onnx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_phi_1_5_2/attn_phi_1_5_2.onnx -------------------------------------------------------------------------------- /testdata/unittest_models/attn_phi_1_5_2/test_data_set_0/input_0.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_phi_1_5_2/test_data_set_0/input_0.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_phi_1_5_2/test_data_set_0/input_1.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_phi_1_5_2/test_data_set_0/input_1.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_phi_1_5_2/test_data_set_0/input_10.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_phi_1_5_2/test_data_set_0/input_10.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_phi_1_5_2/test_data_set_0/input_11.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_phi_1_5_2/test_data_set_0/input_11.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_phi_1_5_2/test_data_set_0/input_12.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_phi_1_5_2/test_data_set_0/input_12.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_phi_1_5_2/test_data_set_0/input_2.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_phi_1_5_2/test_data_set_0/input_2.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_phi_1_5_2/test_data_set_0/input_3.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_phi_1_5_2/test_data_set_0/input_3.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_phi_1_5_2/test_data_set_0/input_4.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_phi_1_5_2/test_data_set_0/input_4.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_phi_1_5_2/test_data_set_0/input_5.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_phi_1_5_2/test_data_set_0/input_5.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_phi_1_5_2/test_data_set_0/input_6.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_phi_1_5_2/test_data_set_0/input_6.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_phi_1_5_2/test_data_set_0/input_7.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_phi_1_5_2/test_data_set_0/input_7.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_phi_1_5_2/test_data_set_0/input_8.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_phi_1_5_2/test_data_set_0/input_8.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_phi_1_5_2/test_data_set_0/input_9.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_phi_1_5_2/test_data_set_0/input_9.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_phi_1_5_2/test_data_set_0/output_0.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_phi_1_5_2/test_data_set_0/output_0.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_phi_1_5_2/test_data_set_0/output_1.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_phi_1_5_2/test_data_set_0/output_1.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_phi_1_5_2/test_data_set_0/output_2.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_phi_1_5_2/test_data_set_0/output_2.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_phi_1_5_3/attn_phi_1_5_3.onnx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_phi_1_5_3/attn_phi_1_5_3.onnx -------------------------------------------------------------------------------- /testdata/unittest_models/attn_phi_1_5_3/test_data_set_0/input_0.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_phi_1_5_3/test_data_set_0/input_0.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_phi_1_5_3/test_data_set_0/input_1.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_phi_1_5_3/test_data_set_0/input_1.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_phi_1_5_3/test_data_set_0/input_10.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_phi_1_5_3/test_data_set_0/input_10.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_phi_1_5_3/test_data_set_0/input_11.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_phi_1_5_3/test_data_set_0/input_11.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_phi_1_5_3/test_data_set_0/input_12.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_phi_1_5_3/test_data_set_0/input_12.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_phi_1_5_3/test_data_set_0/input_2.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_phi_1_5_3/test_data_set_0/input_2.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_phi_1_5_3/test_data_set_0/input_3.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_phi_1_5_3/test_data_set_0/input_3.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_phi_1_5_3/test_data_set_0/input_4.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_phi_1_5_3/test_data_set_0/input_4.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_phi_1_5_3/test_data_set_0/input_5.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_phi_1_5_3/test_data_set_0/input_5.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_phi_1_5_3/test_data_set_0/input_6.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_phi_1_5_3/test_data_set_0/input_6.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_phi_1_5_3/test_data_set_0/input_7.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_phi_1_5_3/test_data_set_0/input_7.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_phi_1_5_3/test_data_set_0/input_8.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_phi_1_5_3/test_data_set_0/input_8.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_phi_1_5_3/test_data_set_0/input_9.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_phi_1_5_3/test_data_set_0/input_9.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_phi_1_5_3/test_data_set_0/output_0.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_phi_1_5_3/test_data_set_0/output_0.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_phi_1_5_3/test_data_set_0/output_1.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_phi_1_5_3/test_data_set_0/output_1.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_phi_1_5_3/test_data_set_0/output_2.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_phi_1_5_3/test_data_set_0/output_2.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_stable_diffusion_unet_0/attn_stable_diffusion_unet_0.onnx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_stable_diffusion_unet_0/attn_stable_diffusion_unet_0.onnx -------------------------------------------------------------------------------- /testdata/unittest_models/attn_stable_diffusion_unet_0/test_data_set_0/input_0.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_stable_diffusion_unet_0/test_data_set_0/input_0.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_stable_diffusion_unet_0/test_data_set_0/input_1.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_stable_diffusion_unet_0/test_data_set_0/input_1.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_stable_diffusion_unet_0/test_data_set_0/input_2.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_stable_diffusion_unet_0/test_data_set_0/input_2.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_stable_diffusion_unet_0/test_data_set_0/input_3.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_stable_diffusion_unet_0/test_data_set_0/input_3.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_stable_diffusion_unet_0/test_data_set_0/input_4.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_stable_diffusion_unet_0/test_data_set_0/input_4.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_stable_diffusion_unet_0/test_data_set_0/input_5.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_stable_diffusion_unet_0/test_data_set_0/input_5.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_stable_diffusion_unet_0/test_data_set_0/input_6.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_stable_diffusion_unet_0/test_data_set_0/input_6.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_stable_diffusion_unet_0/test_data_set_0/output_0.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_stable_diffusion_unet_0/test_data_set_0/output_0.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_stable_diffusion_unet_1/attn_stable_diffusion_unet_1.onnx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_stable_diffusion_unet_1/attn_stable_diffusion_unet_1.onnx -------------------------------------------------------------------------------- /testdata/unittest_models/attn_stable_diffusion_unet_1/test_data_set_0/input_0.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_stable_diffusion_unet_1/test_data_set_0/input_0.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_stable_diffusion_unet_1/test_data_set_0/input_1.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_stable_diffusion_unet_1/test_data_set_0/input_1.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_stable_diffusion_unet_1/test_data_set_0/input_2.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_stable_diffusion_unet_1/test_data_set_0/input_2.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_stable_diffusion_unet_1/test_data_set_0/input_3.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_stable_diffusion_unet_1/test_data_set_0/input_3.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_stable_diffusion_unet_1/test_data_set_0/input_4.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_stable_diffusion_unet_1/test_data_set_0/input_4.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_stable_diffusion_unet_1/test_data_set_0/input_5.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_stable_diffusion_unet_1/test_data_set_0/input_5.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_stable_diffusion_unet_1/test_data_set_0/input_6.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_stable_diffusion_unet_1/test_data_set_0/input_6.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_stable_diffusion_unet_1/test_data_set_0/output_0.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_stable_diffusion_unet_1/test_data_set_0/output_0.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_stable_diffusion_unet_without_encoder_hidden_states_0/test_data_set_0/input_0.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_stable_diffusion_unet_without_encoder_hidden_states_0/test_data_set_0/input_0.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_stable_diffusion_unet_without_encoder_hidden_states_0/test_data_set_0/input_1.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_stable_diffusion_unet_without_encoder_hidden_states_0/test_data_set_0/input_1.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_stable_diffusion_unet_without_encoder_hidden_states_0/test_data_set_0/input_2.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_stable_diffusion_unet_without_encoder_hidden_states_0/test_data_set_0/input_2.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_stable_diffusion_unet_without_encoder_hidden_states_0/test_data_set_0/input_3.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_stable_diffusion_unet_without_encoder_hidden_states_0/test_data_set_0/input_3.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_stable_diffusion_unet_without_encoder_hidden_states_0/test_data_set_0/input_4.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_stable_diffusion_unet_without_encoder_hidden_states_0/test_data_set_0/input_4.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_stable_diffusion_unet_without_encoder_hidden_states_0/test_data_set_0/input_5.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_stable_diffusion_unet_without_encoder_hidden_states_0/test_data_set_0/input_5.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_stable_diffusion_unet_without_encoder_hidden_states_0/test_data_set_0/output_0.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_stable_diffusion_unet_without_encoder_hidden_states_0/test_data_set_0/output_0.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_stable_diffusion_unet_without_encoder_hidden_states_1/test_data_set_0/input_0.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_stable_diffusion_unet_without_encoder_hidden_states_1/test_data_set_0/input_0.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_stable_diffusion_unet_without_encoder_hidden_states_1/test_data_set_0/input_1.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_stable_diffusion_unet_without_encoder_hidden_states_1/test_data_set_0/input_1.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_stable_diffusion_unet_without_encoder_hidden_states_1/test_data_set_0/input_2.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_stable_diffusion_unet_without_encoder_hidden_states_1/test_data_set_0/input_2.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_stable_diffusion_unet_without_encoder_hidden_states_1/test_data_set_0/input_3.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_stable_diffusion_unet_without_encoder_hidden_states_1/test_data_set_0/input_3.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_stable_diffusion_unet_without_encoder_hidden_states_1/test_data_set_0/input_4.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_stable_diffusion_unet_without_encoder_hidden_states_1/test_data_set_0/input_4.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_yi_4_37_0/attn_yi_4_37_0.onnx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_yi_4_37_0/attn_yi_4_37_0.onnx -------------------------------------------------------------------------------- /testdata/unittest_models/attn_yi_4_37_0/test_data_set_0/input_0.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_yi_4_37_0/test_data_set_0/input_0.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_yi_4_37_0/test_data_set_0/input_1.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_yi_4_37_0/test_data_set_0/input_1.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_yi_4_37_0/test_data_set_0/input_2.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_yi_4_37_0/test_data_set_0/input_2.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_yi_4_37_0/test_data_set_0/input_3.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_yi_4_37_0/test_data_set_0/input_3.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_yi_4_37_0/test_data_set_0/input_4.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_yi_4_37_0/test_data_set_0/input_4.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_yi_4_37_0/test_data_set_0/input_5.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_yi_4_37_0/test_data_set_0/input_5.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_yi_4_37_0/test_data_set_0/input_6.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_yi_4_37_0/test_data_set_0/input_6.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_yi_4_37_0/test_data_set_0/input_7.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_yi_4_37_0/test_data_set_0/input_7.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_yi_4_37_0/test_data_set_0/input_8.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_yi_4_37_0/test_data_set_0/input_8.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_yi_4_37_0/test_data_set_0/output_0.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_yi_4_37_0/test_data_set_0/output_0.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_yi_4_37_0/test_data_set_0/output_1.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_yi_4_37_0/test_data_set_0/output_1.pb -------------------------------------------------------------------------------- /testdata/unittest_models/attn_yi_4_37_0/test_data_set_0/output_2.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/attn_yi_4_37_0/test_data_set_0/output_2.pb -------------------------------------------------------------------------------- /testdata/unittest_models/geglu_stable_diffusion_unet_0/geglu_stable_diffusion_unet_0.onnx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/geglu_stable_diffusion_unet_0/geglu_stable_diffusion_unet_0.onnx -------------------------------------------------------------------------------- /testdata/unittest_models/geglu_stable_diffusion_unet_0/test_data_set_0/input_0.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/geglu_stable_diffusion_unet_0/test_data_set_0/input_0.pb -------------------------------------------------------------------------------- /testdata/unittest_models/geglu_stable_diffusion_unet_0/test_data_set_0/input_1.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/geglu_stable_diffusion_unet_0/test_data_set_0/input_1.pb -------------------------------------------------------------------------------- /testdata/unittest_models/geglu_stable_diffusion_unet_0/test_data_set_0/input_2.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/geglu_stable_diffusion_unet_0/test_data_set_0/input_2.pb -------------------------------------------------------------------------------- /testdata/unittest_models/geglu_stable_diffusion_unet_0/test_data_set_0/output_0.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/geglu_stable_diffusion_unet_0/test_data_set_0/output_0.pb -------------------------------------------------------------------------------- /testdata/unittest_models/geglu_stable_diffusion_unet_1/geglu_stable_diffusion_unet_1.onnx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/geglu_stable_diffusion_unet_1/geglu_stable_diffusion_unet_1.onnx -------------------------------------------------------------------------------- /testdata/unittest_models/geglu_stable_diffusion_unet_1/test_data_set_0/input_0.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/geglu_stable_diffusion_unet_1/test_data_set_0/input_0.pb -------------------------------------------------------------------------------- /testdata/unittest_models/geglu_stable_diffusion_unet_1/test_data_set_0/input_1.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/geglu_stable_diffusion_unet_1/test_data_set_0/input_1.pb -------------------------------------------------------------------------------- /testdata/unittest_models/geglu_stable_diffusion_unet_1/test_data_set_0/input_2.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/geglu_stable_diffusion_unet_1/test_data_set_0/input_2.pb -------------------------------------------------------------------------------- /testdata/unittest_models/geglu_stable_diffusion_unet_1/test_data_set_0/output_0.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/geglu_stable_diffusion_unet_1/test_data_set_0/output_0.pb -------------------------------------------------------------------------------- /testdata/unittest_models/geglu_stable_diffusion_unet_2/geglu_stable_diffusion_unet_2.onnx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/geglu_stable_diffusion_unet_2/geglu_stable_diffusion_unet_2.onnx -------------------------------------------------------------------------------- /testdata/unittest_models/geglu_stable_diffusion_unet_2/test_data_set_0/input_0.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/geglu_stable_diffusion_unet_2/test_data_set_0/input_0.pb -------------------------------------------------------------------------------- /testdata/unittest_models/geglu_stable_diffusion_unet_2/test_data_set_0/input_1.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/geglu_stable_diffusion_unet_2/test_data_set_0/input_1.pb -------------------------------------------------------------------------------- /testdata/unittest_models/geglu_stable_diffusion_unet_2/test_data_set_0/input_2.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/geglu_stable_diffusion_unet_2/test_data_set_0/input_2.pb -------------------------------------------------------------------------------- /testdata/unittest_models/geglu_stable_diffusion_unet_2/test_data_set_0/output_0.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/geglu_stable_diffusion_unet_2/test_data_set_0/output_0.pb -------------------------------------------------------------------------------- /testdata/unittest_models/geglu_stable_diffusion_unet_3/geglu_stable_diffusion_unet_3.onnx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/geglu_stable_diffusion_unet_3/geglu_stable_diffusion_unet_3.onnx -------------------------------------------------------------------------------- /testdata/unittest_models/geglu_stable_diffusion_unet_3/test_data_set_0/input_0.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/geglu_stable_diffusion_unet_3/test_data_set_0/input_0.pb -------------------------------------------------------------------------------- /testdata/unittest_models/geglu_stable_diffusion_unet_3/test_data_set_0/input_1.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/geglu_stable_diffusion_unet_3/test_data_set_0/input_1.pb -------------------------------------------------------------------------------- /testdata/unittest_models/geglu_stable_diffusion_unet_3/test_data_set_0/input_2.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/geglu_stable_diffusion_unet_3/test_data_set_0/input_2.pb -------------------------------------------------------------------------------- /testdata/unittest_models/geglu_stable_diffusion_unet_3/test_data_set_0/output_0.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/geglu_stable_diffusion_unet_3/test_data_set_0/output_0.pb -------------------------------------------------------------------------------- /testdata/unittest_models/gelu_phi_1_5_0/gelu_phi_1_5_0.onnx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/gelu_phi_1_5_0/gelu_phi_1_5_0.onnx -------------------------------------------------------------------------------- /testdata/unittest_models/gelu_phi_1_5_0/test_data_set_0/input_0.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/gelu_phi_1_5_0/test_data_set_0/input_0.pb -------------------------------------------------------------------------------- /testdata/unittest_models/gelu_phi_1_5_0/test_data_set_0/output_0.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/gelu_phi_1_5_0/test_data_set_0/output_0.pb -------------------------------------------------------------------------------- /testdata/unittest_models/gelu_phi_1_5_1/gelu_phi_1_5_1.onnx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/gelu_phi_1_5_1/gelu_phi_1_5_1.onnx -------------------------------------------------------------------------------- /testdata/unittest_models/gelu_phi_1_5_1/test_data_set_0/input_0.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/gelu_phi_1_5_1/test_data_set_0/input_0.pb -------------------------------------------------------------------------------- /testdata/unittest_models/gelu_phi_1_5_1/test_data_set_0/output_0.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/gelu_phi_1_5_1/test_data_set_0/output_0.pb -------------------------------------------------------------------------------- /testdata/unittest_models/gelu_phi_1_5_2/gelu_phi_1_5_2.onnx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/gelu_phi_1_5_2/gelu_phi_1_5_2.onnx -------------------------------------------------------------------------------- /testdata/unittest_models/gelu_phi_1_5_2/test_data_set_0/input_0.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/gelu_phi_1_5_2/test_data_set_0/input_0.pb -------------------------------------------------------------------------------- /testdata/unittest_models/gelu_phi_1_5_2/test_data_set_0/output_0.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/gelu_phi_1_5_2/test_data_set_0/output_0.pb -------------------------------------------------------------------------------- /testdata/unittest_models/gelu_phi_1_5_3/gelu_phi_1_5_3.onnx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/gelu_phi_1_5_3/gelu_phi_1_5_3.onnx -------------------------------------------------------------------------------- /testdata/unittest_models/gelu_phi_1_5_3/test_data_set_0/input_0.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/gelu_phi_1_5_3/test_data_set_0/input_0.pb -------------------------------------------------------------------------------- /testdata/unittest_models/gelu_phi_1_5_3/test_data_set_0/output_0.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/gelu_phi_1_5_3/test_data_set_0/output_0.pb -------------------------------------------------------------------------------- /testdata/unittest_models/ln_llama2_0/ln_llama2_0.onnx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/ln_llama2_0/ln_llama2_0.onnx -------------------------------------------------------------------------------- /testdata/unittest_models/ln_llama2_0/test_data_set_0/input_0.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/ln_llama2_0/test_data_set_0/input_0.pb -------------------------------------------------------------------------------- /testdata/unittest_models/ln_llama2_0/test_data_set_0/input_1.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/ln_llama2_0/test_data_set_0/input_1.pb -------------------------------------------------------------------------------- /testdata/unittest_models/ln_llama2_0/test_data_set_0/output_0.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/ln_llama2_0/test_data_set_0/output_0.pb -------------------------------------------------------------------------------- /testdata/unittest_models/ln_llama2_1/ln_llama2_1.onnx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/ln_llama2_1/ln_llama2_1.onnx -------------------------------------------------------------------------------- /testdata/unittest_models/ln_llama2_1/test_data_set_0/input_0.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/ln_llama2_1/test_data_set_0/input_0.pb -------------------------------------------------------------------------------- /testdata/unittest_models/ln_llama2_1/test_data_set_0/input_1.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/ln_llama2_1/test_data_set_0/input_1.pb -------------------------------------------------------------------------------- /testdata/unittest_models/ln_llama2_1/test_data_set_0/output_0.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/ln_llama2_1/test_data_set_0/output_0.pb -------------------------------------------------------------------------------- /testdata/unittest_models/ln_llama2_2/ln_llama2_2.onnx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/ln_llama2_2/ln_llama2_2.onnx -------------------------------------------------------------------------------- /testdata/unittest_models/ln_llama2_2/test_data_set_0/input_0.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/ln_llama2_2/test_data_set_0/input_0.pb -------------------------------------------------------------------------------- /testdata/unittest_models/ln_llama2_2/test_data_set_0/input_1.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/ln_llama2_2/test_data_set_0/input_1.pb -------------------------------------------------------------------------------- /testdata/unittest_models/ln_llama2_2/test_data_set_0/output_0.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/ln_llama2_2/test_data_set_0/output_0.pb -------------------------------------------------------------------------------- /testdata/unittest_models/ln_llama2_3/ln_llama2_3.onnx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/ln_llama2_3/ln_llama2_3.onnx -------------------------------------------------------------------------------- /testdata/unittest_models/ln_llama2_3/test_data_set_0/input_0.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/ln_llama2_3/test_data_set_0/input_0.pb -------------------------------------------------------------------------------- /testdata/unittest_models/ln_llama2_3/test_data_set_0/input_1.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/ln_llama2_3/test_data_set_0/input_1.pb -------------------------------------------------------------------------------- /testdata/unittest_models/ln_llama2_3/test_data_set_0/output_0.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/ln_llama2_3/test_data_set_0/output_0.pb -------------------------------------------------------------------------------- /testdata/unittest_models/sdpa_llama2_0/sdpa_llama2_0.onnx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/sdpa_llama2_0/sdpa_llama2_0.onnx -------------------------------------------------------------------------------- /testdata/unittest_models/sdpa_llama2_0/test_data_set_0/input_0.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/sdpa_llama2_0/test_data_set_0/input_0.pb -------------------------------------------------------------------------------- /testdata/unittest_models/sdpa_llama2_0/test_data_set_0/input_1.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/sdpa_llama2_0/test_data_set_0/input_1.pb -------------------------------------------------------------------------------- /testdata/unittest_models/sdpa_llama2_0/test_data_set_0/input_2.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/sdpa_llama2_0/test_data_set_0/input_2.pb -------------------------------------------------------------------------------- /testdata/unittest_models/sdpa_llama2_0/test_data_set_0/input_3.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/sdpa_llama2_0/test_data_set_0/input_3.pb -------------------------------------------------------------------------------- /testdata/unittest_models/sdpa_llama2_0/test_data_set_0/input_4.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/sdpa_llama2_0/test_data_set_0/input_4.pb -------------------------------------------------------------------------------- /testdata/unittest_models/sdpa_llama2_0/test_data_set_0/input_5.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/sdpa_llama2_0/test_data_set_0/input_5.pb -------------------------------------------------------------------------------- /testdata/unittest_models/sdpa_llama2_0/test_data_set_0/input_6.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/sdpa_llama2_0/test_data_set_0/input_6.pb -------------------------------------------------------------------------------- /testdata/unittest_models/sdpa_llama2_0/test_data_set_0/input_7.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/sdpa_llama2_0/test_data_set_0/input_7.pb -------------------------------------------------------------------------------- /testdata/unittest_models/sdpa_llama2_0/test_data_set_0/output_0.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/sdpa_llama2_0/test_data_set_0/output_0.pb -------------------------------------------------------------------------------- /testdata/unittest_models/sdpa_llama2_0/test_data_set_0/output_1.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/sdpa_llama2_0/test_data_set_0/output_1.pb -------------------------------------------------------------------------------- /testdata/unittest_models/sdpa_llama2_0/test_data_set_0/output_2.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/sdpa_llama2_0/test_data_set_0/output_2.pb -------------------------------------------------------------------------------- /testdata/unittest_models/sdpa_llama2_1/sdpa_llama2_1.onnx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/sdpa_llama2_1/sdpa_llama2_1.onnx -------------------------------------------------------------------------------- /testdata/unittest_models/sdpa_llama2_1/test_data_set_0/input_0.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/sdpa_llama2_1/test_data_set_0/input_0.pb -------------------------------------------------------------------------------- /testdata/unittest_models/sdpa_llama2_1/test_data_set_0/input_1.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/sdpa_llama2_1/test_data_set_0/input_1.pb -------------------------------------------------------------------------------- /testdata/unittest_models/sdpa_llama2_1/test_data_set_0/input_2.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/sdpa_llama2_1/test_data_set_0/input_2.pb -------------------------------------------------------------------------------- /testdata/unittest_models/sdpa_llama2_1/test_data_set_0/input_3.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/sdpa_llama2_1/test_data_set_0/input_3.pb -------------------------------------------------------------------------------- /testdata/unittest_models/sdpa_llama2_1/test_data_set_0/input_4.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/sdpa_llama2_1/test_data_set_0/input_4.pb -------------------------------------------------------------------------------- /testdata/unittest_models/sdpa_llama2_1/test_data_set_0/input_5.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/sdpa_llama2_1/test_data_set_0/input_5.pb -------------------------------------------------------------------------------- /testdata/unittest_models/sdpa_llama2_1/test_data_set_0/input_6.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/sdpa_llama2_1/test_data_set_0/input_6.pb -------------------------------------------------------------------------------- /testdata/unittest_models/sdpa_llama2_1/test_data_set_0/input_7.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/sdpa_llama2_1/test_data_set_0/input_7.pb -------------------------------------------------------------------------------- /testdata/unittest_models/sdpa_llama2_1/test_data_set_0/output_0.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/sdpa_llama2_1/test_data_set_0/output_0.pb -------------------------------------------------------------------------------- /testdata/unittest_models/sdpa_llama2_1/test_data_set_0/output_1.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/sdpa_llama2_1/test_data_set_0/output_1.pb -------------------------------------------------------------------------------- /testdata/unittest_models/sdpa_llama2_1/test_data_set_0/output_2.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/sdpa_llama2_1/test_data_set_0/output_2.pb -------------------------------------------------------------------------------- /testdata/unittest_models/sdpa_llama2_2/sdpa_llama2_2.onnx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/sdpa_llama2_2/sdpa_llama2_2.onnx -------------------------------------------------------------------------------- /testdata/unittest_models/sdpa_llama2_2/test_data_set_0/input_0.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/sdpa_llama2_2/test_data_set_0/input_0.pb -------------------------------------------------------------------------------- /testdata/unittest_models/sdpa_llama2_2/test_data_set_0/input_1.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/sdpa_llama2_2/test_data_set_0/input_1.pb -------------------------------------------------------------------------------- /testdata/unittest_models/sdpa_llama2_2/test_data_set_0/input_2.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/sdpa_llama2_2/test_data_set_0/input_2.pb -------------------------------------------------------------------------------- /testdata/unittest_models/sdpa_llama2_2/test_data_set_0/input_3.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/sdpa_llama2_2/test_data_set_0/input_3.pb -------------------------------------------------------------------------------- /testdata/unittest_models/sdpa_llama2_2/test_data_set_0/input_4.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/sdpa_llama2_2/test_data_set_0/input_4.pb -------------------------------------------------------------------------------- /testdata/unittest_models/sdpa_llama2_2/test_data_set_0/input_5.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/sdpa_llama2_2/test_data_set_0/input_5.pb -------------------------------------------------------------------------------- /testdata/unittest_models/sdpa_llama2_2/test_data_set_0/input_6.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/sdpa_llama2_2/test_data_set_0/input_6.pb -------------------------------------------------------------------------------- /testdata/unittest_models/sdpa_llama2_2/test_data_set_0/input_7.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/sdpa_llama2_2/test_data_set_0/input_7.pb -------------------------------------------------------------------------------- /testdata/unittest_models/sdpa_llama2_2/test_data_set_0/output_0.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/sdpa_llama2_2/test_data_set_0/output_0.pb -------------------------------------------------------------------------------- /testdata/unittest_models/sdpa_llama2_2/test_data_set_0/output_1.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/sdpa_llama2_2/test_data_set_0/output_1.pb -------------------------------------------------------------------------------- /testdata/unittest_models/sdpa_llama2_2/test_data_set_0/output_2.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/sdpa_llama2_2/test_data_set_0/output_2.pb -------------------------------------------------------------------------------- /testdata/unittest_models/sdpa_llama2_3/sdpa_llama2_3.onnx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/sdpa_llama2_3/sdpa_llama2_3.onnx -------------------------------------------------------------------------------- /testdata/unittest_models/sdpa_llama2_3/test_data_set_0/input_0.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/sdpa_llama2_3/test_data_set_0/input_0.pb -------------------------------------------------------------------------------- /testdata/unittest_models/sdpa_llama2_3/test_data_set_0/input_1.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/sdpa_llama2_3/test_data_set_0/input_1.pb -------------------------------------------------------------------------------- /testdata/unittest_models/sdpa_llama2_3/test_data_set_0/input_2.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/sdpa_llama2_3/test_data_set_0/input_2.pb -------------------------------------------------------------------------------- /testdata/unittest_models/sdpa_llama2_3/test_data_set_0/input_3.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/sdpa_llama2_3/test_data_set_0/input_3.pb -------------------------------------------------------------------------------- /testdata/unittest_models/sdpa_llama2_3/test_data_set_0/input_4.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/sdpa_llama2_3/test_data_set_0/input_4.pb -------------------------------------------------------------------------------- /testdata/unittest_models/sdpa_llama2_3/test_data_set_0/input_5.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/sdpa_llama2_3/test_data_set_0/input_5.pb -------------------------------------------------------------------------------- /testdata/unittest_models/sdpa_llama2_3/test_data_set_0/input_6.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/sdpa_llama2_3/test_data_set_0/input_6.pb -------------------------------------------------------------------------------- /testdata/unittest_models/sdpa_llama2_3/test_data_set_0/input_7.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/sdpa_llama2_3/test_data_set_0/input_7.pb -------------------------------------------------------------------------------- /testdata/unittest_models/sdpa_llama2_3/test_data_set_0/output_0.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/sdpa_llama2_3/test_data_set_0/output_0.pb -------------------------------------------------------------------------------- /testdata/unittest_models/sdpa_llama2_3/test_data_set_0/output_1.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/sdpa_llama2_3/test_data_set_0/output_1.pb -------------------------------------------------------------------------------- /testdata/unittest_models/sdpa_llama2_3/test_data_set_0/output_2.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/sdpa_llama2_3/test_data_set_0/output_2.pb -------------------------------------------------------------------------------- /testdata/unittest_models/sdpa_llama2_4_38_0/sdpa_llama2_4_38_0.onnx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/sdpa_llama2_4_38_0/sdpa_llama2_4_38_0.onnx -------------------------------------------------------------------------------- /testdata/unittest_models/sdpa_llama2_4_38_0/test_data_set_0/input_0.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/sdpa_llama2_4_38_0/test_data_set_0/input_0.pb -------------------------------------------------------------------------------- /testdata/unittest_models/sdpa_llama2_4_38_0/test_data_set_0/input_1.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/sdpa_llama2_4_38_0/test_data_set_0/input_1.pb -------------------------------------------------------------------------------- /testdata/unittest_models/sdpa_llama2_4_38_0/test_data_set_0/input_2.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/sdpa_llama2_4_38_0/test_data_set_0/input_2.pb -------------------------------------------------------------------------------- /testdata/unittest_models/sdpa_llama2_4_38_0/test_data_set_0/input_3.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/sdpa_llama2_4_38_0/test_data_set_0/input_3.pb -------------------------------------------------------------------------------- /testdata/unittest_models/sdpa_llama2_4_38_0/test_data_set_0/input_4.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/sdpa_llama2_4_38_0/test_data_set_0/input_4.pb -------------------------------------------------------------------------------- /testdata/unittest_models/sdpa_llama2_4_38_0/test_data_set_0/input_5.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/sdpa_llama2_4_38_0/test_data_set_0/input_5.pb -------------------------------------------------------------------------------- /testdata/unittest_models/sdpa_llama2_4_38_0/test_data_set_0/input_6.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/sdpa_llama2_4_38_0/test_data_set_0/input_6.pb -------------------------------------------------------------------------------- /testdata/unittest_models/sdpa_llama2_4_38_0/test_data_set_0/input_7.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/sdpa_llama2_4_38_0/test_data_set_0/input_7.pb -------------------------------------------------------------------------------- /testdata/unittest_models/sdpa_llama2_4_38_0/test_data_set_0/input_8.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/sdpa_llama2_4_38_0/test_data_set_0/input_8.pb -------------------------------------------------------------------------------- /testdata/unittest_models/sdpa_llama2_4_38_0/test_data_set_0/output_0.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/sdpa_llama2_4_38_0/test_data_set_0/output_0.pb -------------------------------------------------------------------------------- /testdata/unittest_models/sdpa_llama2_4_38_0/test_data_set_0/output_1.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/sdpa_llama2_4_38_0/test_data_set_0/output_1.pb -------------------------------------------------------------------------------- /testdata/unittest_models/sdpa_llama2_4_38_0/test_data_set_0/output_2.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/sdpa_llama2_4_38_0/test_data_set_0/output_2.pb -------------------------------------------------------------------------------- /testdata/unittest_models/sdpa_yi_0/sdpa_yi_0.onnx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/sdpa_yi_0/sdpa_yi_0.onnx -------------------------------------------------------------------------------- /testdata/unittest_models/sdpa_yi_0/test_data_set_0/input_0.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/sdpa_yi_0/test_data_set_0/input_0.pb -------------------------------------------------------------------------------- /testdata/unittest_models/sdpa_yi_0/test_data_set_0/input_1.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/sdpa_yi_0/test_data_set_0/input_1.pb -------------------------------------------------------------------------------- /testdata/unittest_models/sdpa_yi_0/test_data_set_0/input_2.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/sdpa_yi_0/test_data_set_0/input_2.pb -------------------------------------------------------------------------------- /testdata/unittest_models/sdpa_yi_0/test_data_set_0/input_3.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/sdpa_yi_0/test_data_set_0/input_3.pb -------------------------------------------------------------------------------- /testdata/unittest_models/sdpa_yi_0/test_data_set_0/input_4.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/sdpa_yi_0/test_data_set_0/input_4.pb -------------------------------------------------------------------------------- /testdata/unittest_models/sdpa_yi_0/test_data_set_0/input_5.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/sdpa_yi_0/test_data_set_0/input_5.pb -------------------------------------------------------------------------------- /testdata/unittest_models/sdpa_yi_0/test_data_set_0/input_6.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/sdpa_yi_0/test_data_set_0/input_6.pb -------------------------------------------------------------------------------- /testdata/unittest_models/sdpa_yi_0/test_data_set_0/input_7.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/sdpa_yi_0/test_data_set_0/input_7.pb -------------------------------------------------------------------------------- /testdata/unittest_models/sdpa_yi_0/test_data_set_0/output_0.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/sdpa_yi_0/test_data_set_0/output_0.pb -------------------------------------------------------------------------------- /testdata/unittest_models/sdpa_yi_0/test_data_set_0/output_1.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/sdpa_yi_0/test_data_set_0/output_1.pb -------------------------------------------------------------------------------- /testdata/unittest_models/sdpa_yi_0/test_data_set_0/output_2.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/sdpa_yi_0/test_data_set_0/output_2.pb -------------------------------------------------------------------------------- /testdata/unittest_models/sdpa_yi_1/sdpa_yi_1.onnx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/sdpa_yi_1/sdpa_yi_1.onnx -------------------------------------------------------------------------------- /testdata/unittest_models/sdpa_yi_1/test_data_set_0/input_0.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/sdpa_yi_1/test_data_set_0/input_0.pb -------------------------------------------------------------------------------- /testdata/unittest_models/sdpa_yi_1/test_data_set_0/input_1.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/sdpa_yi_1/test_data_set_0/input_1.pb -------------------------------------------------------------------------------- /testdata/unittest_models/sdpa_yi_1/test_data_set_0/input_2.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/sdpa_yi_1/test_data_set_0/input_2.pb -------------------------------------------------------------------------------- /testdata/unittest_models/sdpa_yi_1/test_data_set_0/input_3.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/sdpa_yi_1/test_data_set_0/input_3.pb -------------------------------------------------------------------------------- /testdata/unittest_models/sdpa_yi_1/test_data_set_0/input_4.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/sdpa_yi_1/test_data_set_0/input_4.pb -------------------------------------------------------------------------------- /testdata/unittest_models/sdpa_yi_1/test_data_set_0/input_5.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/sdpa_yi_1/test_data_set_0/input_5.pb -------------------------------------------------------------------------------- /testdata/unittest_models/sdpa_yi_1/test_data_set_0/input_6.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/sdpa_yi_1/test_data_set_0/input_6.pb -------------------------------------------------------------------------------- /testdata/unittest_models/sdpa_yi_1/test_data_set_0/input_7.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/sdpa_yi_1/test_data_set_0/input_7.pb -------------------------------------------------------------------------------- /testdata/unittest_models/sdpa_yi_1/test_data_set_0/output_0.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/sdpa_yi_1/test_data_set_0/output_0.pb -------------------------------------------------------------------------------- /testdata/unittest_models/sdpa_yi_1/test_data_set_0/output_1.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/sdpa_yi_1/test_data_set_0/output_1.pb -------------------------------------------------------------------------------- /testdata/unittest_models/sdpa_yi_1/test_data_set_0/output_2.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/sdpa_yi_1/test_data_set_0/output_2.pb -------------------------------------------------------------------------------- /testdata/unittest_models/sdpa_yi_4_38_0/sdpa_yi_4_38_0.onnx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/sdpa_yi_4_38_0/sdpa_yi_4_38_0.onnx -------------------------------------------------------------------------------- /testdata/unittest_models/sdpa_yi_4_38_0/test_data_set_0/input_0.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/sdpa_yi_4_38_0/test_data_set_0/input_0.pb -------------------------------------------------------------------------------- /testdata/unittest_models/sdpa_yi_4_38_0/test_data_set_0/input_1.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/sdpa_yi_4_38_0/test_data_set_0/input_1.pb -------------------------------------------------------------------------------- /testdata/unittest_models/sdpa_yi_4_38_0/test_data_set_0/input_2.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/sdpa_yi_4_38_0/test_data_set_0/input_2.pb -------------------------------------------------------------------------------- /testdata/unittest_models/sdpa_yi_4_38_0/test_data_set_0/input_3.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/sdpa_yi_4_38_0/test_data_set_0/input_3.pb -------------------------------------------------------------------------------- /testdata/unittest_models/sdpa_yi_4_38_0/test_data_set_0/input_4.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/sdpa_yi_4_38_0/test_data_set_0/input_4.pb -------------------------------------------------------------------------------- /testdata/unittest_models/sdpa_yi_4_38_0/test_data_set_0/input_5.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/sdpa_yi_4_38_0/test_data_set_0/input_5.pb -------------------------------------------------------------------------------- /testdata/unittest_models/sdpa_yi_4_38_0/test_data_set_0/input_6.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/sdpa_yi_4_38_0/test_data_set_0/input_6.pb -------------------------------------------------------------------------------- /testdata/unittest_models/sdpa_yi_4_38_0/test_data_set_0/input_7.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/sdpa_yi_4_38_0/test_data_set_0/input_7.pb -------------------------------------------------------------------------------- /testdata/unittest_models/sdpa_yi_4_38_0/test_data_set_0/input_8.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/sdpa_yi_4_38_0/test_data_set_0/input_8.pb -------------------------------------------------------------------------------- /testdata/unittest_models/sdpa_yi_4_38_0/test_data_set_0/output_0.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/sdpa_yi_4_38_0/test_data_set_0/output_0.pb -------------------------------------------------------------------------------- /testdata/unittest_models/sdpa_yi_4_38_0/test_data_set_0/output_1.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/sdpa_yi_4_38_0/test_data_set_0/output_1.pb -------------------------------------------------------------------------------- /testdata/unittest_models/sdpa_yi_4_38_0/test_data_set_0/output_2.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/testdata/unittest_models/sdpa_yi_4_38_0/test_data_set_0/output_2.pb -------------------------------------------------------------------------------- /tests/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/tests/README.md -------------------------------------------------------------------------------- /tests/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/tests/__init__.py -------------------------------------------------------------------------------- /tests/common/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/tests/common/__init__.py -------------------------------------------------------------------------------- /tests/common/onnx_script_test_case.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/tests/common/onnx_script_test_case.py -------------------------------------------------------------------------------- /tests/common/testutils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/tests/common/testutils.py -------------------------------------------------------------------------------- /tests/eager_mode_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/tests/eager_mode_test.py -------------------------------------------------------------------------------- /tests/eager_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/tests/eager_test.py -------------------------------------------------------------------------------- /tests/external_tensor_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/tests/external_tensor_test.py -------------------------------------------------------------------------------- /tests/function_libs/__init__.py: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /tests/function_libs/torch_lib/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/tests/function_libs/torch_lib/README.md -------------------------------------------------------------------------------- /tests/function_libs/torch_lib/__init__.py: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /tests/function_libs/torch_lib/e2e_ops_tests.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/tests/function_libs/torch_lib/e2e_ops_tests.py -------------------------------------------------------------------------------- /tests/function_libs/torch_lib/error_reproduction.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/tests/function_libs/torch_lib/error_reproduction.py -------------------------------------------------------------------------------- /tests/function_libs/torch_lib/extra_opinfo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/tests/function_libs/torch_lib/extra_opinfo.py -------------------------------------------------------------------------------- /tests/function_libs/torch_lib/ops_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/tests/function_libs/torch_lib/ops_test.py -------------------------------------------------------------------------------- /tests/function_libs/torch_lib/ops_test_common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/tests/function_libs/torch_lib/ops_test_common.py -------------------------------------------------------------------------------- /tests/function_libs/torch_lib/ops_test_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/tests/function_libs/torch_lib/ops_test_data.py -------------------------------------------------------------------------------- /tests/functions/attr_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/tests/functions/attr_test.py -------------------------------------------------------------------------------- /tests/functions/gemmgelu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/tests/functions/gemmgelu.py -------------------------------------------------------------------------------- /tests/functions/gemmgelu_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/tests/functions/gemmgelu_test.py -------------------------------------------------------------------------------- /tests/functions/if_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/tests/functions/if_test.py -------------------------------------------------------------------------------- /tests/functions/onnxfns1A_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/tests/functions/onnxfns1A_test.py -------------------------------------------------------------------------------- /tests/functions/onnxfns2_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/tests/functions/onnxfns2_test.py -------------------------------------------------------------------------------- /tests/functions/onnxfns_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/tests/functions/onnxfns_test.py -------------------------------------------------------------------------------- /tests/functions/ort_custom_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/tests/functions/ort_custom_ops.py -------------------------------------------------------------------------------- /tests/if_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/tests/if_test.py -------------------------------------------------------------------------------- /tests/ir/graph_view_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/tests/ir/graph_view_test.py -------------------------------------------------------------------------------- /tests/ir/serde_roundtrip_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/tests/ir/serde_roundtrip_test.py -------------------------------------------------------------------------------- /tests/loop_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/tests/loop_test.py -------------------------------------------------------------------------------- /tests/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/tests/models/__init__.py -------------------------------------------------------------------------------- /tests/models/attrref.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/tests/models/attrref.py -------------------------------------------------------------------------------- /tests/models/cast_like.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/tests/models/cast_like.py -------------------------------------------------------------------------------- /tests/models/different_opset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/tests/models/different_opset.py -------------------------------------------------------------------------------- /tests/models/dropout.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/tests/models/dropout.py -------------------------------------------------------------------------------- /tests/models/eager_op.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/tests/models/eager_op.py -------------------------------------------------------------------------------- /tests/models/eg1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/tests/models/eg1.py -------------------------------------------------------------------------------- /tests/models/getitem.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/tests/models/getitem.py -------------------------------------------------------------------------------- /tests/models/graph_attr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/tests/models/graph_attr.py -------------------------------------------------------------------------------- /tests/models/identity.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/tests/models/identity.py -------------------------------------------------------------------------------- /tests/models/if_statement.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/tests/models/if_statement.py -------------------------------------------------------------------------------- /tests/models/loops_break.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/tests/models/loops_break.py -------------------------------------------------------------------------------- /tests/models/loops_while.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/tests/models/loops_while.py -------------------------------------------------------------------------------- /tests/models/m1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/tests/models/m1.py -------------------------------------------------------------------------------- /tests/models/multi.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/tests/models/multi.py -------------------------------------------------------------------------------- /tests/models/onnxfns1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/tests/models/onnxfns1.py -------------------------------------------------------------------------------- /tests/models/onnxfns1A.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/tests/models/onnxfns1A.py -------------------------------------------------------------------------------- /tests/models/onnxfns2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/tests/models/onnxfns2.py -------------------------------------------------------------------------------- /tests/models/opt_input.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/tests/models/opt_input.py -------------------------------------------------------------------------------- /tests/models/opt_output.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/tests/models/opt_output.py -------------------------------------------------------------------------------- /tests/models/renaming.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/tests/models/renaming.py -------------------------------------------------------------------------------- /tests/models/sequences.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/tests/models/sequences.py -------------------------------------------------------------------------------- /tests/models/signal_dft.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/tests/models/signal_dft.py -------------------------------------------------------------------------------- /tests/models/subfunction.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/tests/models/subfunction.py -------------------------------------------------------------------------------- /tests/models/type_double.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/tests/models/type_double.py -------------------------------------------------------------------------------- /tests/onnx_types_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/tests/onnx_types_test.py -------------------------------------------------------------------------------- /tests/operator_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/tests/operator_test.py -------------------------------------------------------------------------------- /tests/optimizer/test_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/tests/optimizer/test_models.py -------------------------------------------------------------------------------- /tests/version_converter/version_conversion_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/tests/version_converter/version_conversion_test.py -------------------------------------------------------------------------------- /tools/ir/model_zoo_test/model_zoo_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/tools/ir/model_zoo_test/model_zoo_test.py -------------------------------------------------------------------------------- /tools/onnx2external.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/tools/onnx2external.py -------------------------------------------------------------------------------- /tools/onnx2script.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/tools/onnx2script.py -------------------------------------------------------------------------------- /tools/optimize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/tools/optimize.py -------------------------------------------------------------------------------- /tools/ort_rewriter_profiling/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/tools/ort_rewriter_profiling/README.md -------------------------------------------------------------------------------- /tools/ort_rewriter_profiling/bench_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/tools/ort_rewriter_profiling/bench_model.py -------------------------------------------------------------------------------- /tools/ort_rewriter_profiling/nsys_profile.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/tools/ort_rewriter_profiling/nsys_profile.py -------------------------------------------------------------------------------- /tools/ort_rewriter_profiling/ort_rewrite.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/tools/ort_rewriter_profiling/ort_rewrite.py -------------------------------------------------------------------------------- /tools/ort_rewriter_profiling/profile_analysis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/onnxscript/HEAD/tools/ort_rewriter_profiling/profile_analysis.py --------------------------------------------------------------------------------