onnxruntime-directml 1.24.1__cp314-cp314-win_amd64.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- onnxruntime/LICENSE +21 -0
- onnxruntime/Privacy.md +21 -0
- onnxruntime/ThirdPartyNotices.txt +6121 -0
- onnxruntime/__init__.py +418 -0
- onnxruntime/backend/__init__.py +6 -0
- onnxruntime/backend/backend.py +175 -0
- onnxruntime/backend/backend_rep.py +52 -0
- onnxruntime/capi/DirectML.dll +0 -0
- onnxruntime/capi/__init__.py +4 -0
- onnxruntime/capi/_ld_preload.py +7 -0
- onnxruntime/capi/_pybind_state.py +33 -0
- onnxruntime/capi/build_and_package_info.py +2 -0
- onnxruntime/capi/convert_npz_to_onnx_adapter.py +48 -0
- onnxruntime/capi/onnxruntime.dll +0 -0
- onnxruntime/capi/onnxruntime_collect_build_info.py +47 -0
- onnxruntime/capi/onnxruntime_inference_collection.py +1440 -0
- onnxruntime/capi/onnxruntime_providers_shared.dll +0 -0
- onnxruntime/capi/onnxruntime_pybind11_state.pyd +0 -0
- onnxruntime/capi/onnxruntime_validation.py +154 -0
- onnxruntime/capi/version_info.py +2 -0
- onnxruntime/datasets/__init__.py +18 -0
- onnxruntime/datasets/logreg_iris.onnx +0 -0
- onnxruntime/datasets/mul_1.onnx +0 -0
- onnxruntime/datasets/sigmoid.onnx +13 -0
- onnxruntime/quantization/CalTableFlatBuffers/KeyValue.py +78 -0
- onnxruntime/quantization/CalTableFlatBuffers/TrtTable.py +90 -0
- onnxruntime/quantization/CalTableFlatBuffers/__init__.py +0 -0
- onnxruntime/quantization/__init__.py +19 -0
- onnxruntime/quantization/base_quantizer.py +529 -0
- onnxruntime/quantization/calibrate.py +1267 -0
- onnxruntime/quantization/execution_providers/qnn/__init__.py +2 -0
- onnxruntime/quantization/execution_providers/qnn/fusion_lpnorm.py +132 -0
- onnxruntime/quantization/execution_providers/qnn/fusion_spacetodepth.py +162 -0
- onnxruntime/quantization/execution_providers/qnn/mixed_precision_overrides_utils.py +413 -0
- onnxruntime/quantization/execution_providers/qnn/preprocess.py +353 -0
- onnxruntime/quantization/execution_providers/qnn/quant_config.py +389 -0
- onnxruntime/quantization/fusions/__init__.py +4 -0
- onnxruntime/quantization/fusions/fusion.py +311 -0
- onnxruntime/quantization/fusions/fusion_gelu.py +272 -0
- onnxruntime/quantization/fusions/fusion_layernorm.py +146 -0
- onnxruntime/quantization/fusions/replace_upsample_with_resize.py +96 -0
- onnxruntime/quantization/matmul_bnb4_quantizer.py +239 -0
- onnxruntime/quantization/matmul_nbits_quantizer.py +1638 -0
- onnxruntime/quantization/neural_compressor/__init__.py +1 -0
- onnxruntime/quantization/neural_compressor/onnx_model.py +1251 -0
- onnxruntime/quantization/neural_compressor/util.py +80 -0
- onnxruntime/quantization/neural_compressor/weight_only.py +932 -0
- onnxruntime/quantization/onnx_model.py +600 -0
- onnxruntime/quantization/onnx_quantizer.py +1163 -0
- onnxruntime/quantization/operators/__init__.py +2 -0
- onnxruntime/quantization/operators/activation.py +119 -0
- onnxruntime/quantization/operators/argmax.py +18 -0
- onnxruntime/quantization/operators/attention.py +73 -0
- onnxruntime/quantization/operators/base_operator.py +26 -0
- onnxruntime/quantization/operators/binary_op.py +72 -0
- onnxruntime/quantization/operators/concat.py +62 -0
- onnxruntime/quantization/operators/conv.py +260 -0
- onnxruntime/quantization/operators/direct_q8.py +78 -0
- onnxruntime/quantization/operators/embed_layernorm.py +121 -0
- onnxruntime/quantization/operators/gather.py +64 -0
- onnxruntime/quantization/operators/gavgpool.py +62 -0
- onnxruntime/quantization/operators/gemm.py +172 -0
- onnxruntime/quantization/operators/lstm.py +121 -0
- onnxruntime/quantization/operators/matmul.py +231 -0
- onnxruntime/quantization/operators/maxpool.py +34 -0
- onnxruntime/quantization/operators/norm.py +40 -0
- onnxruntime/quantization/operators/pad.py +172 -0
- onnxruntime/quantization/operators/pooling.py +67 -0
- onnxruntime/quantization/operators/qdq_base_operator.py +22 -0
- onnxruntime/quantization/operators/resize.py +34 -0
- onnxruntime/quantization/operators/softmax.py +74 -0
- onnxruntime/quantization/operators/split.py +63 -0
- onnxruntime/quantization/operators/where.py +87 -0
- onnxruntime/quantization/preprocess.py +141 -0
- onnxruntime/quantization/qdq_loss_debug.py +389 -0
- onnxruntime/quantization/qdq_quantizer.py +1477 -0
- onnxruntime/quantization/quant_utils.py +1051 -0
- onnxruntime/quantization/quantize.py +953 -0
- onnxruntime/quantization/registry.py +110 -0
- onnxruntime/quantization/shape_inference.py +204 -0
- onnxruntime/quantization/static_quantize_runner.py +256 -0
- onnxruntime/quantization/tensor_quant_overrides.py +520 -0
- onnxruntime/tools/__init__.py +10 -0
- onnxruntime/tools/check_onnx_model_mobile_usability.py +47 -0
- onnxruntime/tools/convert_onnx_models_to_ort.py +380 -0
- onnxruntime/tools/file_utils.py +47 -0
- onnxruntime/tools/logger.py +11 -0
- onnxruntime/tools/make_dynamic_shape_fixed.py +73 -0
- onnxruntime/tools/mobile_helpers/__init__.py +0 -0
- onnxruntime/tools/mobile_helpers/coreml_supported_mlprogram_ops.md +53 -0
- onnxruntime/tools/mobile_helpers/coreml_supported_neuralnetwork_ops.md +43 -0
- onnxruntime/tools/mobile_helpers/nnapi_supported_ops.md +58 -0
- onnxruntime/tools/mobile_helpers/usability_checker.py +738 -0
- onnxruntime/tools/offline_tuning.py +169 -0
- onnxruntime/tools/onnx_model_utils.py +416 -0
- onnxruntime/tools/onnx_randomizer.py +85 -0
- onnxruntime/tools/onnxruntime_test.py +164 -0
- onnxruntime/tools/optimize_onnx_model.py +56 -0
- onnxruntime/tools/ort_format_model/__init__.py +27 -0
- onnxruntime/tools/ort_format_model/operator_type_usage_processors.py +653 -0
- onnxruntime/tools/ort_format_model/ort_flatbuffers_py/__init__.py +0 -0
- onnxruntime/tools/ort_format_model/ort_flatbuffers_py/fbs/ArgType.py +7 -0
- onnxruntime/tools/ort_format_model/ort_flatbuffers_py/fbs/ArgTypeAndIndex.py +67 -0
- onnxruntime/tools/ort_format_model/ort_flatbuffers_py/fbs/Attribute.py +337 -0
- onnxruntime/tools/ort_format_model/ort_flatbuffers_py/fbs/AttributeType.py +18 -0
- onnxruntime/tools/ort_format_model/ort_flatbuffers_py/fbs/Checkpoint.py +125 -0
- onnxruntime/tools/ort_format_model/ort_flatbuffers_py/fbs/DeprecatedKernelCreateInfos.py +120 -0
- onnxruntime/tools/ort_format_model/ort_flatbuffers_py/fbs/DeprecatedNodeIndexAndKernelDefHash.py +68 -0
- onnxruntime/tools/ort_format_model/ort_flatbuffers_py/fbs/DeprecatedSessionState.py +96 -0
- onnxruntime/tools/ort_format_model/ort_flatbuffers_py/fbs/DeprecatedSubGraphSessionState.py +72 -0
- onnxruntime/tools/ort_format_model/ort_flatbuffers_py/fbs/Dimension.py +71 -0
- onnxruntime/tools/ort_format_model/ort_flatbuffers_py/fbs/DimensionValue.py +80 -0
- onnxruntime/tools/ort_format_model/ort_flatbuffers_py/fbs/DimensionValueType.py +8 -0
- onnxruntime/tools/ort_format_model/ort_flatbuffers_py/fbs/EdgeEnd.py +32 -0
- onnxruntime/tools/ort_format_model/ort_flatbuffers_py/fbs/FloatProperty.py +67 -0
- onnxruntime/tools/ort_format_model/ort_flatbuffers_py/fbs/Graph.py +320 -0
- onnxruntime/tools/ort_format_model/ort_flatbuffers_py/fbs/InferenceSession.py +88 -0
- onnxruntime/tools/ort_format_model/ort_flatbuffers_py/fbs/IntProperty.py +67 -0
- onnxruntime/tools/ort_format_model/ort_flatbuffers_py/fbs/KernelTypeStrArgsEntry.py +91 -0
- onnxruntime/tools/ort_format_model/ort_flatbuffers_py/fbs/KernelTypeStrResolver.py +78 -0
- onnxruntime/tools/ort_format_model/ort_flatbuffers_py/fbs/MapType.py +71 -0
- onnxruntime/tools/ort_format_model/ort_flatbuffers_py/fbs/Model.py +223 -0
- onnxruntime/tools/ort_format_model/ort_flatbuffers_py/fbs/ModuleState.py +141 -0
- onnxruntime/tools/ort_format_model/ort_flatbuffers_py/fbs/Node.py +317 -0
- onnxruntime/tools/ort_format_model/ort_flatbuffers_py/fbs/NodeEdge.py +126 -0
- onnxruntime/tools/ort_format_model/ort_flatbuffers_py/fbs/NodeType.py +7 -0
- onnxruntime/tools/ort_format_model/ort_flatbuffers_py/fbs/NodesToOptimizeIndices.py +160 -0
- onnxruntime/tools/ort_format_model/ort_flatbuffers_py/fbs/OpIdKernelTypeStrArgsEntry.py +91 -0
- onnxruntime/tools/ort_format_model/ort_flatbuffers_py/fbs/OperatorSetId.py +67 -0
- onnxruntime/tools/ort_format_model/ort_flatbuffers_py/fbs/OptimizerGroup.py +117 -0
- onnxruntime/tools/ort_format_model/ort_flatbuffers_py/fbs/ParameterOptimizerState.py +91 -0
- onnxruntime/tools/ort_format_model/ort_flatbuffers_py/fbs/PropertyBag.py +152 -0
- onnxruntime/tools/ort_format_model/ort_flatbuffers_py/fbs/RuntimeOptimizationRecord.py +105 -0
- onnxruntime/tools/ort_format_model/ort_flatbuffers_py/fbs/RuntimeOptimizationRecordContainerEntry.py +91 -0
- onnxruntime/tools/ort_format_model/ort_flatbuffers_py/fbs/RuntimeOptimizations.py +79 -0
- onnxruntime/tools/ort_format_model/ort_flatbuffers_py/fbs/SequenceType.py +58 -0
- onnxruntime/tools/ort_format_model/ort_flatbuffers_py/fbs/Shape.py +78 -0
- onnxruntime/tools/ort_format_model/ort_flatbuffers_py/fbs/SparseTensor.py +114 -0
- onnxruntime/tools/ort_format_model/ort_flatbuffers_py/fbs/StringProperty.py +67 -0
- onnxruntime/tools/ort_format_model/ort_flatbuffers_py/fbs/StringStringEntry.py +67 -0
- onnxruntime/tools/ort_format_model/ort_flatbuffers_py/fbs/Tensor.py +203 -0
- onnxruntime/tools/ort_format_model/ort_flatbuffers_py/fbs/TensorDataType.py +26 -0
- onnxruntime/tools/ort_format_model/ort_flatbuffers_py/fbs/TensorTypeAndShape.py +71 -0
- onnxruntime/tools/ort_format_model/ort_flatbuffers_py/fbs/TypeInfo.py +83 -0
- onnxruntime/tools/ort_format_model/ort_flatbuffers_py/fbs/TypeInfoValue.py +9 -0
- onnxruntime/tools/ort_format_model/ort_flatbuffers_py/fbs/ValueInfo.py +84 -0
- onnxruntime/tools/ort_format_model/ort_flatbuffers_py/fbs/__init__.py +6 -0
- onnxruntime/tools/ort_format_model/ort_model_processor.py +86 -0
- onnxruntime/tools/ort_format_model/types.py +85 -0
- onnxruntime/tools/ort_format_model/utils.py +61 -0
- onnxruntime/tools/pytorch_export_contrib_ops.py +129 -0
- onnxruntime/tools/pytorch_export_helpers.py +131 -0
- onnxruntime/tools/qdq_helpers/__init__.py +0 -0
- onnxruntime/tools/qdq_helpers/optimize_qdq_model.py +37 -0
- onnxruntime/tools/qnn/add_trans_cast.py +292 -0
- onnxruntime/tools/qnn/gen_qnn_ctx_onnx_model.py +364 -0
- onnxruntime/tools/qnn/preprocess.py +165 -0
- onnxruntime/tools/reduced_build_config_parser.py +203 -0
- onnxruntime/tools/remove_initializer_from_input.py +37 -0
- onnxruntime/tools/symbolic_shape_infer.py +3094 -0
- onnxruntime/tools/update_onnx_opset.py +31 -0
- onnxruntime/transformers/__init__.py +8 -0
- onnxruntime/transformers/affinity_helper.py +40 -0
- onnxruntime/transformers/benchmark.py +942 -0
- onnxruntime/transformers/benchmark_helper.py +643 -0
- onnxruntime/transformers/bert_perf_test.py +629 -0
- onnxruntime/transformers/bert_test_data.py +641 -0
- onnxruntime/transformers/compare_bert_results.py +256 -0
- onnxruntime/transformers/constants.py +47 -0
- onnxruntime/transformers/convert_generation.py +3605 -0
- onnxruntime/transformers/convert_tf_models_to_pytorch.py +205 -0
- onnxruntime/transformers/convert_to_packing_mode.py +385 -0
- onnxruntime/transformers/dynamo_onnx_helper.py +205 -0
- onnxruntime/transformers/float16.py +501 -0
- onnxruntime/transformers/fusion_attention.py +1189 -0
- onnxruntime/transformers/fusion_attention_clip.py +340 -0
- onnxruntime/transformers/fusion_attention_sam2.py +533 -0
- onnxruntime/transformers/fusion_attention_unet.py +1307 -0
- onnxruntime/transformers/fusion_attention_vae.py +300 -0
- onnxruntime/transformers/fusion_bart_attention.py +435 -0
- onnxruntime/transformers/fusion_base.py +141 -0
- onnxruntime/transformers/fusion_bias_add.py +57 -0
- onnxruntime/transformers/fusion_biasgelu.py +66 -0
- onnxruntime/transformers/fusion_biassplitgelu.py +110 -0
- onnxruntime/transformers/fusion_conformer_attention.py +222 -0
- onnxruntime/transformers/fusion_constant_fold.py +144 -0
- onnxruntime/transformers/fusion_embedlayer.py +810 -0
- onnxruntime/transformers/fusion_fastgelu.py +492 -0
- onnxruntime/transformers/fusion_gelu.py +258 -0
- onnxruntime/transformers/fusion_gelu_approximation.py +25 -0
- onnxruntime/transformers/fusion_gemmfastgelu.py +121 -0
- onnxruntime/transformers/fusion_gpt_attention.py +546 -0
- onnxruntime/transformers/fusion_gpt_attention_megatron.py +355 -0
- onnxruntime/transformers/fusion_gpt_attention_no_past.py +260 -0
- onnxruntime/transformers/fusion_group_norm.py +180 -0
- onnxruntime/transformers/fusion_layernorm.py +489 -0
- onnxruntime/transformers/fusion_mha_mmdit.py +667 -0
- onnxruntime/transformers/fusion_nhwc_conv.py +99 -0
- onnxruntime/transformers/fusion_options.py +340 -0
- onnxruntime/transformers/fusion_qordered_attention.py +420 -0
- onnxruntime/transformers/fusion_qordered_gelu.py +118 -0
- onnxruntime/transformers/fusion_qordered_layernorm.py +122 -0
- onnxruntime/transformers/fusion_qordered_matmul.py +216 -0
- onnxruntime/transformers/fusion_quickgelu.py +74 -0
- onnxruntime/transformers/fusion_reshape.py +173 -0
- onnxruntime/transformers/fusion_rotary_attention.py +1591 -0
- onnxruntime/transformers/fusion_shape.py +109 -0
- onnxruntime/transformers/fusion_simplified_layernorm.py +165 -0
- onnxruntime/transformers/fusion_skip_group_norm.py +254 -0
- onnxruntime/transformers/fusion_skiplayernorm.py +209 -0
- onnxruntime/transformers/fusion_transpose.py +167 -0
- onnxruntime/transformers/fusion_utils.py +321 -0
- onnxruntime/transformers/huggingface_models.py +74 -0
- onnxruntime/transformers/import_utils.py +20 -0
- onnxruntime/transformers/io_binding_helper.py +487 -0
- onnxruntime/transformers/large_model_exporter.py +395 -0
- onnxruntime/transformers/machine_info.py +230 -0
- onnxruntime/transformers/metrics.py +163 -0
- onnxruntime/transformers/models/bart/__init__.py +12 -0
- onnxruntime/transformers/models/bart/export.py +98 -0
- onnxruntime/transformers/models/bert/__init__.py +12 -0
- onnxruntime/transformers/models/bert/eval_squad.py +329 -0
- onnxruntime/transformers/models/gpt2/__init__.py +12 -0
- onnxruntime/transformers/models/gpt2/benchmark_gpt2.py +413 -0
- onnxruntime/transformers/models/gpt2/convert_to_onnx.py +566 -0
- onnxruntime/transformers/models/gpt2/gpt2_helper.py +1031 -0
- onnxruntime/transformers/models/gpt2/gpt2_parity.py +513 -0
- onnxruntime/transformers/models/gpt2/gpt2_tester.py +501 -0
- onnxruntime/transformers/models/gpt2/parity_check_helper.py +146 -0
- onnxruntime/transformers/models/llama/__init__.py +12 -0
- onnxruntime/transformers/models/llama/benchmark.py +700 -0
- onnxruntime/transformers/models/llama/benchmark_all.py +488 -0
- onnxruntime/transformers/models/llama/benchmark_e2e.py +608 -0
- onnxruntime/transformers/models/llama/convert_to_onnx.py +1064 -0
- onnxruntime/transformers/models/llama/dist_settings.py +57 -0
- onnxruntime/transformers/models/llama/llama_inputs.py +504 -0
- onnxruntime/transformers/models/llama/llama_parity.py +343 -0
- onnxruntime/transformers/models/llama/llama_torch.py +47 -0
- onnxruntime/transformers/models/llama/quant_kv_dataloader.py +108 -0
- onnxruntime/transformers/models/longformer/__init__.py +12 -0
- onnxruntime/transformers/models/longformer/benchmark_longformer.py +821 -0
- onnxruntime/transformers/models/longformer/convert_to_onnx.py +413 -0
- onnxruntime/transformers/models/longformer/generate_test_data.py +347 -0
- onnxruntime/transformers/models/longformer/longformer_helper.py +76 -0
- onnxruntime/transformers/models/phi2/__init__.py +12 -0
- onnxruntime/transformers/models/phi2/convert_to_onnx.py +590 -0
- onnxruntime/transformers/models/phi2/inference_example.py +414 -0
- onnxruntime/transformers/models/sam2/__init__.py +12 -0
- onnxruntime/transformers/models/sam2/benchmark_sam2.py +638 -0
- onnxruntime/transformers/models/sam2/convert_to_onnx.py +270 -0
- onnxruntime/transformers/models/sam2/image_decoder.py +272 -0
- onnxruntime/transformers/models/sam2/image_encoder.py +236 -0
- onnxruntime/transformers/models/sam2/mask_decoder.py +208 -0
- onnxruntime/transformers/models/sam2/nvtx_helper.py +33 -0
- onnxruntime/transformers/models/sam2/prompt_encoder.py +189 -0
- onnxruntime/transformers/models/sam2/sam2_demo.py +321 -0
- onnxruntime/transformers/models/sam2/sam2_image_onnx_predictor.py +279 -0
- onnxruntime/transformers/models/sam2/sam2_utils.py +147 -0
- onnxruntime/transformers/models/stable_diffusion/__init__.py +12 -0
- onnxruntime/transformers/models/stable_diffusion/benchmark.py +1519 -0
- onnxruntime/transformers/models/stable_diffusion/benchmark_controlnet.py +426 -0
- onnxruntime/transformers/models/stable_diffusion/demo_txt2img.py +103 -0
- onnxruntime/transformers/models/stable_diffusion/demo_txt2img_xl.py +269 -0
- onnxruntime/transformers/models/stable_diffusion/demo_utils.py +778 -0
- onnxruntime/transformers/models/stable_diffusion/diffusion_models.py +1318 -0
- onnxruntime/transformers/models/stable_diffusion/diffusion_schedulers.py +1179 -0
- onnxruntime/transformers/models/stable_diffusion/engine_builder.py +295 -0
- onnxruntime/transformers/models/stable_diffusion/engine_builder_ort_cuda.py +387 -0
- onnxruntime/transformers/models/stable_diffusion/engine_builder_ort_trt.py +288 -0
- onnxruntime/transformers/models/stable_diffusion/engine_builder_tensorrt.py +395 -0
- onnxruntime/transformers/models/stable_diffusion/engine_builder_torch.py +108 -0
- onnxruntime/transformers/models/stable_diffusion/optimize_pipeline.py +590 -0
- onnxruntime/transformers/models/stable_diffusion/ort_optimizer.py +136 -0
- onnxruntime/transformers/models/stable_diffusion/pipeline_stable_diffusion.py +831 -0
- onnxruntime/transformers/models/stable_diffusion/trt_utilities.py +12 -0
- onnxruntime/transformers/models/t5/__init__.py +12 -0
- onnxruntime/transformers/models/t5/convert_to_onnx.py +318 -0
- onnxruntime/transformers/models/t5/t5_decoder.py +437 -0
- onnxruntime/transformers/models/t5/t5_encoder.py +70 -0
- onnxruntime/transformers/models/t5/t5_encoder_decoder_init.py +361 -0
- onnxruntime/transformers/models/t5/t5_helper.py +302 -0
- onnxruntime/transformers/models/whisper/__init__.py +12 -0
- onnxruntime/transformers/models/whisper/benchmark.py +585 -0
- onnxruntime/transformers/models/whisper/benchmark_all.py +526 -0
- onnxruntime/transformers/models/whisper/convert_to_onnx.py +609 -0
- onnxruntime/transformers/models/whisper/whisper_chain.py +334 -0
- onnxruntime/transformers/models/whisper/whisper_decoder.py +464 -0
- onnxruntime/transformers/models/whisper/whisper_encoder.py +164 -0
- onnxruntime/transformers/models/whisper/whisper_encoder_decoder_init.py +371 -0
- onnxruntime/transformers/models/whisper/whisper_helper.py +1035 -0
- onnxruntime/transformers/models/whisper/whisper_inputs.py +380 -0
- onnxruntime/transformers/models/whisper/whisper_jump_times.py +477 -0
- onnxruntime/transformers/onnx_exporter.py +719 -0
- onnxruntime/transformers/onnx_model.py +1636 -0
- onnxruntime/transformers/onnx_model_bart.py +141 -0
- onnxruntime/transformers/onnx_model_bert.py +488 -0
- onnxruntime/transformers/onnx_model_bert_keras.py +474 -0
- onnxruntime/transformers/onnx_model_bert_tf.py +588 -0
- onnxruntime/transformers/onnx_model_clip.py +42 -0
- onnxruntime/transformers/onnx_model_conformer.py +32 -0
- onnxruntime/transformers/onnx_model_gpt2.py +101 -0
- onnxruntime/transformers/onnx_model_mmdit.py +112 -0
- onnxruntime/transformers/onnx_model_phi.py +929 -0
- onnxruntime/transformers/onnx_model_sam2.py +137 -0
- onnxruntime/transformers/onnx_model_t5.py +985 -0
- onnxruntime/transformers/onnx_model_tnlr.py +226 -0
- onnxruntime/transformers/onnx_model_unet.py +258 -0
- onnxruntime/transformers/onnx_model_vae.py +42 -0
- onnxruntime/transformers/onnx_utils.py +55 -0
- onnxruntime/transformers/optimizer.py +620 -0
- onnxruntime/transformers/past_helper.py +149 -0
- onnxruntime/transformers/profile_result_processor.py +358 -0
- onnxruntime/transformers/profiler.py +434 -0
- onnxruntime/transformers/quantize_helper.py +76 -0
- onnxruntime/transformers/shape_infer_helper.py +121 -0
- onnxruntime/transformers/shape_optimizer.py +400 -0
- onnxruntime/transformers/torch_onnx_export_helper.py +74 -0
- onnxruntime_directml-1.24.1.dist-info/METADATA +216 -0
- onnxruntime_directml-1.24.1.dist-info/RECORD +322 -0
- onnxruntime_directml-1.24.1.dist-info/WHEEL +5 -0
- onnxruntime_directml-1.24.1.dist-info/entry_points.txt +2 -0
- onnxruntime_directml-1.24.1.dist-info/top_level.txt +1 -0
|
@@ -0,0 +1,322 @@
|
|
|
1
|
+
onnxruntime/LICENSE,sha256=wlDWJ48LR6ZDn7dZKwi1ilXrn1NapJodtjIRw_mCtnQ,1094
|
|
2
|
+
onnxruntime/Privacy.md,sha256=v7dxKwdfPwfj6-5dwqKW0d4y2_ca0oZj9z0VOMtsOwg,2490
|
|
3
|
+
onnxruntime/ThirdPartyNotices.txt,sha256=-wr3dLTXz_xbnQRvKq6t4vN98vgKv4AzyV3__Md6iGY,331175
|
|
4
|
+
onnxruntime/__init__.py,sha256=4m-eN4dxafHphXgWLRArkH_TvwmMfqvHUhibMspWiBg,17693
|
|
5
|
+
onnxruntime/backend/__init__.py,sha256=5I1Ylsawf9w6MNmK4RiN1wA-EEQqlKKwYTNZB-m_k6M,334
|
|
6
|
+
onnxruntime/backend/backend.py,sha256=xPA69Lf7rwwwgeWoZ3CgB2JSoExuIAdhHmd6ROp19sc,8187
|
|
7
|
+
onnxruntime/backend/backend_rep.py,sha256=A7S4GqxLC6IfkbEXLlWiWpCD9AJ5x-xAhnR8BCM2cNk,1776
|
|
8
|
+
onnxruntime/capi/DirectML.dll,sha256=gbAgac6Yki97R_HA3Ch67twSI76_xoxNsg9CDlgVNj4,18527776
|
|
9
|
+
onnxruntime/capi/__init__.py,sha256=uRp4pMtfoayBhZgEsiFqFCD13Y6LUo82FdZsQX8X8LI,251
|
|
10
|
+
onnxruntime/capi/_ld_preload.py,sha256=li6cbZ64hDfUndat4mprUWzowLa3RQdw0q2E56sXFwE,413
|
|
11
|
+
onnxruntime/capi/_pybind_state.py,sha256=nbUpnUncwBv5pgJA8yugDYJRA4TTfC0gaYOED5jD-SA,1533
|
|
12
|
+
onnxruntime/capi/build_and_package_info.py,sha256=0--YOikLSN6lL1MJYVdzG4ap3ie-ZYixf9ZZV551CWw,63
|
|
13
|
+
onnxruntime/capi/convert_npz_to_onnx_adapter.py,sha256=N0ShYr30vBQcOr9KyFd4AUdEcqWW89KVd80qSYCgdQ4,1581
|
|
14
|
+
onnxruntime/capi/onnxruntime.dll,sha256=nOYI6LS7BIIpzFRHTmcO8uRQJOsP64VmnNROqFEtO9g,21038624
|
|
15
|
+
onnxruntime/capi/onnxruntime_collect_build_info.py,sha256=sD8Z2S15QHSvuO1j7tgqJKeORDUatwzUmzvC8Uj9cAM,2109
|
|
16
|
+
onnxruntime/capi/onnxruntime_inference_collection.py,sha256=ZX2LwB_CkOy0peurgUkxQjr727Wd5jtQI8YJ0jrvjwA,62504
|
|
17
|
+
onnxruntime/capi/onnxruntime_providers_shared.dll,sha256=B4hV-DB6kpwuKcIaQ2FiyyWo9YjANJz1vDqmKV_z1m4,21576
|
|
18
|
+
onnxruntime/capi/onnxruntime_pybind11_state.pyd,sha256=AFpnAwxn8QHcIl6bv6OWl9SQ87_W8PbvI4l8l8Ul_X0,26236952
|
|
19
|
+
onnxruntime/capi/onnxruntime_validation.py,sha256=p9AveXPu78B-Fd4wOtBSisvW3lauMmQf9CxJirttfk8,6949
|
|
20
|
+
onnxruntime/capi/version_info.py,sha256=8mm1VTXF8xgx6N8vFNe0Tiik9qdg9Vvi9f32bPE9ktw,34
|
|
21
|
+
onnxruntime/datasets/__init__.py,sha256=DqRdpMfRtDfhVkCQu5lTmfSQ-GG4dETHNWdoB4fA7lU,473
|
|
22
|
+
onnxruntime/datasets/logreg_iris.onnx,sha256=giR4TJjXNBLZ_ZmrzVejhWi9WQmA0PvlkWRkUxxS6Pw,670
|
|
23
|
+
onnxruntime/datasets/mul_1.onnx,sha256=cfQxxOkyHsb76xWNAu0kBFmn3MmGc_p5pPQ5zkLvrxA,130
|
|
24
|
+
onnxruntime/datasets/sigmoid.onnx,sha256=U0Crpnp-NHUWKteUN4r1XxcY9V-aXXS0r2Dsx_emJLY,103
|
|
25
|
+
onnxruntime/quantization/__init__.py,sha256=gL1o5I5h988WNCfzJ1KMpSAvk2xjgw0HcqYsfXcKurw,647
|
|
26
|
+
onnxruntime/quantization/base_quantizer.py,sha256=Ur85CgEWLdi2AZ2OcUOQkH2xEWJ8wa1bBu20zX1WElY,26179
|
|
27
|
+
onnxruntime/quantization/calibrate.py,sha256=YByAhxDFOPmLPZs64tekzowIVxWPSDoTLMxXY7u8b1E,54638
|
|
28
|
+
onnxruntime/quantization/matmul_bnb4_quantizer.py,sha256=aXTPtK3uQ80YTeSnf_bvBITqcoTnvxNCzJltA4ZS8iM,9263
|
|
29
|
+
onnxruntime/quantization/matmul_nbits_quantizer.py,sha256=jJiYRt1qvS1_KMjwirVtWfwhV1nsRDWUFoYv4GNHiIs,69088
|
|
30
|
+
onnxruntime/quantization/onnx_model.py,sha256=Eabr8tVPnaYCZnsqW3d8Fd0Hm28OZbqsoYZHNPqSEFY,24528
|
|
31
|
+
onnxruntime/quantization/onnx_quantizer.py,sha256=7cK-89RJ016NDCMtOc7i7WdXWewH_te166xjJ-3_DBM,50385
|
|
32
|
+
onnxruntime/quantization/preprocess.py,sha256=VU4iX7g8gOgVH0zehOcOXsVWkZpx6kG_LFlwGM3Bs6c,5045
|
|
33
|
+
onnxruntime/quantization/qdq_loss_debug.py,sha256=1L-BuA1Jck9J5lgRoj7QPEbtaKHMkoQLMTD4iYFOaxo,15829
|
|
34
|
+
onnxruntime/quantization/qdq_quantizer.py,sha256=2Y_xa1DJPoV-XpVN-O41TX5es6cJsOkj-Dnrhg_FS4s,71354
|
|
35
|
+
onnxruntime/quantization/quant_utils.py,sha256=vutak1PcDXY8UpYiiGf3i3scvEzw1EX4aRNVpiUFHGY,41408
|
|
36
|
+
onnxruntime/quantization/quantize.py,sha256=KLsgW2OKThDkXMg4EbX9SUF2YqpQmt90IZM1YhH50fw,53809
|
|
37
|
+
onnxruntime/quantization/registry.py,sha256=8C-Z05D4LAZI27EyMsRiGrxO6bptyFLuJbRJbqEhYLI,3856
|
|
38
|
+
onnxruntime/quantization/shape_inference.py,sha256=tVxgnM1DxuNdsCxaz3pajPbKz3F8S71eprI-guO0_RU,9892
|
|
39
|
+
onnxruntime/quantization/static_quantize_runner.py,sha256=0iYhx49fXs0F38wsqSa_hyyAP1a_ZZxqIhEEM7FakU0,11341
|
|
40
|
+
onnxruntime/quantization/tensor_quant_overrides.py,sha256=Qz2D0NAI72xLA5aXqDfZgBXMYpU5jh5HnCfe-RnLyk0,21304
|
|
41
|
+
onnxruntime/quantization/CalTableFlatBuffers/KeyValue.py,sha256=e-jJFhw9fb775fDCLnWdbRSdoJ6vGD0c7qTnkIG-vNs,2250
|
|
42
|
+
onnxruntime/quantization/CalTableFlatBuffers/TrtTable.py,sha256=vrueLPy2RV5R3MSdIQkqW7TxDSSKPMoOixSuzIsv2oE,2682
|
|
43
|
+
onnxruntime/quantization/CalTableFlatBuffers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
|
44
|
+
onnxruntime/quantization/execution_providers/qnn/__init__.py,sha256=nKKB7VEbO574HDL2xdJPD8VeXoK2a3jd8nLBxULiVvI,120
|
|
45
|
+
onnxruntime/quantization/execution_providers/qnn/fusion_lpnorm.py,sha256=vUrbMNorHH7_uKjeL1jlkPghnplPIDPz0kmN0Tt03mc,5327
|
|
46
|
+
onnxruntime/quantization/execution_providers/qnn/fusion_spacetodepth.py,sha256=u_OyHhp519VoBhUVZTAVkey501E91U0EYMkCUmNwMTE,6285
|
|
47
|
+
onnxruntime/quantization/execution_providers/qnn/mixed_precision_overrides_utils.py,sha256=PTdZPaP1cZAUXu14e4L0j71wO53lcD1s8yPkZlan5G0,18995
|
|
48
|
+
onnxruntime/quantization/execution_providers/qnn/preprocess.py,sha256=m6hj1J-1E8kkXbWJzQgqk0zpWfGPtSkN_jUYPYvPWIE,16736
|
|
49
|
+
onnxruntime/quantization/execution_providers/qnn/quant_config.py,sha256=nspz3rKtOfgRLObEJRLPLVKTeWt8XaX5IYgN7xbM8KE,18986
|
|
50
|
+
onnxruntime/quantization/fusions/__init__.py,sha256=QDG2mzyaAHnVmHkOBpc4Mdwn5NcNrattW5yTavMWs8M,246
|
|
51
|
+
onnxruntime/quantization/fusions/fusion.py,sha256=A6_77l5uw-hIwyoX7DPOFL6O-y3qXk-S16SMLv1Ncis,12088
|
|
52
|
+
onnxruntime/quantization/fusions/fusion_gelu.py,sha256=3qOO4U95ATD6S14dyC-5-vGeaQBr5U-GCsjfvHqoL98,10647
|
|
53
|
+
onnxruntime/quantization/fusions/fusion_layernorm.py,sha256=IPV4wPl0fsK1V02_Lg0ocsuEgpAlrJqxGZfFiTwpjc4,6146
|
|
54
|
+
onnxruntime/quantization/fusions/replace_upsample_with_resize.py,sha256=Ewuca08Yu_6XQMock17J20oRIaoHwLzONx3Nd0bGS_A,3427
|
|
55
|
+
onnxruntime/quantization/neural_compressor/__init__.py,sha256=4SEilQJJDoKxMpoontHVtxLctL4kzZD50jiHylxeMWQ,68
|
|
56
|
+
onnxruntime/quantization/neural_compressor/onnx_model.py,sha256=iNsKGuRhkFo1xTed2sunrbAM4I-ZMiLWRcU7yIeRWi0,51478
|
|
57
|
+
onnxruntime/quantization/neural_compressor/util.py,sha256=4AbSArXCHxT8MTPhRn3rp-eysd9aju5ciha1CDMW2D4,2762
|
|
58
|
+
onnxruntime/quantization/neural_compressor/weight_only.py,sha256=I623O0GdFuKwgQUvh4txNWPIbvZ5HYjMSDzdOv83uE4,37227
|
|
59
|
+
onnxruntime/quantization/operators/__init__.py,sha256=IfKXrFWtRSye1mkgD9lpwxio0fw9cVr_1CdV1cvefig,85
|
|
60
|
+
onnxruntime/quantization/operators/activation.py,sha256=G6XAKZaRIV3vrtjpK4kEvbzUSlLOwwEauXX9exgIHZg,4545
|
|
61
|
+
onnxruntime/quantization/operators/argmax.py,sha256=pfE9_eSTZ2otTkUcWwlLi7HJKtN10kE5c2Lz0SeVADQ,589
|
|
62
|
+
onnxruntime/quantization/operators/attention.py,sha256=eH7-Z3MfP6xRZCdhDAyNxWG2s2nZILxIEFVAHtqj7EQ,2637
|
|
63
|
+
onnxruntime/quantization/operators/base_operator.py,sha256=vrAVfKJXZvF7ZherKw4JUGonNyNuoU2TWnwBy-EQ3QE,1118
|
|
64
|
+
onnxruntime/quantization/operators/binary_op.py,sha256=pEQHRAS75EMp7LG6jzWV7gDQt_vzEPLJEI00eIOuoiA,2544
|
|
65
|
+
onnxruntime/quantization/operators/concat.py,sha256=fZFwnaqoOZ9b0ZvGpBK_MrJzVteeJguWRQ396kUh8QQ,2143
|
|
66
|
+
onnxruntime/quantization/operators/conv.py,sha256=whgjZwx8iWCUDhiGoMfZ8ADzj9lMzIamYrY2Ps7l8oU,10203
|
|
67
|
+
onnxruntime/quantization/operators/direct_q8.py,sha256=xWWLXoO1fJ0FXOFs0OdX5pvLVw3m7b6TBpU1YRYycO0,3389
|
|
68
|
+
onnxruntime/quantization/operators/embed_layernorm.py,sha256=2LsZk5Um0ELaRESWjScgYyQioJelRZK6oQbzAclSgXI,4058
|
|
69
|
+
onnxruntime/quantization/operators/gather.py,sha256=HL79Csv--zTAs5XmG1xP7Fjw72RTlqpadxB6w-OmqN8,2230
|
|
70
|
+
onnxruntime/quantization/operators/gavgpool.py,sha256=wYyjEf3h-_QChWKnsZ2N-haBG1RSvqRitZ-Yvfwo9Dk,2445
|
|
71
|
+
onnxruntime/quantization/operators/gemm.py,sha256=SaqWitL-QotPO9KS6A4iZ5H_OUJ7dC_ZQhr5x6epGDQ,6226
|
|
72
|
+
onnxruntime/quantization/operators/lstm.py,sha256=gO3AqC3tvoiVlGRDKOLrBRPRyEfnC2EfazcMLOZ4AkE,5238
|
|
73
|
+
onnxruntime/quantization/operators/matmul.py,sha256=Z98R4saKkbQhps_jRCCJ91cmptxc8y7b5TmG3gvHIcw,8499
|
|
74
|
+
onnxruntime/quantization/operators/maxpool.py,sha256=QyDmHyBo0QKf6kNFbp2a9v6ThrBO-OL3tW0PFdN6bkI,961
|
|
75
|
+
onnxruntime/quantization/operators/norm.py,sha256=f3fUiSN4WdM8iuXCSjETorRyDFUSJJWxO10vegj_dEs,1649
|
|
76
|
+
onnxruntime/quantization/operators/pad.py,sha256=yM83jrZC88p5Fa53w3KLQc7zIlSz6Gv1iSOPvq5guM4,7951
|
|
77
|
+
onnxruntime/quantization/operators/pooling.py,sha256=L0IT7G6-2XSx9-wUz5BX59Mc43FfJEg79NwW3yqEDhI,2285
|
|
78
|
+
onnxruntime/quantization/operators/qdq_base_operator.py,sha256=Fco9JZxrXQoVgjKvmHFuzT0mogWo9-wHiDa51CjTioo,823
|
|
79
|
+
onnxruntime/quantization/operators/resize.py,sha256=BMeym-7GHOSnGpZisa9BkdQkVmCXwKANA5NpnKRnaLI,962
|
|
80
|
+
onnxruntime/quantization/operators/softmax.py,sha256=e3ThVOh2TG1B8luG6xWkoT_hCdvsMRjvTlTje8CW-YQ,2714
|
|
81
|
+
onnxruntime/quantization/operators/split.py,sha256=82R65-_Rw5g23f0uekUWpA3nhOzeUWdOQgr2JZXwrOc,2258
|
|
82
|
+
onnxruntime/quantization/operators/where.py,sha256=wd6PQ7LlbrJTqamFMch_Fipnbt4IewMJSAPozMTrwKI,3127
|
|
83
|
+
onnxruntime/tools/__init__.py,sha256=7up7iKcklVy6UcpIIIIlBaK690O32vaOxyaaTWvwyxU,528
|
|
84
|
+
onnxruntime/tools/check_onnx_model_mobile_usability.py,sha256=hzqjrI9Xz7LP1idPdviRU7EEeLMvUB0qlyTqHofxlFs,1717
|
|
85
|
+
onnxruntime/tools/convert_onnx_models_to_ort.py,sha256=FJIv_geqXwObnTEH5T90NM7_NhVfxjf_jXKrDzlJoWY,16939
|
|
86
|
+
onnxruntime/tools/file_utils.py,sha256=HrNK4UjpR-u43tLBC2vwKieisCVRBSFnaIDKaswwEGI,1572
|
|
87
|
+
onnxruntime/tools/logger.py,sha256=s3M5-Akb69zubXNhCpsjIoJ052gYieHV5FsOfBZ6lrI,333
|
|
88
|
+
onnxruntime/tools/make_dynamic_shape_fixed.py,sha256=28zdPjT8yz8AEv5hkFl_YNI8qQzbmP64hl-GqXO-Or0,2642
|
|
89
|
+
onnxruntime/tools/offline_tuning.py,sha256=IswzPfMIbWwZKBBX4pat8zu4s1D5sUzVhLjDmWzQlkI,6368
|
|
90
|
+
onnxruntime/tools/onnx_model_utils.py,sha256=G9Wsgfiq46pgbkYTZAYupZKnqgZI-GT0RverFPn2sC4,16818
|
|
91
|
+
onnxruntime/tools/onnx_randomizer.py,sha256=9L96dzIf59cQ2oQsmR2EEsdrR4hHwEGrpZkajEgUPAY,3361
|
|
92
|
+
onnxruntime/tools/onnxruntime_test.py,sha256=SvqgwrjiIpf_vsZfHmkE_FPXJkDA18mZpwYoyjMv5g0,5770
|
|
93
|
+
onnxruntime/tools/optimize_onnx_model.py,sha256=DUm9R3qC6g1ZnXRdc7Mu7U03ArJF5j5oICJLh_amhXI,2015
|
|
94
|
+
onnxruntime/tools/pytorch_export_contrib_ops.py,sha256=DypErhw_nTIDpINn2omBl4Wb-tmvLzEtUjkl_Krgo3I,4909
|
|
95
|
+
onnxruntime/tools/pytorch_export_helpers.py,sha256=MRegHn3z3VhVbZQ4O-kTGedIE-pufyxhq1A1GVIdCjY,5971
|
|
96
|
+
onnxruntime/tools/reduced_build_config_parser.py,sha256=F9bN5_yi5V7nLQUS2vsfo7gbYB1cZnSAQiUTZOdvqM8,10161
|
|
97
|
+
onnxruntime/tools/remove_initializer_from_input.py,sha256=C1vDZZBek-TK6SV1q6e_7kqkyQzS8iqHE8GzZgehhiE,1031
|
|
98
|
+
onnxruntime/tools/symbolic_shape_infer.py,sha256=Ya-aBHWAzZ5AHh75UE9mhhp3_3HiONkdPExSeWtu3XY,145673
|
|
99
|
+
onnxruntime/tools/update_onnx_opset.py,sha256=fplb1ypV-pFhu8Xsi5u_bDfI7EsC4zamJkTziccgQ2c,1182
|
|
100
|
+
onnxruntime/tools/mobile_helpers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
|
101
|
+
onnxruntime/tools/mobile_helpers/coreml_supported_mlprogram_ops.md,sha256=PnuZOiccxNgBY4umDVi_gFWi9tlFz7e7XsSirXpSTx8,2456
|
|
102
|
+
onnxruntime/tools/mobile_helpers/coreml_supported_neuralnetwork_ops.md,sha256=JJBwv04C_o2K1UWOttDtarnC7xFFZQ2uvvIZg4EFmWA,1958
|
|
103
|
+
onnxruntime/tools/mobile_helpers/nnapi_supported_ops.md,sha256=uJznEyy7ZAdlrkKQeoWFFs55rPE-kOePIJiv741r98Q,2385
|
|
104
|
+
onnxruntime/tools/mobile_helpers/usability_checker.py,sha256=sEaT80-YwNwUxQbev3R2RzMzaaGR06slGdJuUpuSdRc,32378
|
|
105
|
+
onnxruntime/tools/ort_format_model/__init__.py,sha256=Xsq0LM-zEkNsULbTTvyxnz4dDzqepeE8BEgHzJGYOD8,1307
|
|
106
|
+
onnxruntime/tools/ort_format_model/operator_type_usage_processors.py,sha256=rl9Ub_QRC5B0F9Mu0LbY86bsVnkUf7XYn1PtwC1NPNM,27032
|
|
107
|
+
onnxruntime/tools/ort_format_model/ort_model_processor.py,sha256=Zh07RmcGYf0-qbXRiim2DWUGHbsrEeG6_rOcx3VLexw,4472
|
|
108
|
+
onnxruntime/tools/ort_format_model/types.py,sha256=8r38Y7ghaGn6_4muEWSpeDHkY3rsIwAB76t5YD8uMFk,4468
|
|
109
|
+
onnxruntime/tools/ort_format_model/utils.py,sha256=rURCBd9_214ScVY9dkHF3xWbmh3hAGbeDPHXJVlMmmk,2595
|
|
110
|
+
onnxruntime/tools/ort_format_model/ort_flatbuffers_py/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
|
111
|
+
onnxruntime/tools/ort_format_model/ort_flatbuffers_py/fbs/ArgType.py,sha256=ErRXrmwza1xgVW4OAENw_B8yVc7WfWbAYYhiAO3Cc2g,147
|
|
112
|
+
onnxruntime/tools/ort_format_model/ort_flatbuffers_py/fbs/ArgTypeAndIndex.py,sha256=8Sx0mmUqjN4uN7MeTY6scIgeZO6gruARh8MvlEQdUdQ,2093
|
|
113
|
+
onnxruntime/tools/ort_format_model/ort_flatbuffers_py/fbs/Attribute.py,sha256=S5SA5y9FB2HWcPIPx99jz_ebTV85O8cTETJgGe_FKLo,11187
|
|
114
|
+
onnxruntime/tools/ort_format_model/ort_flatbuffers_py/fbs/AttributeType.py,sha256=piF8u_U7zNE5eSrwcQhRLgvr2axK21-455McxIje2HQ,346
|
|
115
|
+
onnxruntime/tools/ort_format_model/ort_flatbuffers_py/fbs/Checkpoint.py,sha256=0cZ14luQy_w8GlbDF_xoGPWtee2UueQM4jt59YX5B9o,4342
|
|
116
|
+
onnxruntime/tools/ort_format_model/ort_flatbuffers_py/fbs/DeprecatedKernelCreateInfos.py,sha256=K6Sl9lPxPE8huGE0tunG9MXDFYbzJs5Hc4nzKj8ft88,4648
|
|
117
|
+
onnxruntime/tools/ort_format_model/ort_flatbuffers_py/fbs/DeprecatedNodeIndexAndKernelDefHash.py,sha256=Lr3cs1e80M1zLbZ1hBKwDsSl6d8zxjkbIfAguiTkJko,2526
|
|
118
|
+
onnxruntime/tools/ort_format_model/ort_flatbuffers_py/fbs/DeprecatedSessionState.py,sha256=N5CN72wJObCmYGcbA5z9o3u1khYxJ9bVSIaFIJCDR3E,3678
|
|
119
|
+
onnxruntime/tools/ort_format_model/ort_flatbuffers_py/fbs/DeprecatedSubGraphSessionState.py,sha256=ARjFKPJX_pB2KTrSUCHz_iN_UmrB3xRDKtUbIepTX-Q,2682
|
|
120
|
+
onnxruntime/tools/ort_format_model/ort_flatbuffers_py/fbs/Dimension.py,sha256=VHQcCV5ip0BIhm1EFtsAXA-uMp8tqlu1BfiJP2LZuQc,2262
|
|
121
|
+
onnxruntime/tools/ort_format_model/ort_flatbuffers_py/fbs/DimensionValue.py,sha256=14W1geQj5_V_5UiqgrG2oFGDgCkAIS5d71xemZwmO_Y,2574
|
|
122
|
+
onnxruntime/tools/ort_format_model/ort_flatbuffers_py/fbs/DimensionValueType.py,sha256=2kmuPePZe_rWSMkI7ceIhewHL3wvvsDEhzH-LBusm0M,174
|
|
123
|
+
onnxruntime/tools/ort_format_model/ort_flatbuffers_py/fbs/EdgeEnd.py,sha256=aNdY1SA8fDyvH1Vsqo5xklVuhGtw2sUmyy86HmtWEbk,1137
|
|
124
|
+
onnxruntime/tools/ort_format_model/ort_flatbuffers_py/fbs/FloatProperty.py,sha256=qqpRpfy9QXs7WHhHUARUjv5q_zs8wiT2-iOjKD9v4NQ,2075
|
|
125
|
+
onnxruntime/tools/ort_format_model/ort_flatbuffers_py/fbs/Graph.py,sha256=Y_AnjssgrJ7o4oU1w-KG-rOl8EE8YuEnvDWu7rGN57E,11039
|
|
126
|
+
onnxruntime/tools/ort_format_model/ort_flatbuffers_py/fbs/InferenceSession.py,sha256=0FgbOyXfWicKshG131GzHqPTFJvoVorBxTVP1xM6ZNU,3125
|
|
127
|
+
onnxruntime/tools/ort_format_model/ort_flatbuffers_py/fbs/IntProperty.py,sha256=EjWio-yFwfkE_N764bYTaDRz0EHSZ1n0JePkLSNGitk,2037
|
|
128
|
+
onnxruntime/tools/ort_format_model/ort_flatbuffers_py/fbs/KernelTypeStrArgsEntry.py,sha256=lKy6VMEjvkiv5BGOjugWEZtlobNMwvlfcSRSYzjCiAk,3193
|
|
129
|
+
onnxruntime/tools/ort_format_model/ort_flatbuffers_py/fbs/KernelTypeStrResolver.py,sha256=cufEs5uTENEF6bvZQemzDztbErNtvDuzMpDoCnSBpik,2867
|
|
130
|
+
onnxruntime/tools/ort_format_model/ort_flatbuffers_py/fbs/MapType.py,sha256=wlC76nL8dC0vdZVlEoUC9qwgbVDwnSt5RCRO5r_Wfmg,2194
|
|
131
|
+
onnxruntime/tools/ort_format_model/ort_flatbuffers_py/fbs/Model.py,sha256=CPbe2q2OcKU8LbShHGtnIG_jrW83f3nvMR5Trg1Nx90,7663
|
|
132
|
+
onnxruntime/tools/ort_format_model/ort_flatbuffers_py/fbs/ModuleState.py,sha256=rGMagBBqyCNbewI-xUygxN07m7fLDQmH2EWISmFd7ss,4994
|
|
133
|
+
onnxruntime/tools/ort_format_model/ort_flatbuffers_py/fbs/Node.py,sha256=IjM4YjGq_swHTiLw01xsSWmAYaZmkVZUR3cei3byY68,10718
|
|
134
|
+
onnxruntime/tools/ort_format_model/ort_flatbuffers_py/fbs/NodeEdge.py,sha256=LiB9vyQTsFkfRRTuiDcwrRfVaWrRzAALP5Rop0rtGKg,4183
|
|
135
|
+
onnxruntime/tools/ort_format_model/ort_flatbuffers_py/fbs/NodeType.py,sha256=bBLbvDWziLLRxhjCAqlS44k_hdIWbWk3vsRxCDOLz7k,151
|
|
136
|
+
onnxruntime/tools/ort_format_model/ort_flatbuffers_py/fbs/NodesToOptimizeIndices.py,sha256=KoWmACgRnK8qGNyXiKo6NywUvzpvYYxGQ0AyayjyaFQ,6144
|
|
137
|
+
onnxruntime/tools/ort_format_model/ort_flatbuffers_py/fbs/OpIdKernelTypeStrArgsEntry.py,sha256=6WV2Pj3vkqJ2NZ19VWrPAxKtoGCeiJ6NTVOvs89BIQ4,3387
|
|
138
|
+
onnxruntime/tools/ort_format_model/ort_flatbuffers_py/fbs/OperatorSetId.py,sha256=UOiU-CCvexY3OOWk2hiZyBI0OItcA6jiZTsERDZxu2U,2099
|
|
139
|
+
onnxruntime/tools/ort_format_model/ort_flatbuffers_py/fbs/OptimizerGroup.py,sha256=Xri5OFTQ881fzaVdQZjVviLyE5vH2aFGGArQE5H1ZbU,4135
|
|
140
|
+
onnxruntime/tools/ort_format_model/ort_flatbuffers_py/fbs/ParameterOptimizerState.py,sha256=AIEaCRcEln5wC7X3C8vp7TmmpxuaH7t93UCphxbxRJA,3218
|
|
141
|
+
onnxruntime/tools/ort_format_model/ort_flatbuffers_py/fbs/PropertyBag.py,sha256=QXs064QNug8pG9zjX9aWEPmO63CIWJcL-kMLgMPk7vI,5099
|
|
142
|
+
onnxruntime/tools/ort_format_model/ort_flatbuffers_py/fbs/RuntimeOptimizationRecord.py,sha256=Lrl5lCzbUL323Y1nnij-7vn95d3qNHkiVPwkbg6SSds,4067
|
|
143
|
+
onnxruntime/tools/ort_format_model/ort_flatbuffers_py/fbs/RuntimeOptimizationRecordContainerEntry.py,sha256=PM7_R0kbnF3K32qoLOIPXmeudQfj0pvabMQLB-FjREw,3832
|
|
144
|
+
onnxruntime/tools/ort_format_model/ort_flatbuffers_py/fbs/RuntimeOptimizations.py,sha256=qYkC65wMqu7FinW2X9pPTeRD4bY7xdIsOfXnIJoFDFU,2800
|
|
145
|
+
onnxruntime/tools/ort_format_model/ort_flatbuffers_py/fbs/SequenceType.py,sha256=iiMOzyjzXzjO3zoad_oG2-ct786Wv8u4LWoFOZdmc0A,1829
|
|
146
|
+
onnxruntime/tools/ort_format_model/ort_flatbuffers_py/fbs/Shape.py,sha256=4Y6yU35JHSXKzb2BDzObkss4yiopwS3IXkSBpFf88FI,2352
|
|
147
|
+
onnxruntime/tools/ort_format_model/ort_flatbuffers_py/fbs/SparseTensor.py,sha256=3EZahzlyO_73WOzK9jZHwoVqsQkq3407ATBOG56KLEE,3806
|
|
148
|
+
onnxruntime/tools/ort_format_model/ort_flatbuffers_py/fbs/StringProperty.py,sha256=Zd5QCcSgvf5ojuOBRV19gGyQPhqjJkoBaMdZ06kOc7c,2110
|
|
149
|
+
onnxruntime/tools/ort_format_model/ort_flatbuffers_py/fbs/StringStringEntry.py,sha256=Iih7xRq4zvu4cVV4_TAC4sbjaH-dLqZGU68X0oLaWoI,2147
|
|
150
|
+
onnxruntime/tools/ort_format_model/ort_flatbuffers_py/fbs/Tensor.py,sha256=CZItxbQ-B9cJnaS5HwBNWT1buat_JgX-W-zOn9m11a4,6802
|
|
151
|
+
onnxruntime/tools/ort_format_model/ort_flatbuffers_py/fbs/TensorDataType.py,sha256=WmblM8GoLES62L12Ea3dC655_ruO2T3R2Q085OLCFc8,500
|
|
152
|
+
onnxruntime/tools/ort_format_model/ort_flatbuffers_py/fbs/TensorTypeAndShape.py,sha256=noUmY1JdkO-Bjhghyuevf7cy0Gokhnh0r_AcmIt1fl4,2326
|
|
153
|
+
onnxruntime/tools/ort_format_model/ort_flatbuffers_py/fbs/TypeInfo.py,sha256=ToFYZzH0Vit5VVIztodfzP-fCOc97vgBHXz6Y_tHyWk,2599
|
|
154
|
+
onnxruntime/tools/ort_format_model/ort_flatbuffers_py/fbs/TypeInfoValue.py,sha256=a2VKzgcR6v4qiffa5a5HYNUaHKDLdmsFbAdsNwrmT5Q,198
|
|
155
|
+
onnxruntime/tools/ort_format_model/ort_flatbuffers_py/fbs/ValueInfo.py,sha256=ZC6peiqYRYKYw_si4BmtZUbRRLf9BkkD4za4xky3QHY,2655
|
|
156
|
+
onnxruntime/tools/ort_format_model/ort_flatbuffers_py/fbs/__init__.py,sha256=EfkIrreUF6TrcYpBo1NJ8GOV_p_o_YXg3fSptBN5XUo,251
|
|
157
|
+
onnxruntime/tools/qdq_helpers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
|
158
|
+
onnxruntime/tools/qdq_helpers/optimize_qdq_model.py,sha256=9kpU0Dukc0ZEHnjwO7NqNFUFVRtOx8xgFPVWZpXkEcQ,1279
|
|
159
|
+
onnxruntime/tools/qnn/add_trans_cast.py,sha256=0NY0qI3b_8LGlJcXOZzfsHyZ48GSKyx0TOErVdDl6P8,13748
|
|
160
|
+
onnxruntime/tools/qnn/gen_qnn_ctx_onnx_model.py,sha256=mDumDp_8zChijEitCcfY7kWxr5uBqC2evKQlVV4P_cs,16562
|
|
161
|
+
onnxruntime/tools/qnn/preprocess.py,sha256=ubAAXztNQSVVBByQazXWKRt-1xzmq3EWAb1rRHkgdbg,7440
|
|
162
|
+
onnxruntime/transformers/__init__.py,sha256=2c213CqzXrc0N6Cqf__Te5d_SH_stfLdNdeNrugB7SQ,321
|
|
163
|
+
onnxruntime/transformers/affinity_helper.py,sha256=KOKBvzoBr-wOk0QHMGKzY1uy1iI7E6eHpYwBdTHM-Y4,1442
|
|
164
|
+
onnxruntime/transformers/benchmark.py,sha256=kpNQNO-ZWp_DVQYu26-TNbtwifkXsx3T3fGwrecQA58,33496
|
|
165
|
+
onnxruntime/transformers/benchmark_helper.py,sha256=k3rLMZHI0YKinGeBj-dgYNFJRwc5vaYwkQC1AfJzWZM,23043
|
|
166
|
+
onnxruntime/transformers/bert_perf_test.py,sha256=F8AQjgGUUrmN0loMLiTgstM-M08aufoPI4-uYJkYtYc,20792
|
|
167
|
+
onnxruntime/transformers/bert_test_data.py,sha256=8AFHdTtsKqaOARwRmXhQV59fBwatohpFeHVSeANn4kY,23446
|
|
168
|
+
onnxruntime/transformers/compare_bert_results.py,sha256=WzGCIGUwgOWzxesaWKGii0Xy3JfcY8LFkr8qceRR2vg,8448
|
|
169
|
+
onnxruntime/transformers/constants.py,sha256=HBmbfaRLrLCen1q0C9tZYJApn2NBPX2p2h05hc5DTsE,1127
|
|
170
|
+
onnxruntime/transformers/convert_generation.py,sha256=04WN23AxUEcaWpq-CaC5CBlxrbKWSItiBCFpV893EtI,145070
|
|
171
|
+
onnxruntime/transformers/convert_tf_models_to_pytorch.py,sha256=DMOWS-tr3BlcEhk_35LyHfC6E15s7gsVWkb_HLaVzQU,6841
|
|
172
|
+
onnxruntime/transformers/convert_to_packing_mode.py,sha256=9v4VmtOV6-WT4FrgIQILSlu2hJJFwNXSiSj1W5nb6Eg,16804
|
|
173
|
+
onnxruntime/transformers/dynamo_onnx_helper.py,sha256=88r04Lcz88tZM9xFGkdbyMyc5tIWD1VoyYl180sgwXA,7802
|
|
174
|
+
onnxruntime/transformers/float16.py,sha256=nMTYghlnKLfqnnAVnsUYCKEwt_nkcPe3T9JIfg4Un3A,24687
|
|
175
|
+
onnxruntime/transformers/fusion_attention.py,sha256=VVYBo0ZegDgzXMlgiAXU_rgpNWDnOsS5ACU-M0DsRwo,51393
|
|
176
|
+
onnxruntime/transformers/fusion_attention_clip.py,sha256=a3iO4ZTYoCN8bpOmbLgQSvdUp1zKsL-U_ofsPa9v4ok,14096
|
|
177
|
+
onnxruntime/transformers/fusion_attention_sam2.py,sha256=s5TNIr5hi8UiorlfAzEAlJKUfcn6WhEvw5CEG1bH3Pg,21307
|
|
178
|
+
onnxruntime/transformers/fusion_attention_unet.py,sha256=CAQikTpvz4KbCJUx9jonZzgy30l_w9ul6hW1RNnXUtw,56955
|
|
179
|
+
onnxruntime/transformers/fusion_attention_vae.py,sha256=1JwY-DnsngQymXCIhb_huu3SzvgXcuKDkfC7duSlfgY,12379
|
|
180
|
+
onnxruntime/transformers/fusion_bart_attention.py,sha256=EcTHIMM-pUIQ1expZ0eCtpWHMq59uZkdqq9ANjy2JGQ,20277
|
|
181
|
+
onnxruntime/transformers/fusion_base.py,sha256=AP8WDXSHPbcqsGzRORyiHOvPDYyomMDXlC-oUFV34yQ,5966
|
|
182
|
+
onnxruntime/transformers/fusion_bias_add.py,sha256=YOOZSfui3NOV0GUcNJmRSz-hoZOjQnRimjfAZPNOTzM,2041
|
|
183
|
+
onnxruntime/transformers/fusion_biasgelu.py,sha256=vGamxthOu6jXsxCRVdTFaP25-_tnjz9TVq91pIRV_Is,2300
|
|
184
|
+
onnxruntime/transformers/fusion_biassplitgelu.py,sha256=qN_YFjwT8jmBUaDobT9eGRZDCT6vSlbJvD0ZStgNFJs,4491
|
|
185
|
+
onnxruntime/transformers/fusion_conformer_attention.py,sha256=G8fq5BVB6t8--8xCpr1_tKmD6Vf1kw9UncCKB7CxkKk,8424
|
|
186
|
+
onnxruntime/transformers/fusion_constant_fold.py,sha256=qkthyYCZlSl1RMvfncCCmfkDzCsZ4tlOE9kf1-qENOA,6014
|
|
187
|
+
onnxruntime/transformers/fusion_embedlayer.py,sha256=RxQ2ZhT96d2g7-m9QcRAj4mD63Dn2s1Zz2J59AykKp8,36674
|
|
188
|
+
onnxruntime/transformers/fusion_fastgelu.py,sha256=RwvKIBJGRMk2MoH9j6HtFfQByH_Hw6eGmjeX9dtx_Lg,18181
|
|
189
|
+
onnxruntime/transformers/fusion_gelu.py,sha256=S9Jia08I3jXwafbo73xrqPOCT7rNHCnpASjZLHWiRcU,10451
|
|
190
|
+
onnxruntime/transformers/fusion_gelu_approximation.py,sha256=Xsa2v5mHjEuZrwnf1bm3UCCJ8I1is0dmuzzXgf5zDl4,1029
|
|
191
|
+
onnxruntime/transformers/fusion_gemmfastgelu.py,sha256=iRdK6jX-LlUIBGxjGavJsZKgLrj-hPBXpLvA13UIoR4,4208
|
|
192
|
+
onnxruntime/transformers/fusion_gpt_attention.py,sha256=20ZhplkAVJ3rq1VWwcNRmRs6OZu7lTHKIop3SAyDSUw,22508
|
|
193
|
+
onnxruntime/transformers/fusion_gpt_attention_megatron.py,sha256=HhoweTBxleb1niPOU_cfQzvUwM4LjxCVuZZWVEy3Imw,13639
|
|
194
|
+
onnxruntime/transformers/fusion_gpt_attention_no_past.py,sha256=qQb8WekiDJeQUV8egoCTrLoZki018veZTVVE-w3p3ds,10794
|
|
195
|
+
onnxruntime/transformers/fusion_group_norm.py,sha256=NBilP_TOZjlhuUS81EEEZMDNDzRPvje_r4_VTx9YlJ4,7645
|
|
196
|
+
onnxruntime/transformers/fusion_layernorm.py,sha256=qWb5Zp7itAXFyGeH4yBRr_XSvMZt1iTukBi_uzWWReU,20845
|
|
197
|
+
onnxruntime/transformers/fusion_mha_mmdit.py,sha256=l6KDhrKj0-9Q-t7hCYcWxzzkUUuPu6HFUPaPJZ_GwT4,25816
|
|
198
|
+
onnxruntime/transformers/fusion_nhwc_conv.py,sha256=Se5emPntX7hA0q20yklwu5RSyn_6D_vpUHMo4Z5U5HE,3948
|
|
199
|
+
onnxruntime/transformers/fusion_options.py,sha256=NSeQVc9Hz9S-4uGpxuYfsigzdLDOdsRqXyujbfRPGhw,12704
|
|
200
|
+
onnxruntime/transformers/fusion_qordered_attention.py,sha256=s1BQ8OolahrgxaSriVmli2AqTmn_fNQZH2KxU1HhmRM,17137
|
|
201
|
+
onnxruntime/transformers/fusion_qordered_gelu.py,sha256=4nuNJ0W9mDKH9Uw3tuEOc8O_ld2iRLA-8t0CUaPx7IE,4410
|
|
202
|
+
onnxruntime/transformers/fusion_qordered_layernorm.py,sha256=hzWVzxKVlL_fYd4Nv-xlOb9AYSQjHY_j3HzkFv-p_XI,4932
|
|
203
|
+
onnxruntime/transformers/fusion_qordered_matmul.py,sha256=5mjdEPTb8aT_ajAZFqEynM6fXWjJmg4iBpz3MrR-ayY,8541
|
|
204
|
+
onnxruntime/transformers/fusion_quickgelu.py,sha256=e7MMq9G2aiIotoxnkd0a7-XmvyY-wYyefdTmXtBO_f8,2869
|
|
205
|
+
onnxruntime/transformers/fusion_reshape.py,sha256=AfT88v22G6PgZPzVKM39_QduUnlIbe8dbxvPCh-5dkg,6403
|
|
206
|
+
onnxruntime/transformers/fusion_rotary_attention.py,sha256=W0qhTvG7_AZdfpUZsujGYBY7KVJenqgQ5n46Iz-zeh8,68222
|
|
207
|
+
onnxruntime/transformers/fusion_shape.py,sha256=SbwDCkucLt3U-kCdxEEaJAJa4nWRtKdU81qHYYcytfk,3763
|
|
208
|
+
onnxruntime/transformers/fusion_simplified_layernorm.py,sha256=RovL8O0nVTMZTWgvXQ-vfzH_1k9fd0hDY70oKAu0Ly4,7831
|
|
209
|
+
onnxruntime/transformers/fusion_skip_group_norm.py,sha256=_dK_FotisHnc4s0zuyAzYNjHBznh_0iGWHuSD6BgGzo,10855
|
|
210
|
+
onnxruntime/transformers/fusion_skiplayernorm.py,sha256=P4r7cXVTjRLZIyOqkOXl5gUjiOBiXRQaaV7u4Xm__I0,9168
|
|
211
|
+
onnxruntime/transformers/fusion_transpose.py,sha256=OkcS-03fQuIau8MmSXxuXyUIZqlfp7xYkE_DyuzOneM,7004
|
|
212
|
+
onnxruntime/transformers/fusion_utils.py,sha256=CozUg6Awde9EeJGJcOemOpSkgIpPhSOUo1zO4w5off8,13324
|
|
213
|
+
onnxruntime/transformers/huggingface_models.py,sha256=9X08Ad57xm-GiPcB6YcHRJT10QBE-DmGUL32xkP3OIU,4005
|
|
214
|
+
onnxruntime/transformers/import_utils.py,sha256=_ILscQRcSyaJHt1l6jqcny5FWy7Qr6N_7hKs5aav8oM,651
|
|
215
|
+
onnxruntime/transformers/io_binding_helper.py,sha256=DKBWGVi3Cv8eCcWCZOY0v85HD_wDDI0FlSJc5a0EkKw,19440
|
|
216
|
+
onnxruntime/transformers/large_model_exporter.py,sha256=f9VswBG1yIu_z6I4OfU6rYefIA2zladP9INzeo2Z-c8,15296
|
|
217
|
+
onnxruntime/transformers/machine_info.py,sha256=suxpmfU-7yB9UVji_hum2nkc7AALZnivUKkgj18ajgs,7581
|
|
218
|
+
onnxruntime/transformers/metrics.py,sha256=DEjoDQGAf2Yq_BAz66x3w0KZm1QhvG4ZENk8rIIC2UQ,5223
|
|
219
|
+
onnxruntime/transformers/onnx_exporter.py,sha256=jFFFq0gGDtDtLgi0zoDZuIUR3ba_r6J-a2b7IgoRAnQ,25340
|
|
220
|
+
onnxruntime/transformers/onnx_model.py,sha256=tg3iprgpcEFwnRk4ADiXrKxD-Sj-kI4-IeEYEDOiCko,69954
|
|
221
|
+
onnxruntime/transformers/onnx_model_bart.py,sha256=yLPRW-4YuZ_H_6xN5H4hE2Wr1bYp7Em4ZrP_T0hC20s,5547
|
|
222
|
+
onnxruntime/transformers/onnx_model_bert.py,sha256=4UF9Omcq2cKQ4VeycaH9olH8h4RvlEQUsrkuR3qXG3Q,20402
|
|
223
|
+
onnxruntime/transformers/onnx_model_bert_keras.py,sha256=iAlnQJJCZMGBUx5fLErc7FInVzBcz14ncc-RAO3svqU,19010
|
|
224
|
+
onnxruntime/transformers/onnx_model_bert_tf.py,sha256=Ebq6LeUyxIzE8_LG_xvaPRxtRIlD2gTtVKE5UyeGoRo,25503
|
|
225
|
+
onnxruntime/transformers/onnx_model_clip.py,sha256=RyBZh4nOvV7TP_b_SN_nSNgK5noVhtwOHbKhiTCifds,1394
|
|
226
|
+
onnxruntime/transformers/onnx_model_conformer.py,sha256=lP4r-iid7Jl1AAOGdZM5S_MCQ__oRq-KTibGcM6Gri4,1412
|
|
227
|
+
onnxruntime/transformers/onnx_model_gpt2.py,sha256=3LmzgHuLvO5tyNHKWGidttyqrcpIE7aLBYbRqzjolUg,3913
|
|
228
|
+
onnxruntime/transformers/onnx_model_mmdit.py,sha256=gyQ7AuF9ZU6o6oOEytM4HMo-B5p6JvXPxM5xGAdjEpk,4209
|
|
229
|
+
onnxruntime/transformers/onnx_model_phi.py,sha256=lChW-NHkl5YWRhwyMCpwMYDSfXFdYmIx456SB3TQbKQ,36339
|
|
230
|
+
onnxruntime/transformers/onnx_model_sam2.py,sha256=3mojj1dGPaw8gb8jJABQzw9rLHD1IBdpVEM-h-Tpt6U,4981
|
|
231
|
+
onnxruntime/transformers/onnx_model_t5.py,sha256=-cl2iOALXNMff_zFsi9eCiwIJKWHFDmjVnfQ8cIQTxo,38056
|
|
232
|
+
onnxruntime/transformers/onnx_model_tnlr.py,sha256=6MlEn9yRlF3jVW1zd9NCR0qxnPgIa2dEhaylvbf8dCo,8405
|
|
233
|
+
onnxruntime/transformers/onnx_model_unet.py,sha256=OsQqUi3Ak5eMK2w2-j294o9bwchH0zH0gYgmEZUhLRQ,9513
|
|
234
|
+
onnxruntime/transformers/onnx_model_vae.py,sha256=JTp3ctfgHR7YERn9cNYpXB4ZM82X3CgTYHtVdgrJqxY,1513
|
|
235
|
+
onnxruntime/transformers/onnx_utils.py,sha256=GGkO0ebsoPvVOPNfeYC0tN9DJWiBaLXXzjQhBrAXvNE,2175
|
|
236
|
+
onnxruntime/transformers/optimizer.py,sha256=zjKsbqvXRYmZEKXcxaOk1OusCXDnfQtvLQS_6695U10,25691
|
|
237
|
+
onnxruntime/transformers/past_helper.py,sha256=uDRilngPAzWQ-ZCNiOivB_qQUroOSSZu6A2EZg6mu0o,6955
|
|
238
|
+
onnxruntime/transformers/profile_result_processor.py,sha256=fZSWV2p3QWkG2cgqfEt1dC9rglnY8UgviCWs6W4agek,12940
|
|
239
|
+
onnxruntime/transformers/profiler.py,sha256=fp42DHIBu4VZMDJTVclEm83TEh-HNwPQGz0_AvtLNmg,13736
|
|
240
|
+
onnxruntime/transformers/quantize_helper.py,sha256=wcptMQ4RiSZ18DhhPFAZ8AbKrbNx--UGj63S6lBVOig,2927
|
|
241
|
+
onnxruntime/transformers/shape_infer_helper.py,sha256=v7VKU2m7blQp21PBbDPeySgwFMoP_BC1OM_rUIlnAFo,4566
|
|
242
|
+
onnxruntime/transformers/shape_optimizer.py,sha256=McbgtLsmb3j0AVHWnVigktDgxMktE0UXAdyemg3NNjU,15468
|
|
243
|
+
onnxruntime/transformers/torch_onnx_export_helper.py,sha256=DOTqWF9DEbxsxqKWtq3NCqcA7de-JSMgjS-MyczJimg,2575
|
|
244
|
+
onnxruntime/transformers/models/bart/__init__.py,sha256=F8Gml7gD7jmMyMIkKvHiXLfj7kWyz175X1-5_iaFx5k,495
|
|
245
|
+
onnxruntime/transformers/models/bart/export.py,sha256=PNlhkbvrxTxSSLXpzqoa02Lektzf8rdZpcVFBxw-qcI,4285
|
|
246
|
+
onnxruntime/transformers/models/bert/__init__.py,sha256=F8Gml7gD7jmMyMIkKvHiXLfj7kWyz175X1-5_iaFx5k,495
|
|
247
|
+
onnxruntime/transformers/models/bert/eval_squad.py,sha256=If_RsXJHUt0O78-FLnfbS4DXeccihgm7HFdhC8NzGMA,12355
|
|
248
|
+
onnxruntime/transformers/models/gpt2/__init__.py,sha256=F8Gml7gD7jmMyMIkKvHiXLfj7kWyz175X1-5_iaFx5k,495
|
|
249
|
+
onnxruntime/transformers/models/gpt2/benchmark_gpt2.py,sha256=53krYHNeJBk8ZixWoWrZHxCbe9z4bDYiGW02tzxR-88,15930
|
|
250
|
+
onnxruntime/transformers/models/gpt2/convert_to_onnx.py,sha256=5UwCRcAjW2MK-81M1igXMCpdlOKJ8eA0iN5mSaY5C5E,20837
|
|
251
|
+
onnxruntime/transformers/models/gpt2/gpt2_helper.py,sha256=1mZeKeDKm5jXgVg2sE5Jofbyg_0HQ84d9Gqg26dLPes,41331
|
|
252
|
+
onnxruntime/transformers/models/gpt2/gpt2_parity.py,sha256=A963JoK1zQb6UgFUOp1strqjoJ_lZQ2pvrFLa0UjIMA,18248
|
|
253
|
+
onnxruntime/transformers/models/gpt2/gpt2_tester.py,sha256=JqQKJvOfHKcNcnwtDIp_k98YHuT9BSmVfz2th8IIY-c,20071
|
|
254
|
+
onnxruntime/transformers/models/gpt2/parity_check_helper.py,sha256=x54knOO4LgzWvle3N3HuWyRTTNhhcanwMa12AredDTo,5840
|
|
255
|
+
onnxruntime/transformers/models/llama/__init__.py,sha256=yR2FucNw-jt_3CbNt-zuM7DmldPq1rJK3SV8gRISzN0,490
|
|
256
|
+
onnxruntime/transformers/models/llama/benchmark.py,sha256=fuh9oKRKldxzZ7ccnlb84B6g9NV520CiMR6Z0xnsflM,27294
|
|
257
|
+
onnxruntime/transformers/models/llama/benchmark_all.py,sha256=wiZ6LhbOmt_vNVuzgcLRaXPBudPoydL9XUbGi4TyTpw,15793
|
|
258
|
+
onnxruntime/transformers/models/llama/benchmark_e2e.py,sha256=741bV8ZSqA0CmP5quL3gMuIXYp3hwBoihPrsqPEI2Po,25477
|
|
259
|
+
onnxruntime/transformers/models/llama/convert_to_onnx.py,sha256=GFTkqaEXHXFdg2iXYXSQQLox_X3_sYq0pxyEcB3OnvQ,43312
|
|
260
|
+
onnxruntime/transformers/models/llama/dist_settings.py,sha256=swRa5c26FmnPJloTRqELnK0OrYmE1mI_RrM9yf_Q5ho,1659
|
|
261
|
+
onnxruntime/transformers/models/llama/llama_inputs.py,sha256=x7vxfIAMA0QmMVtjDVNSOoA8nrKv22bTnRNQfSyK1YY,20726
|
|
262
|
+
onnxruntime/transformers/models/llama/llama_parity.py,sha256=MViDkW2T_Ln8yLTarZLbtZTjE0mlVTgpBclbmKo_yfs,11900
|
|
263
|
+
onnxruntime/transformers/models/llama/llama_torch.py,sha256=OIbSpg7bjfPN9SzyGfaBKmzsa-bGHQTx0X-etiEl0Zk,1732
|
|
264
|
+
onnxruntime/transformers/models/llama/quant_kv_dataloader.py,sha256=piVldpGm9eBmF4wzgmKJprhujqTPddqORxZyLizcJdA,4959
|
|
265
|
+
onnxruntime/transformers/models/longformer/__init__.py,sha256=F8Gml7gD7jmMyMIkKvHiXLfj7kWyz175X1-5_iaFx5k,495
|
|
266
|
+
onnxruntime/transformers/models/longformer/benchmark_longformer.py,sha256=G6IlIJfIDblXW2iBIlALaf_Lf9mQvAxvE_TNXldwmD8,30230
|
|
267
|
+
onnxruntime/transformers/models/longformer/convert_to_onnx.py,sha256=SZwLsRhYpP7di289903R768eSK0zRZlRniSWEACmX_8,15207
|
|
268
|
+
onnxruntime/transformers/models/longformer/generate_test_data.py,sha256=wQxpgo_vZBhKRlquJwUB9FH3_xxvyDC3aCCZdkvADLM,9964
|
|
269
|
+
onnxruntime/transformers/models/longformer/longformer_helper.py,sha256=n3crEYnVYZxNzoyTNMzX3LFgxHH-FcaqtdfRmyAMg-4,3123
|
|
270
|
+
onnxruntime/transformers/models/phi2/__init__.py,sha256=yR2FucNw-jt_3CbNt-zuM7DmldPq1rJK3SV8gRISzN0,490
|
|
271
|
+
onnxruntime/transformers/models/phi2/convert_to_onnx.py,sha256=nfsADQV9T6d2CMaSmOzGXFxDtqrLqXk191FibxGBMkk,20998
|
|
272
|
+
onnxruntime/transformers/models/phi2/inference_example.py,sha256=4Q7VjNGCvrjIbC1Ul8W8SRndaiI42cGQ1L4iO0vMwlg,17705
|
|
273
|
+
onnxruntime/transformers/models/sam2/__init__.py,sha256=F8Gml7gD7jmMyMIkKvHiXLfj7kWyz175X1-5_iaFx5k,495
|
|
274
|
+
onnxruntime/transformers/models/sam2/benchmark_sam2.py,sha256=2ABs0vz1B5bN0EaRUUMstvmez4hjt15teLK3PdqECcg,22468
|
|
275
|
+
onnxruntime/transformers/models/sam2/convert_to_onnx.py,sha256=g1y2o679JtjTEkCHJn3r-QSvosIBKBTeUz6mGTHctnI,10669
|
|
276
|
+
onnxruntime/transformers/models/sam2/image_decoder.py,sha256=XCxCgIXSKLKvO4CQXGuyEuoXeDn9yxy8RnmfQgALOWc,11093
|
|
277
|
+
onnxruntime/transformers/models/sam2/image_encoder.py,sha256=AEFh2pdqwmJkdKVsowms0Va8w8-iKhsRk5Hu5eajWyQ,9788
|
|
278
|
+
onnxruntime/transformers/models/sam2/mask_decoder.py,sha256=QwnZgjWwFsLdoNmnr7OG8Kgk5lrpZfHOeFGg5LLXQG0,9066
|
|
279
|
+
onnxruntime/transformers/models/sam2/nvtx_helper.py,sha256=XJFujDb-p27zCJs-O-ZjcKEAetc5dc0keFvsIYgFQ6E,1312
|
|
280
|
+
onnxruntime/transformers/models/sam2/prompt_encoder.py,sha256=STPG6PNZ5oVmuNdfeWuHkxxR2cOiwXN4lYKnvm3pyNA,8530
|
|
281
|
+
onnxruntime/transformers/models/sam2/sam2_demo.py,sha256=SH522SETO-RHeqH2Em848TTqe2iBXXYgp5iDc-xPA-0,10822
|
|
282
|
+
onnxruntime/transformers/models/sam2/sam2_image_onnx_predictor.py,sha256=bB-f7bi7NSb0XVeAA6J3v9jkMMJDHdHmS5PdX1UO0nU,12702
|
|
283
|
+
onnxruntime/transformers/models/sam2/sam2_utils.py,sha256=gNzqfMwdmIbkWCvRrKeJ1uYpz9fqFxrGNGQ9ikMxRfc,5667
|
|
284
|
+
onnxruntime/transformers/models/stable_diffusion/__init__.py,sha256=F8Gml7gD7jmMyMIkKvHiXLfj7kWyz175X1-5_iaFx5k,495
|
|
285
|
+
onnxruntime/transformers/models/stable_diffusion/benchmark.py,sha256=94DvZFoBCJEfrENa2sJwMjdRn-MBXHPLErQs-WhPt4s,51563
|
|
286
|
+
onnxruntime/transformers/models/stable_diffusion/benchmark_controlnet.py,sha256=hd1yCasPxxcu7g9O2Bg-CRW4IsPM-PE8BbNze2TdtW0,13355
|
|
287
|
+
onnxruntime/transformers/models/stable_diffusion/demo_txt2img.py,sha256=mNgUXe2Ta6aNK4b-hom-0Zo7eMI5lXClyw-dMyAIxvM,3415
|
|
288
|
+
onnxruntime/transformers/models/stable_diffusion/demo_txt2img_xl.py,sha256=2NH8ySNa9hHHX0ZoAYO8gSUUIhi_wHMsTB9-NLPlLC8,10217
|
|
289
|
+
onnxruntime/transformers/models/stable_diffusion/demo_utils.py,sha256=nz5QzXHy12J5VoSvt-lKxLt4c_VFf58K0OI3hoiZP1A,29376
|
|
290
|
+
onnxruntime/transformers/models/stable_diffusion/diffusion_models.py,sha256=SwAEjw6mI08qNxpr6f8xfeg_TrhafUeLmSSkfePWuUs,51716
|
|
291
|
+
onnxruntime/transformers/models/stable_diffusion/diffusion_schedulers.py,sha256=ohAEuJNU_-rgz9wwz7pRc7tJI1QJOc5zjJYkvuh2jKs,49468
|
|
292
|
+
onnxruntime/transformers/models/stable_diffusion/engine_builder.py,sha256=YFe6_0DI97qhfdlkyliqXCBnobWrTVWfJl2uL-HaAXs,11965
|
|
293
|
+
onnxruntime/transformers/models/stable_diffusion/engine_builder_ort_cuda.py,sha256=t9cjyv617MpQt_S7crnspAHbw0WT8Rf3lCVRkeLwuVo,16241
|
|
294
|
+
onnxruntime/transformers/models/stable_diffusion/engine_builder_ort_trt.py,sha256=0M-JLT3Z1zYPEVkJ0TPCZuhbIFCstbBi5Wh623VLcww,11451
|
|
295
|
+
onnxruntime/transformers/models/stable_diffusion/engine_builder_tensorrt.py,sha256=-VCFZgeT233jWR1B8mzSJWEpoPOUbHNWTnWVMXmf0k4,16016
|
|
296
|
+
onnxruntime/transformers/models/stable_diffusion/engine_builder_torch.py,sha256=I_viNNv16MEMklC6yVByLa3fIKwdp2iSLqwCCflQK3M,4306
|
|
297
|
+
onnxruntime/transformers/models/stable_diffusion/optimize_pipeline.py,sha256=3K2FTtM3AKsshQ-JQSNVIkI6i0fbyu_boboPE9kEvI0,22238
|
|
298
|
+
onnxruntime/transformers/models/stable_diffusion/ort_optimizer.py,sha256=Yl8skQoXYbouhw5ti2zCOfbMLbSm3iXniA5GLmQ9A3s,5853
|
|
299
|
+
onnxruntime/transformers/models/stable_diffusion/pipeline_stable_diffusion.py,sha256=EDU3Hnvo4nX3u4VRxw5uyHQZC59u-RiXrIxMhzGPvPg,34029
|
|
300
|
+
onnxruntime/transformers/models/stable_diffusion/trt_utilities.py,sha256=XZCfqG_kZ72e-L9p7PlGqc4NLvFZF1h40A6Guyj6z8k,432
|
|
301
|
+
onnxruntime/transformers/models/t5/__init__.py,sha256=F8Gml7gD7jmMyMIkKvHiXLfj7kWyz175X1-5_iaFx5k,495
|
|
302
|
+
onnxruntime/transformers/models/t5/convert_to_onnx.py,sha256=70tYNOGfx0HgIHgMdbRTLbsJlhQkEbW4h2xZveSvNeE,10510
|
|
303
|
+
onnxruntime/transformers/models/t5/t5_decoder.py,sha256=GmUTeBavhX2TfIjZIfMNuMML_3-6BRqetUEXAvOxFPo,17188
|
|
304
|
+
onnxruntime/transformers/models/t5/t5_encoder.py,sha256=xSjy9AmscMdyCDzSn-v59xXqS634lAs1EmsJUsBaB0E,2319
|
|
305
|
+
onnxruntime/transformers/models/t5/t5_encoder_decoder_init.py,sha256=hf7ozFvgXSE_FH17p88MS0fpPBbnlsUYfKgncCkmnvU,15421
|
|
306
|
+
onnxruntime/transformers/models/t5/t5_helper.py,sha256=FKwaOB-z4uvEBpsKZbJgVjMP6xl2D9fryfHjD-4eneA,12515
|
|
307
|
+
onnxruntime/transformers/models/whisper/__init__.py,sha256=yR2FucNw-jt_3CbNt-zuM7DmldPq1rJK3SV8gRISzN0,490
|
|
308
|
+
onnxruntime/transformers/models/whisper/benchmark.py,sha256=pb-SPFQUbyDB6apStWARVgdu4y_P92-KabYPgLeFka8,22447
|
|
309
|
+
onnxruntime/transformers/models/whisper/benchmark_all.py,sha256=zLzB84_cyIDJSR-dyZCXjJ3qGvvpKx97LonodvboZ7s,19368
|
|
310
|
+
onnxruntime/transformers/models/whisper/convert_to_onnx.py,sha256=pra2cOGGkE8Zvi4yuDO5blGK8rfx5o4pDDTKsflLNGk,21750
|
|
311
|
+
onnxruntime/transformers/models/whisper/whisper_chain.py,sha256=CF9UoGWWmHwljfaLkj8zyrxrjVymYMvGYDNJ7dFLIbI,15238
|
|
312
|
+
onnxruntime/transformers/models/whisper/whisper_decoder.py,sha256=za4PBjUSbx-lwQReCMs7ODBK19kUVvROwGY2_01AyRk,21862
|
|
313
|
+
onnxruntime/transformers/models/whisper/whisper_encoder.py,sha256=vyykcthAejmd2K4lJyqd3JxXxTwcOVhxqMdj7AAeyUU,6342
|
|
314
|
+
onnxruntime/transformers/models/whisper/whisper_encoder_decoder_init.py,sha256=PoseFY-kxybyiCd0JVhPT1bC9rG_qzKmY76mKl37eGk,16802
|
|
315
|
+
onnxruntime/transformers/models/whisper/whisper_helper.py,sha256=AVbtUmuBhpcxycjiKJqbkozIBR_5PZTX65aTYiqBfFI,51334
|
|
316
|
+
onnxruntime/transformers/models/whisper/whisper_inputs.py,sha256=rSjFJ9Llz2FmpRXDz8cK2KFzr2b5l-n4eo8VN463iUE,16041
|
|
317
|
+
onnxruntime/transformers/models/whisper/whisper_jump_times.py,sha256=bgxGRs00NAf5F7hzJBJbt7JB3IMuCOxylhCgnf16-l8,19950
|
|
318
|
+
onnxruntime_directml-1.24.1.dist-info/METADATA,sha256=bH_R3gXdEyZWGSYkHHv85rxIVUOJdULR4OWKMK9a9qI,5155
|
|
319
|
+
onnxruntime_directml-1.24.1.dist-info/WHEEL,sha256=z-AL790RFdWKt_PD8upif4biP-wiSujrp37waEcSdi0,102
|
|
320
|
+
onnxruntime_directml-1.24.1.dist-info/entry_points.txt,sha256=7qLS4FbGXwPZjfdpVAGpnmk9I6m6H5CxEnwcCx1Imjs,77
|
|
321
|
+
onnxruntime_directml-1.24.1.dist-info/top_level.txt,sha256=zk_fJEekrTm9DLxX2LwGegokVqP6blqPhFoMIuh0Nv8,12
|
|
322
|
+
onnxruntime_directml-1.24.1.dist-info/RECORD,,
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
onnxruntime
|