From e097e4e93c599d2a81346d6da5979ea5068d78e6 Mon Sep 17 00:00:00 2001 From: Albert Grigoryan <52666717+albert-grigoryan@users.noreply.github.com> Date: Sat, 25 Feb 2023 11:25:16 +0400 Subject: [PATCH] Objective-C lib: Added support for int64 and uint64. (#14405) ### Description Added support for int64 and uint64 in Objective-C lib. ### Motivation and Context Int64 is rarely used, but we needed it. The Int64 inference worked after the change (tested). --- objectivec/include/ort_enums.h | 2 ++ objectivec/src/ort_enums.mm | 2 ++ 2 files changed, 4 insertions(+) diff --git a/objectivec/include/ort_enums.h b/objectivec/include/ort_enums.h index e82765b535..d505ce2c4e 100644 --- a/objectivec/include/ort_enums.h +++ b/objectivec/include/ort_enums.h @@ -36,6 +36,8 @@ typedef NS_ENUM(int32_t, ORTTensorElementDataType) { ORTTensorElementDataTypeUInt8, ORTTensorElementDataTypeInt32, ORTTensorElementDataTypeUInt32, + ORTTensorElementDataTypeInt64, + ORTTensorElementDataTypeUInt64, }; /** diff --git a/objectivec/src/ort_enums.mm b/objectivec/src/ort_enums.mm index 1a7d63decd..82191b7801 100644 --- a/objectivec/src/ort_enums.mm +++ b/objectivec/src/ort_enums.mm @@ -51,6 +51,8 @@ constexpr TensorElementTypeInfo kElementTypeInfos[]{ {ORTTensorElementDataTypeUInt8, ONNX_TENSOR_ELEMENT_DATA_TYPE_UINT8, sizeof(uint8_t)}, {ORTTensorElementDataTypeInt32, ONNX_TENSOR_ELEMENT_DATA_TYPE_INT32, sizeof(int32_t)}, {ORTTensorElementDataTypeUInt32, ONNX_TENSOR_ELEMENT_DATA_TYPE_UINT32, sizeof(uint32_t)}, + {ORTTensorElementDataTypeInt64, ONNX_TENSOR_ELEMENT_DATA_TYPE_INT64, sizeof(int64_t)}, + {ORTTensorElementDataTypeUInt64, ONNX_TENSOR_ELEMENT_DATA_TYPE_UINT64, sizeof(uint64_t)}, }; struct GraphOptimizationLevelInfo {