onnxruntime-directml 1.20.0__cp313-cp313-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 +6508 -0
- onnxruntime/__init__.py +78 -0
- onnxruntime/backend/__init__.py +6 -0
- onnxruntime/backend/backend.py +174 -0
- onnxruntime/backend/backend_rep.py +53 -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/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 +1108 -0
- onnxruntime/capi/onnxruntime_providers_shared.dll +0 -0
- onnxruntime/capi/onnxruntime_pybind11_state.pyd +0 -0
- onnxruntime/capi/onnxruntime_validation.py +150 -0
- onnxruntime/capi/version_info.py +2 -0
- onnxruntime/datasets/__init__.py +17 -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 +16 -0
- onnxruntime/quantization/base_quantizer.py +532 -0
- onnxruntime/quantization/calibrate.py +1245 -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/mixed_precision_overrides_utils.py +413 -0
- onnxruntime/quantization/execution_providers/qnn/preprocess.py +307 -0
- onnxruntime/quantization/execution_providers/qnn/quant_config.py +387 -0
- onnxruntime/quantization/fusions/__init__.py +3 -0
- onnxruntime/quantization/fusions/fusion.py +311 -0
- onnxruntime/quantization/fusions/fusion_gelu.py +272 -0
- onnxruntime/quantization/fusions/fusion_layernorm.py +135 -0
- onnxruntime/quantization/matmul_4bits_quantizer.py +1480 -0
- onnxruntime/quantization/matmul_bnb4_quantizer.py +240 -0
- onnxruntime/quantization/onnx_model.py +580 -0
- onnxruntime/quantization/onnx_quantizer.py +1008 -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 +258 -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 +166 -0
- onnxruntime/quantization/operators/lstm.py +117 -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 +100 -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 +1187 -0
- onnxruntime/quantization/quant_utils.py +891 -0
- onnxruntime/quantization/quantize.py +748 -0
- onnxruntime/quantization/registry.py +106 -0
- onnxruntime/quantization/shape_inference.py +187 -0
- onnxruntime/quantization/tensor_quant_overrides.py +516 -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 +377 -0
- onnxruntime/tools/file_utils.py +46 -0
- onnxruntime/tools/logger.py +11 -0
- onnxruntime/tools/make_dynamic_shape_fixed.py +72 -0
- onnxruntime/tools/mobile_helpers/__init__.py +0 -0
- onnxruntime/tools/mobile_helpers/coreml_supported_mlprogram_ops.md +33 -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 +739 -0
- onnxruntime/tools/offline_tuning.py +169 -0
- onnxruntime/tools/onnx_model_utils.py +413 -0
- onnxruntime/tools/onnx_randomizer.py +85 -0
- onnxruntime/tools/onnxruntime_test.py +164 -0
- onnxruntime/tools/optimize_onnx_model.py +55 -0
- onnxruntime/tools/ort_format_model/__init__.py +25 -0
- onnxruntime/tools/ort_format_model/operator_type_usage_processors.py +663 -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 +84 -0
- onnxruntime/tools/ort_format_model/utils.py +62 -0
- onnxruntime/tools/pytorch_export_contrib_ops.py +108 -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/reduced_build_config_parser.py +202 -0
- onnxruntime/tools/symbolic_shape_infer.py +3016 -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 +944 -0
- onnxruntime/transformers/benchmark_helper.py +646 -0
- onnxruntime/transformers/bert_perf_test.py +634 -0
- onnxruntime/transformers/bert_test_data.py +642 -0
- onnxruntime/transformers/compare_bert_results.py +246 -0
- onnxruntime/transformers/constants.py +47 -0
- onnxruntime/transformers/convert_generation.py +3124 -0
- onnxruntime/transformers/convert_tf_models_to_pytorch.py +205 -0
- onnxruntime/transformers/convert_to_packing_mode.py +387 -0
- onnxruntime/transformers/dynamo_onnx_helper.py +104 -0
- onnxruntime/transformers/float16.py +501 -0
- onnxruntime/transformers/fusion_attention.py +1235 -0
- onnxruntime/transformers/fusion_attention_clip.py +257 -0
- onnxruntime/transformers/fusion_attention_sam2.py +534 -0
- onnxruntime/transformers/fusion_attention_unet.py +1304 -0
- onnxruntime/transformers/fusion_attention_vae.py +301 -0
- onnxruntime/transformers/fusion_bart_attention.py +640 -0
- onnxruntime/transformers/fusion_base.py +137 -0
- onnxruntime/transformers/fusion_bias_add.py +58 -0
- onnxruntime/transformers/fusion_biasgelu.py +66 -0
- onnxruntime/transformers/fusion_biassplitgelu.py +111 -0
- onnxruntime/transformers/fusion_conformer_attention.py +143 -0
- onnxruntime/transformers/fusion_embedlayer.py +811 -0
- onnxruntime/transformers/fusion_fastgelu.py +360 -0
- onnxruntime/transformers/fusion_gelu.py +259 -0
- onnxruntime/transformers/fusion_gelu_approximation.py +25 -0
- onnxruntime/transformers/fusion_gemmfastgelu.py +122 -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 +179 -0
- onnxruntime/transformers/fusion_layernorm.py +465 -0
- onnxruntime/transformers/fusion_nhwc_conv.py +100 -0
- onnxruntime/transformers/fusion_options.py +340 -0
- onnxruntime/transformers/fusion_qordered_attention.py +421 -0
- onnxruntime/transformers/fusion_qordered_gelu.py +119 -0
- onnxruntime/transformers/fusion_qordered_layernorm.py +123 -0
- onnxruntime/transformers/fusion_qordered_matmul.py +217 -0
- onnxruntime/transformers/fusion_quickgelu.py +74 -0
- onnxruntime/transformers/fusion_reshape.py +173 -0
- onnxruntime/transformers/fusion_rotary_attention.py +1592 -0
- onnxruntime/transformers/fusion_shape.py +110 -0
- onnxruntime/transformers/fusion_simplified_layernorm.py +159 -0
- onnxruntime/transformers/fusion_skip_group_norm.py +255 -0
- onnxruntime/transformers/fusion_skiplayernorm.py +209 -0
- onnxruntime/transformers/fusion_transpose.py +168 -0
- onnxruntime/transformers/fusion_utils.py +307 -0
- onnxruntime/transformers/huggingface_models.py +167 -0
- onnxruntime/transformers/import_utils.py +20 -0
- onnxruntime/transformers/io_binding_helper.py +442 -0
- onnxruntime/transformers/large_model_exporter.py +395 -0
- onnxruntime/transformers/machine_info.py +221 -0
- onnxruntime/transformers/metrics.py +164 -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 +561 -0
- onnxruntime/transformers/models/gpt2/gpt2_helper.py +1032 -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 +703 -0
- onnxruntime/transformers/models/llama/benchmark_all.py +488 -0
- onnxruntime/transformers/models/llama/benchmark_e2e.py +606 -0
- onnxruntime/transformers/models/llama/convert_to_onnx.py +1027 -0
- onnxruntime/transformers/models/llama/dist_settings.py +57 -0
- onnxruntime/transformers/models/llama/llama_inputs.py +503 -0
- onnxruntime/transformers/models/llama/llama_parity.py +309 -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 +77 -0
- onnxruntime/transformers/models/phi2/__init__.py +12 -0
- onnxruntime/transformers/models/phi2/convert_to_onnx.py +576 -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 +625 -0
- onnxruntime/transformers/models/sam2/convert_to_onnx.py +260 -0
- onnxruntime/transformers/models/sam2/image_decoder.py +273 -0
- onnxruntime/transformers/models/sam2/image_encoder.py +186 -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 +322 -0
- onnxruntime/transformers/models/sam2/sam2_image_onnx_predictor.py +280 -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 +1429 -0
- onnxruntime/transformers/models/stable_diffusion/benchmark_controlnet.py +426 -0
- onnxruntime/transformers/models/stable_diffusion/demo_txt2img.py +102 -0
- onnxruntime/transformers/models/stable_diffusion/demo_txt2img_xl.py +268 -0
- onnxruntime/transformers/models/stable_diffusion/demo_utils.py +778 -0
- onnxruntime/transformers/models/stable_diffusion/diffusion_models.py +1319 -0
- onnxruntime/transformers/models/stable_diffusion/diffusion_schedulers.py +1181 -0
- onnxruntime/transformers/models/stable_diffusion/engine_builder.py +296 -0
- onnxruntime/transformers/models/stable_diffusion/engine_builder_ort_cuda.py +388 -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 +350 -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 +278 -0
- onnxruntime/transformers/models/t5/past_helper.py +150 -0
- onnxruntime/transformers/models/t5/t5_decoder.py +438 -0
- onnxruntime/transformers/models/t5/t5_encoder.py +171 -0
- onnxruntime/transformers/models/t5/t5_encoder_decoder_init.py +299 -0
- onnxruntime/transformers/models/t5/t5_helper.py +272 -0
- onnxruntime/transformers/models/whisper/__init__.py +12 -0
- onnxruntime/transformers/models/whisper/benchmark.py +610 -0
- onnxruntime/transformers/models/whisper/benchmark_all.py +528 -0
- onnxruntime/transformers/models/whisper/convert_to_onnx.py +536 -0
- onnxruntime/transformers/models/whisper/whisper_chain.py +329 -0
- onnxruntime/transformers/models/whisper/whisper_decoder.py +402 -0
- onnxruntime/transformers/models/whisper/whisper_encoder.py +164 -0
- onnxruntime/transformers/models/whisper/whisper_encoder_decoder_init.py +306 -0
- onnxruntime/transformers/models/whisper/whisper_helper.py +524 -0
- onnxruntime/transformers/models/whisper/whisper_openai_helper.py +84 -0
- onnxruntime/transformers/onnx_exporter.py +717 -0
- onnxruntime/transformers/onnx_model.py +1569 -0
- onnxruntime/transformers/onnx_model_bart.py +142 -0
- onnxruntime/transformers/onnx_model_bert.py +481 -0
- onnxruntime/transformers/onnx_model_bert_keras.py +475 -0
- onnxruntime/transformers/onnx_model_bert_tf.py +589 -0
- onnxruntime/transformers/onnx_model_clip.py +40 -0
- onnxruntime/transformers/onnx_model_conformer.py +33 -0
- onnxruntime/transformers/onnx_model_gpt2.py +101 -0
- onnxruntime/transformers/onnx_model_phi.py +930 -0
- onnxruntime/transformers/onnx_model_sam2.py +138 -0
- onnxruntime/transformers/onnx_model_t5.py +791 -0
- onnxruntime/transformers/onnx_model_tnlr.py +227 -0
- onnxruntime/transformers/onnx_model_unet.py +259 -0
- onnxruntime/transformers/onnx_model_vae.py +43 -0
- onnxruntime/transformers/onnx_utils.py +55 -0
- onnxruntime/transformers/optimizer.py +612 -0
- onnxruntime/transformers/profiler.py +725 -0
- onnxruntime/transformers/quantize_helper.py +76 -0
- onnxruntime/transformers/shape_infer_helper.py +122 -0
- onnxruntime/transformers/shape_optimizer.py +401 -0
- onnxruntime/transformers/torch_onnx_export_helper.py +74 -0
- onnxruntime_directml-1.20.0.dist-info/METADATA +187 -0
- onnxruntime_directml-1.20.0.dist-info/RECORD +305 -0
- onnxruntime_directml-1.20.0.dist-info/WHEEL +5 -0
- onnxruntime_directml-1.20.0.dist-info/entry_points.txt +2 -0
- onnxruntime_directml-1.20.0.dist-info/top_level.txt +1 -0
|
@@ -0,0 +1,67 @@
|
|
|
1
|
+
# automatically generated by the FlatBuffers compiler, do not modify
|
|
2
|
+
|
|
3
|
+
# namespace: fbs
|
|
4
|
+
|
|
5
|
+
import flatbuffers
|
|
6
|
+
from flatbuffers.compat import import_numpy
|
|
7
|
+
np = import_numpy()
|
|
8
|
+
|
|
9
|
+
class OperatorSetId(object):
|
|
10
|
+
__slots__ = ['_tab']
|
|
11
|
+
|
|
12
|
+
@classmethod
|
|
13
|
+
def GetRootAs(cls, buf, offset=0):
|
|
14
|
+
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
15
|
+
x = OperatorSetId()
|
|
16
|
+
x.Init(buf, n + offset)
|
|
17
|
+
return x
|
|
18
|
+
|
|
19
|
+
@classmethod
|
|
20
|
+
def GetRootAsOperatorSetId(cls, buf, offset=0):
|
|
21
|
+
"""This method is deprecated. Please switch to GetRootAs."""
|
|
22
|
+
return cls.GetRootAs(buf, offset)
|
|
23
|
+
@classmethod
|
|
24
|
+
def OperatorSetIdBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
25
|
+
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x4F\x52\x54\x4D", size_prefixed=size_prefixed)
|
|
26
|
+
|
|
27
|
+
# OperatorSetId
|
|
28
|
+
def Init(self, buf, pos):
|
|
29
|
+
self._tab = flatbuffers.table.Table(buf, pos)
|
|
30
|
+
|
|
31
|
+
# OperatorSetId
|
|
32
|
+
def Domain(self):
|
|
33
|
+
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
34
|
+
if o != 0:
|
|
35
|
+
return self._tab.String(o + self._tab.Pos)
|
|
36
|
+
return None
|
|
37
|
+
|
|
38
|
+
# OperatorSetId
|
|
39
|
+
def Version(self):
|
|
40
|
+
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(6))
|
|
41
|
+
if o != 0:
|
|
42
|
+
return self._tab.Get(flatbuffers.number_types.Int64Flags, o + self._tab.Pos)
|
|
43
|
+
return 0
|
|
44
|
+
|
|
45
|
+
def OperatorSetIdStart(builder):
|
|
46
|
+
builder.StartObject(2)
|
|
47
|
+
|
|
48
|
+
def Start(builder):
|
|
49
|
+
OperatorSetIdStart(builder)
|
|
50
|
+
|
|
51
|
+
def OperatorSetIdAddDomain(builder, domain):
|
|
52
|
+
builder.PrependUOffsetTRelativeSlot(0, flatbuffers.number_types.UOffsetTFlags.py_type(domain), 0)
|
|
53
|
+
|
|
54
|
+
def AddDomain(builder, domain):
|
|
55
|
+
OperatorSetIdAddDomain(builder, domain)
|
|
56
|
+
|
|
57
|
+
def OperatorSetIdAddVersion(builder, version):
|
|
58
|
+
builder.PrependInt64Slot(1, version, 0)
|
|
59
|
+
|
|
60
|
+
def AddVersion(builder, version):
|
|
61
|
+
OperatorSetIdAddVersion(builder, version)
|
|
62
|
+
|
|
63
|
+
def OperatorSetIdEnd(builder):
|
|
64
|
+
return builder.EndObject()
|
|
65
|
+
|
|
66
|
+
def End(builder):
|
|
67
|
+
return OperatorSetIdEnd(builder)
|
|
@@ -0,0 +1,117 @@
|
|
|
1
|
+
# automatically generated by the FlatBuffers compiler, do not modify
|
|
2
|
+
|
|
3
|
+
# namespace: fbs
|
|
4
|
+
|
|
5
|
+
import flatbuffers
|
|
6
|
+
from flatbuffers.compat import import_numpy
|
|
7
|
+
np = import_numpy()
|
|
8
|
+
|
|
9
|
+
class OptimizerGroup(object):
|
|
10
|
+
__slots__ = ['_tab']
|
|
11
|
+
|
|
12
|
+
@classmethod
|
|
13
|
+
def GetRootAs(cls, buf, offset=0):
|
|
14
|
+
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
15
|
+
x = OptimizerGroup()
|
|
16
|
+
x.Init(buf, n + offset)
|
|
17
|
+
return x
|
|
18
|
+
|
|
19
|
+
@classmethod
|
|
20
|
+
def GetRootAsOptimizerGroup(cls, buf, offset=0):
|
|
21
|
+
"""This method is deprecated. Please switch to GetRootAs."""
|
|
22
|
+
return cls.GetRootAs(buf, offset)
|
|
23
|
+
@classmethod
|
|
24
|
+
def OptimizerGroupBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
25
|
+
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x4F\x44\x54\x43", size_prefixed=size_prefixed)
|
|
26
|
+
|
|
27
|
+
# OptimizerGroup
|
|
28
|
+
def Init(self, buf, pos):
|
|
29
|
+
self._tab = flatbuffers.table.Table(buf, pos)
|
|
30
|
+
|
|
31
|
+
# OptimizerGroup
|
|
32
|
+
def GroupName(self):
|
|
33
|
+
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
34
|
+
if o != 0:
|
|
35
|
+
return self._tab.String(o + self._tab.Pos)
|
|
36
|
+
return None
|
|
37
|
+
|
|
38
|
+
# OptimizerGroup
|
|
39
|
+
def Step(self):
|
|
40
|
+
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(6))
|
|
41
|
+
if o != 0:
|
|
42
|
+
return self._tab.Get(flatbuffers.number_types.Int64Flags, o + self._tab.Pos)
|
|
43
|
+
return 0
|
|
44
|
+
|
|
45
|
+
# OptimizerGroup
|
|
46
|
+
def InitialLearningRate(self):
|
|
47
|
+
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(8))
|
|
48
|
+
if o != 0:
|
|
49
|
+
return self._tab.Get(flatbuffers.number_types.Float32Flags, o + self._tab.Pos)
|
|
50
|
+
return 0.0
|
|
51
|
+
|
|
52
|
+
# OptimizerGroup
|
|
53
|
+
def OptimizerStates(self, j):
|
|
54
|
+
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(10))
|
|
55
|
+
if o != 0:
|
|
56
|
+
x = self._tab.Vector(o)
|
|
57
|
+
x += flatbuffers.number_types.UOffsetTFlags.py_type(j) * 4
|
|
58
|
+
x = self._tab.Indirect(x)
|
|
59
|
+
from ort_flatbuffers_py.fbs.ParameterOptimizerState import ParameterOptimizerState
|
|
60
|
+
obj = ParameterOptimizerState()
|
|
61
|
+
obj.Init(self._tab.Bytes, x)
|
|
62
|
+
return obj
|
|
63
|
+
return None
|
|
64
|
+
|
|
65
|
+
# OptimizerGroup
|
|
66
|
+
def OptimizerStatesLength(self):
|
|
67
|
+
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(10))
|
|
68
|
+
if o != 0:
|
|
69
|
+
return self._tab.VectorLen(o)
|
|
70
|
+
return 0
|
|
71
|
+
|
|
72
|
+
# OptimizerGroup
|
|
73
|
+
def OptimizerStatesIsNone(self):
|
|
74
|
+
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(10))
|
|
75
|
+
return o == 0
|
|
76
|
+
|
|
77
|
+
def OptimizerGroupStart(builder):
|
|
78
|
+
builder.StartObject(4)
|
|
79
|
+
|
|
80
|
+
def Start(builder):
|
|
81
|
+
OptimizerGroupStart(builder)
|
|
82
|
+
|
|
83
|
+
def OptimizerGroupAddGroupName(builder, groupName):
|
|
84
|
+
builder.PrependUOffsetTRelativeSlot(0, flatbuffers.number_types.UOffsetTFlags.py_type(groupName), 0)
|
|
85
|
+
|
|
86
|
+
def AddGroupName(builder, groupName):
|
|
87
|
+
OptimizerGroupAddGroupName(builder, groupName)
|
|
88
|
+
|
|
89
|
+
def OptimizerGroupAddStep(builder, step):
|
|
90
|
+
builder.PrependInt64Slot(1, step, 0)
|
|
91
|
+
|
|
92
|
+
def AddStep(builder, step):
|
|
93
|
+
OptimizerGroupAddStep(builder, step)
|
|
94
|
+
|
|
95
|
+
def OptimizerGroupAddInitialLearningRate(builder, initialLearningRate):
|
|
96
|
+
builder.PrependFloat32Slot(2, initialLearningRate, 0.0)
|
|
97
|
+
|
|
98
|
+
def AddInitialLearningRate(builder, initialLearningRate):
|
|
99
|
+
OptimizerGroupAddInitialLearningRate(builder, initialLearningRate)
|
|
100
|
+
|
|
101
|
+
def OptimizerGroupAddOptimizerStates(builder, optimizerStates):
|
|
102
|
+
builder.PrependUOffsetTRelativeSlot(3, flatbuffers.number_types.UOffsetTFlags.py_type(optimizerStates), 0)
|
|
103
|
+
|
|
104
|
+
def AddOptimizerStates(builder, optimizerStates):
|
|
105
|
+
OptimizerGroupAddOptimizerStates(builder, optimizerStates)
|
|
106
|
+
|
|
107
|
+
def OptimizerGroupStartOptimizerStatesVector(builder, numElems):
|
|
108
|
+
return builder.StartVector(4, numElems, 4)
|
|
109
|
+
|
|
110
|
+
def StartOptimizerStatesVector(builder, numElems: int) -> int:
|
|
111
|
+
return OptimizerGroupStartOptimizerStatesVector(builder, numElems)
|
|
112
|
+
|
|
113
|
+
def OptimizerGroupEnd(builder):
|
|
114
|
+
return builder.EndObject()
|
|
115
|
+
|
|
116
|
+
def End(builder):
|
|
117
|
+
return OptimizerGroupEnd(builder)
|
|
@@ -0,0 +1,91 @@
|
|
|
1
|
+
# automatically generated by the FlatBuffers compiler, do not modify
|
|
2
|
+
|
|
3
|
+
# namespace: fbs
|
|
4
|
+
|
|
5
|
+
import flatbuffers
|
|
6
|
+
from flatbuffers.compat import import_numpy
|
|
7
|
+
np = import_numpy()
|
|
8
|
+
|
|
9
|
+
class ParameterOptimizerState(object):
|
|
10
|
+
__slots__ = ['_tab']
|
|
11
|
+
|
|
12
|
+
@classmethod
|
|
13
|
+
def GetRootAs(cls, buf, offset=0):
|
|
14
|
+
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
15
|
+
x = ParameterOptimizerState()
|
|
16
|
+
x.Init(buf, n + offset)
|
|
17
|
+
return x
|
|
18
|
+
|
|
19
|
+
@classmethod
|
|
20
|
+
def GetRootAsParameterOptimizerState(cls, buf, offset=0):
|
|
21
|
+
"""This method is deprecated. Please switch to GetRootAs."""
|
|
22
|
+
return cls.GetRootAs(buf, offset)
|
|
23
|
+
@classmethod
|
|
24
|
+
def ParameterOptimizerStateBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
25
|
+
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x4F\x44\x54\x43", size_prefixed=size_prefixed)
|
|
26
|
+
|
|
27
|
+
# ParameterOptimizerState
|
|
28
|
+
def Init(self, buf, pos):
|
|
29
|
+
self._tab = flatbuffers.table.Table(buf, pos)
|
|
30
|
+
|
|
31
|
+
# ParameterOptimizerState
|
|
32
|
+
def ParamName(self):
|
|
33
|
+
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
34
|
+
if o != 0:
|
|
35
|
+
return self._tab.String(o + self._tab.Pos)
|
|
36
|
+
return None
|
|
37
|
+
|
|
38
|
+
# ParameterOptimizerState
|
|
39
|
+
def Momentums(self, j):
|
|
40
|
+
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(6))
|
|
41
|
+
if o != 0:
|
|
42
|
+
x = self._tab.Vector(o)
|
|
43
|
+
x += flatbuffers.number_types.UOffsetTFlags.py_type(j) * 4
|
|
44
|
+
x = self._tab.Indirect(x)
|
|
45
|
+
from ort_flatbuffers_py.fbs.Tensor import Tensor
|
|
46
|
+
obj = Tensor()
|
|
47
|
+
obj.Init(self._tab.Bytes, x)
|
|
48
|
+
return obj
|
|
49
|
+
return None
|
|
50
|
+
|
|
51
|
+
# ParameterOptimizerState
|
|
52
|
+
def MomentumsLength(self):
|
|
53
|
+
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(6))
|
|
54
|
+
if o != 0:
|
|
55
|
+
return self._tab.VectorLen(o)
|
|
56
|
+
return 0
|
|
57
|
+
|
|
58
|
+
# ParameterOptimizerState
|
|
59
|
+
def MomentumsIsNone(self):
|
|
60
|
+
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(6))
|
|
61
|
+
return o == 0
|
|
62
|
+
|
|
63
|
+
def ParameterOptimizerStateStart(builder):
|
|
64
|
+
builder.StartObject(2)
|
|
65
|
+
|
|
66
|
+
def Start(builder):
|
|
67
|
+
ParameterOptimizerStateStart(builder)
|
|
68
|
+
|
|
69
|
+
def ParameterOptimizerStateAddParamName(builder, paramName):
|
|
70
|
+
builder.PrependUOffsetTRelativeSlot(0, flatbuffers.number_types.UOffsetTFlags.py_type(paramName), 0)
|
|
71
|
+
|
|
72
|
+
def AddParamName(builder, paramName):
|
|
73
|
+
ParameterOptimizerStateAddParamName(builder, paramName)
|
|
74
|
+
|
|
75
|
+
def ParameterOptimizerStateAddMomentums(builder, momentums):
|
|
76
|
+
builder.PrependUOffsetTRelativeSlot(1, flatbuffers.number_types.UOffsetTFlags.py_type(momentums), 0)
|
|
77
|
+
|
|
78
|
+
def AddMomentums(builder, momentums):
|
|
79
|
+
ParameterOptimizerStateAddMomentums(builder, momentums)
|
|
80
|
+
|
|
81
|
+
def ParameterOptimizerStateStartMomentumsVector(builder, numElems):
|
|
82
|
+
return builder.StartVector(4, numElems, 4)
|
|
83
|
+
|
|
84
|
+
def StartMomentumsVector(builder, numElems: int) -> int:
|
|
85
|
+
return ParameterOptimizerStateStartMomentumsVector(builder, numElems)
|
|
86
|
+
|
|
87
|
+
def ParameterOptimizerStateEnd(builder):
|
|
88
|
+
return builder.EndObject()
|
|
89
|
+
|
|
90
|
+
def End(builder):
|
|
91
|
+
return ParameterOptimizerStateEnd(builder)
|
|
@@ -0,0 +1,152 @@
|
|
|
1
|
+
# automatically generated by the FlatBuffers compiler, do not modify
|
|
2
|
+
|
|
3
|
+
# namespace: fbs
|
|
4
|
+
|
|
5
|
+
import flatbuffers
|
|
6
|
+
from flatbuffers.compat import import_numpy
|
|
7
|
+
np = import_numpy()
|
|
8
|
+
|
|
9
|
+
class PropertyBag(object):
|
|
10
|
+
__slots__ = ['_tab']
|
|
11
|
+
|
|
12
|
+
@classmethod
|
|
13
|
+
def GetRootAs(cls, buf, offset=0):
|
|
14
|
+
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
15
|
+
x = PropertyBag()
|
|
16
|
+
x.Init(buf, n + offset)
|
|
17
|
+
return x
|
|
18
|
+
|
|
19
|
+
@classmethod
|
|
20
|
+
def GetRootAsPropertyBag(cls, buf, offset=0):
|
|
21
|
+
"""This method is deprecated. Please switch to GetRootAs."""
|
|
22
|
+
return cls.GetRootAs(buf, offset)
|
|
23
|
+
@classmethod
|
|
24
|
+
def PropertyBagBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
25
|
+
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x4F\x44\x54\x43", size_prefixed=size_prefixed)
|
|
26
|
+
|
|
27
|
+
# PropertyBag
|
|
28
|
+
def Init(self, buf, pos):
|
|
29
|
+
self._tab = flatbuffers.table.Table(buf, pos)
|
|
30
|
+
|
|
31
|
+
# PropertyBag
|
|
32
|
+
def Ints(self, j):
|
|
33
|
+
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
34
|
+
if o != 0:
|
|
35
|
+
x = self._tab.Vector(o)
|
|
36
|
+
x += flatbuffers.number_types.UOffsetTFlags.py_type(j) * 4
|
|
37
|
+
x = self._tab.Indirect(x)
|
|
38
|
+
from ort_flatbuffers_py.fbs.IntProperty import IntProperty
|
|
39
|
+
obj = IntProperty()
|
|
40
|
+
obj.Init(self._tab.Bytes, x)
|
|
41
|
+
return obj
|
|
42
|
+
return None
|
|
43
|
+
|
|
44
|
+
# PropertyBag
|
|
45
|
+
def IntsLength(self):
|
|
46
|
+
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
47
|
+
if o != 0:
|
|
48
|
+
return self._tab.VectorLen(o)
|
|
49
|
+
return 0
|
|
50
|
+
|
|
51
|
+
# PropertyBag
|
|
52
|
+
def IntsIsNone(self):
|
|
53
|
+
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
54
|
+
return o == 0
|
|
55
|
+
|
|
56
|
+
# PropertyBag
|
|
57
|
+
def Floats(self, j):
|
|
58
|
+
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(6))
|
|
59
|
+
if o != 0:
|
|
60
|
+
x = self._tab.Vector(o)
|
|
61
|
+
x += flatbuffers.number_types.UOffsetTFlags.py_type(j) * 4
|
|
62
|
+
x = self._tab.Indirect(x)
|
|
63
|
+
from ort_flatbuffers_py.fbs.FloatProperty import FloatProperty
|
|
64
|
+
obj = FloatProperty()
|
|
65
|
+
obj.Init(self._tab.Bytes, x)
|
|
66
|
+
return obj
|
|
67
|
+
return None
|
|
68
|
+
|
|
69
|
+
# PropertyBag
|
|
70
|
+
def FloatsLength(self):
|
|
71
|
+
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(6))
|
|
72
|
+
if o != 0:
|
|
73
|
+
return self._tab.VectorLen(o)
|
|
74
|
+
return 0
|
|
75
|
+
|
|
76
|
+
# PropertyBag
|
|
77
|
+
def FloatsIsNone(self):
|
|
78
|
+
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(6))
|
|
79
|
+
return o == 0
|
|
80
|
+
|
|
81
|
+
# PropertyBag
|
|
82
|
+
def Strings(self, j):
|
|
83
|
+
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(8))
|
|
84
|
+
if o != 0:
|
|
85
|
+
x = self._tab.Vector(o)
|
|
86
|
+
x += flatbuffers.number_types.UOffsetTFlags.py_type(j) * 4
|
|
87
|
+
x = self._tab.Indirect(x)
|
|
88
|
+
from ort_flatbuffers_py.fbs.StringProperty import StringProperty
|
|
89
|
+
obj = StringProperty()
|
|
90
|
+
obj.Init(self._tab.Bytes, x)
|
|
91
|
+
return obj
|
|
92
|
+
return None
|
|
93
|
+
|
|
94
|
+
# PropertyBag
|
|
95
|
+
def StringsLength(self):
|
|
96
|
+
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(8))
|
|
97
|
+
if o != 0:
|
|
98
|
+
return self._tab.VectorLen(o)
|
|
99
|
+
return 0
|
|
100
|
+
|
|
101
|
+
# PropertyBag
|
|
102
|
+
def StringsIsNone(self):
|
|
103
|
+
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(8))
|
|
104
|
+
return o == 0
|
|
105
|
+
|
|
106
|
+
def PropertyBagStart(builder):
|
|
107
|
+
builder.StartObject(3)
|
|
108
|
+
|
|
109
|
+
def Start(builder):
|
|
110
|
+
PropertyBagStart(builder)
|
|
111
|
+
|
|
112
|
+
def PropertyBagAddInts(builder, ints):
|
|
113
|
+
builder.PrependUOffsetTRelativeSlot(0, flatbuffers.number_types.UOffsetTFlags.py_type(ints), 0)
|
|
114
|
+
|
|
115
|
+
def AddInts(builder, ints):
|
|
116
|
+
PropertyBagAddInts(builder, ints)
|
|
117
|
+
|
|
118
|
+
def PropertyBagStartIntsVector(builder, numElems):
|
|
119
|
+
return builder.StartVector(4, numElems, 4)
|
|
120
|
+
|
|
121
|
+
def StartIntsVector(builder, numElems: int) -> int:
|
|
122
|
+
return PropertyBagStartIntsVector(builder, numElems)
|
|
123
|
+
|
|
124
|
+
def PropertyBagAddFloats(builder, floats):
|
|
125
|
+
builder.PrependUOffsetTRelativeSlot(1, flatbuffers.number_types.UOffsetTFlags.py_type(floats), 0)
|
|
126
|
+
|
|
127
|
+
def AddFloats(builder, floats):
|
|
128
|
+
PropertyBagAddFloats(builder, floats)
|
|
129
|
+
|
|
130
|
+
def PropertyBagStartFloatsVector(builder, numElems):
|
|
131
|
+
return builder.StartVector(4, numElems, 4)
|
|
132
|
+
|
|
133
|
+
def StartFloatsVector(builder, numElems: int) -> int:
|
|
134
|
+
return PropertyBagStartFloatsVector(builder, numElems)
|
|
135
|
+
|
|
136
|
+
def PropertyBagAddStrings(builder, strings):
|
|
137
|
+
builder.PrependUOffsetTRelativeSlot(2, flatbuffers.number_types.UOffsetTFlags.py_type(strings), 0)
|
|
138
|
+
|
|
139
|
+
def AddStrings(builder, strings):
|
|
140
|
+
PropertyBagAddStrings(builder, strings)
|
|
141
|
+
|
|
142
|
+
def PropertyBagStartStringsVector(builder, numElems):
|
|
143
|
+
return builder.StartVector(4, numElems, 4)
|
|
144
|
+
|
|
145
|
+
def StartStringsVector(builder, numElems: int) -> int:
|
|
146
|
+
return PropertyBagStartStringsVector(builder, numElems)
|
|
147
|
+
|
|
148
|
+
def PropertyBagEnd(builder):
|
|
149
|
+
return builder.EndObject()
|
|
150
|
+
|
|
151
|
+
def End(builder):
|
|
152
|
+
return PropertyBagEnd(builder)
|
|
@@ -0,0 +1,105 @@
|
|
|
1
|
+
# automatically generated by the FlatBuffers compiler, do not modify
|
|
2
|
+
|
|
3
|
+
# namespace: fbs
|
|
4
|
+
|
|
5
|
+
import flatbuffers
|
|
6
|
+
from flatbuffers.compat import import_numpy
|
|
7
|
+
np = import_numpy()
|
|
8
|
+
|
|
9
|
+
# a single runtime optimization
|
|
10
|
+
# see corresponding type in onnxruntime/core/graph/runtime_optimization_record.h
|
|
11
|
+
class RuntimeOptimizationRecord(object):
|
|
12
|
+
__slots__ = ['_tab']
|
|
13
|
+
|
|
14
|
+
@classmethod
|
|
15
|
+
def GetRootAs(cls, buf, offset=0):
|
|
16
|
+
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
17
|
+
x = RuntimeOptimizationRecord()
|
|
18
|
+
x.Init(buf, n + offset)
|
|
19
|
+
return x
|
|
20
|
+
|
|
21
|
+
@classmethod
|
|
22
|
+
def GetRootAsRuntimeOptimizationRecord(cls, buf, offset=0):
|
|
23
|
+
"""This method is deprecated. Please switch to GetRootAs."""
|
|
24
|
+
return cls.GetRootAs(buf, offset)
|
|
25
|
+
@classmethod
|
|
26
|
+
def RuntimeOptimizationRecordBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
27
|
+
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x4F\x52\x54\x4D", size_prefixed=size_prefixed)
|
|
28
|
+
|
|
29
|
+
# RuntimeOptimizationRecord
|
|
30
|
+
def Init(self, buf, pos):
|
|
31
|
+
self._tab = flatbuffers.table.Table(buf, pos)
|
|
32
|
+
|
|
33
|
+
# RuntimeOptimizationRecord
|
|
34
|
+
def ActionId(self):
|
|
35
|
+
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
36
|
+
if o != 0:
|
|
37
|
+
return self._tab.String(o + self._tab.Pos)
|
|
38
|
+
return None
|
|
39
|
+
|
|
40
|
+
# RuntimeOptimizationRecord
|
|
41
|
+
def NodesToOptimizeIndices(self):
|
|
42
|
+
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(6))
|
|
43
|
+
if o != 0:
|
|
44
|
+
x = self._tab.Indirect(o + self._tab.Pos)
|
|
45
|
+
from ort_flatbuffers_py.fbs.NodesToOptimizeIndices import NodesToOptimizeIndices
|
|
46
|
+
obj = NodesToOptimizeIndices()
|
|
47
|
+
obj.Init(self._tab.Bytes, x)
|
|
48
|
+
return obj
|
|
49
|
+
return None
|
|
50
|
+
|
|
51
|
+
# RuntimeOptimizationRecord
|
|
52
|
+
def ProducedOpIds(self, j):
|
|
53
|
+
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(10))
|
|
54
|
+
if o != 0:
|
|
55
|
+
a = self._tab.Vector(o)
|
|
56
|
+
return self._tab.String(a + flatbuffers.number_types.UOffsetTFlags.py_type(j * 4))
|
|
57
|
+
return ""
|
|
58
|
+
|
|
59
|
+
# RuntimeOptimizationRecord
|
|
60
|
+
def ProducedOpIdsLength(self):
|
|
61
|
+
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(10))
|
|
62
|
+
if o != 0:
|
|
63
|
+
return self._tab.VectorLen(o)
|
|
64
|
+
return 0
|
|
65
|
+
|
|
66
|
+
# RuntimeOptimizationRecord
|
|
67
|
+
def ProducedOpIdsIsNone(self):
|
|
68
|
+
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(10))
|
|
69
|
+
return o == 0
|
|
70
|
+
|
|
71
|
+
def RuntimeOptimizationRecordStart(builder):
|
|
72
|
+
builder.StartObject(4)
|
|
73
|
+
|
|
74
|
+
def Start(builder):
|
|
75
|
+
RuntimeOptimizationRecordStart(builder)
|
|
76
|
+
|
|
77
|
+
def RuntimeOptimizationRecordAddActionId(builder, actionId):
|
|
78
|
+
builder.PrependUOffsetTRelativeSlot(0, flatbuffers.number_types.UOffsetTFlags.py_type(actionId), 0)
|
|
79
|
+
|
|
80
|
+
def AddActionId(builder, actionId):
|
|
81
|
+
RuntimeOptimizationRecordAddActionId(builder, actionId)
|
|
82
|
+
|
|
83
|
+
def RuntimeOptimizationRecordAddNodesToOptimizeIndices(builder, nodesToOptimizeIndices):
|
|
84
|
+
builder.PrependUOffsetTRelativeSlot(1, flatbuffers.number_types.UOffsetTFlags.py_type(nodesToOptimizeIndices), 0)
|
|
85
|
+
|
|
86
|
+
def AddNodesToOptimizeIndices(builder, nodesToOptimizeIndices):
|
|
87
|
+
RuntimeOptimizationRecordAddNodesToOptimizeIndices(builder, nodesToOptimizeIndices)
|
|
88
|
+
|
|
89
|
+
def RuntimeOptimizationRecordAddProducedOpIds(builder, producedOpIds):
|
|
90
|
+
builder.PrependUOffsetTRelativeSlot(3, flatbuffers.number_types.UOffsetTFlags.py_type(producedOpIds), 0)
|
|
91
|
+
|
|
92
|
+
def AddProducedOpIds(builder, producedOpIds):
|
|
93
|
+
RuntimeOptimizationRecordAddProducedOpIds(builder, producedOpIds)
|
|
94
|
+
|
|
95
|
+
def RuntimeOptimizationRecordStartProducedOpIdsVector(builder, numElems):
|
|
96
|
+
return builder.StartVector(4, numElems, 4)
|
|
97
|
+
|
|
98
|
+
def StartProducedOpIdsVector(builder, numElems: int) -> int:
|
|
99
|
+
return RuntimeOptimizationRecordStartProducedOpIdsVector(builder, numElems)
|
|
100
|
+
|
|
101
|
+
def RuntimeOptimizationRecordEnd(builder):
|
|
102
|
+
return builder.EndObject()
|
|
103
|
+
|
|
104
|
+
def End(builder):
|
|
105
|
+
return RuntimeOptimizationRecordEnd(builder)
|
onnxruntime/tools/ort_format_model/ort_flatbuffers_py/fbs/RuntimeOptimizationRecordContainerEntry.py
ADDED
|
@@ -0,0 +1,91 @@
|
|
|
1
|
+
# automatically generated by the FlatBuffers compiler, do not modify
|
|
2
|
+
|
|
3
|
+
# namespace: fbs
|
|
4
|
+
|
|
5
|
+
import flatbuffers
|
|
6
|
+
from flatbuffers.compat import import_numpy
|
|
7
|
+
np = import_numpy()
|
|
8
|
+
|
|
9
|
+
class RuntimeOptimizationRecordContainerEntry(object):
|
|
10
|
+
__slots__ = ['_tab']
|
|
11
|
+
|
|
12
|
+
@classmethod
|
|
13
|
+
def GetRootAs(cls, buf, offset=0):
|
|
14
|
+
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
15
|
+
x = RuntimeOptimizationRecordContainerEntry()
|
|
16
|
+
x.Init(buf, n + offset)
|
|
17
|
+
return x
|
|
18
|
+
|
|
19
|
+
@classmethod
|
|
20
|
+
def GetRootAsRuntimeOptimizationRecordContainerEntry(cls, buf, offset=0):
|
|
21
|
+
"""This method is deprecated. Please switch to GetRootAs."""
|
|
22
|
+
return cls.GetRootAs(buf, offset)
|
|
23
|
+
@classmethod
|
|
24
|
+
def RuntimeOptimizationRecordContainerEntryBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
25
|
+
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x4F\x52\x54\x4D", size_prefixed=size_prefixed)
|
|
26
|
+
|
|
27
|
+
# RuntimeOptimizationRecordContainerEntry
|
|
28
|
+
def Init(self, buf, pos):
|
|
29
|
+
self._tab = flatbuffers.table.Table(buf, pos)
|
|
30
|
+
|
|
31
|
+
# RuntimeOptimizationRecordContainerEntry
|
|
32
|
+
def OptimizerName(self):
|
|
33
|
+
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
34
|
+
if o != 0:
|
|
35
|
+
return self._tab.String(o + self._tab.Pos)
|
|
36
|
+
return None
|
|
37
|
+
|
|
38
|
+
# RuntimeOptimizationRecordContainerEntry
|
|
39
|
+
def RuntimeOptimizationRecords(self, j):
|
|
40
|
+
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(6))
|
|
41
|
+
if o != 0:
|
|
42
|
+
x = self._tab.Vector(o)
|
|
43
|
+
x += flatbuffers.number_types.UOffsetTFlags.py_type(j) * 4
|
|
44
|
+
x = self._tab.Indirect(x)
|
|
45
|
+
from ort_flatbuffers_py.fbs.RuntimeOptimizationRecord import RuntimeOptimizationRecord
|
|
46
|
+
obj = RuntimeOptimizationRecord()
|
|
47
|
+
obj.Init(self._tab.Bytes, x)
|
|
48
|
+
return obj
|
|
49
|
+
return None
|
|
50
|
+
|
|
51
|
+
# RuntimeOptimizationRecordContainerEntry
|
|
52
|
+
def RuntimeOptimizationRecordsLength(self):
|
|
53
|
+
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(6))
|
|
54
|
+
if o != 0:
|
|
55
|
+
return self._tab.VectorLen(o)
|
|
56
|
+
return 0
|
|
57
|
+
|
|
58
|
+
# RuntimeOptimizationRecordContainerEntry
|
|
59
|
+
def RuntimeOptimizationRecordsIsNone(self):
|
|
60
|
+
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(6))
|
|
61
|
+
return o == 0
|
|
62
|
+
|
|
63
|
+
def RuntimeOptimizationRecordContainerEntryStart(builder):
|
|
64
|
+
builder.StartObject(2)
|
|
65
|
+
|
|
66
|
+
def Start(builder):
|
|
67
|
+
RuntimeOptimizationRecordContainerEntryStart(builder)
|
|
68
|
+
|
|
69
|
+
def RuntimeOptimizationRecordContainerEntryAddOptimizerName(builder, optimizerName):
|
|
70
|
+
builder.PrependUOffsetTRelativeSlot(0, flatbuffers.number_types.UOffsetTFlags.py_type(optimizerName), 0)
|
|
71
|
+
|
|
72
|
+
def AddOptimizerName(builder, optimizerName):
|
|
73
|
+
RuntimeOptimizationRecordContainerEntryAddOptimizerName(builder, optimizerName)
|
|
74
|
+
|
|
75
|
+
def RuntimeOptimizationRecordContainerEntryAddRuntimeOptimizationRecords(builder, runtimeOptimizationRecords):
|
|
76
|
+
builder.PrependUOffsetTRelativeSlot(1, flatbuffers.number_types.UOffsetTFlags.py_type(runtimeOptimizationRecords), 0)
|
|
77
|
+
|
|
78
|
+
def AddRuntimeOptimizationRecords(builder, runtimeOptimizationRecords):
|
|
79
|
+
RuntimeOptimizationRecordContainerEntryAddRuntimeOptimizationRecords(builder, runtimeOptimizationRecords)
|
|
80
|
+
|
|
81
|
+
def RuntimeOptimizationRecordContainerEntryStartRuntimeOptimizationRecordsVector(builder, numElems):
|
|
82
|
+
return builder.StartVector(4, numElems, 4)
|
|
83
|
+
|
|
84
|
+
def StartRuntimeOptimizationRecordsVector(builder, numElems: int) -> int:
|
|
85
|
+
return RuntimeOptimizationRecordContainerEntryStartRuntimeOptimizationRecordsVector(builder, numElems)
|
|
86
|
+
|
|
87
|
+
def RuntimeOptimizationRecordContainerEntryEnd(builder):
|
|
88
|
+
return builder.EndObject()
|
|
89
|
+
|
|
90
|
+
def End(builder):
|
|
91
|
+
return RuntimeOptimizationRecordContainerEntryEnd(builder)
|