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 {