Enum for Tensor scalar type and shape dynamism.

Differential Revision: D71752748

Pull Request resolved: #9609

Author: shoumikhin

extension/apple/ExecuTorch/Exported/ExecuTorchTensor.h

@@ -10,6 +10,57 @@
 
 NS_ASSUME_NONNULL_BEGIN
 
+/**
+ * Enum to define the data type of a Tensor.
+ * Values can be a subset, but must numerically match exactly those defined in
+ * runtime/core/portable_type/scalar_type.h
+ */
+typedef NS_ENUM(int8_t, ExecuTorchDataType) {
+  ExecuTorchDataTypeByte,
+  ExecuTorchDataTypeChar,
+  ExecuTorchDataTypeShort,
+  ExecuTorchDataTypeInt,
+  ExecuTorchDataTypeLong,
+  ExecuTorchDataTypeHalf,
+  ExecuTorchDataTypeFloat,
+  ExecuTorchDataTypeDouble,
+  ExecuTorchDataTypeComplexHalf,
+  ExecuTorchDataTypeComplexFloat,
+  ExecuTorchDataTypeComplexDouble,
+  ExecuTorchDataTypeBool,
+  ExecuTorchDataTypeQInt8,
+  ExecuTorchDataTypeQUInt8,
+  ExecuTorchDataTypeQInt32,
+  ExecuTorchDataTypeBFloat16,
+  ExecuTorchDataTypeQUInt4x2,
+  ExecuTorchDataTypeQUInt2x4,
+  ExecuTorchDataTypeBits1x8,
+  ExecuTorchDataTypeBits2x4,
+  ExecuTorchDataTypeBits4x2,
+  ExecuTorchDataTypeBits8,
+  ExecuTorchDataTypeBits16,
+  ExecuTorchDataTypeFloat8_e5m2,
+  ExecuTorchDataTypeFloat8_e4m3fn,
+  ExecuTorchDataTypeFloat8_e5m2fnuz,
+  ExecuTorchDataTypeFloat8_e4m3fnuz,
+  ExecuTorchDataTypeUInt16,
+  ExecuTorchDataTypeUInt32,
+  ExecuTorchDataTypeUInt64,
+  ExecuTorchDataTypeUndefined,
+  ExecuTorchDataTypeNumOptions,
+} NS_SWIFT_NAME(DataType);
+
+/**
+ * Enum to define the shape dynamism of a Tensor.
+ * Values can be a subset, but must numerically match exactly those defined in
+ * runtime/core/tensor_shape_dynamism.h
+ */
+typedef NS_ENUM(uint8_t, ExecuTorchShapeDynamism) {
+  ExecuTorchShapeDynamismStatic,
+  ExecuTorchShapeDynamismDynamicBound,
+  ExecuTorchShapeDynamismDynamicUnbound,
+} NS_SWIFT_NAME(ShapeDynamism);
+
 NS_SWIFT_NAME(Tensor)
 __attribute__((deprecated("This API is experimental.")))
 @interface ExecuTorchTensor : NSObject