diff options
| -rw-r--r-- | include/tosa_generated.h | 13 | ||||
| -rw-r--r-- | python/serializer/tosa_serializer.py | 1 | ||||
| -rw-r--r-- | python/tosa/DType.py | 1 | ||||
| -rw-r--r-- | schema/tosa.fbs | 1 |
4 files changed, 11 insertions, 5 deletions
diff --git a/include/tosa_generated.h b/include/tosa_generated.h index 3bc1546..c30a04d 100644 --- a/include/tosa_generated.h +++ b/include/tosa_generated.h @@ -96,11 +96,12 @@ enum DType : uint32_t { DType_INT32 = 6, DType_INT48 = 7, DType_FLOAT = 8, + DType_UINT16 = 9, DType_MIN = DType_UNKNOWN, - DType_MAX = DType_FLOAT + DType_MAX = DType_UINT16 }; -inline const DType (&EnumValuesDType())[9] { +inline const DType (&EnumValuesDType())[10] { static const DType values[] = { DType_UNKNOWN, DType_BOOL, @@ -110,13 +111,14 @@ inline const DType (&EnumValuesDType())[9] { DType_INT16, DType_INT32, DType_INT48, - DType_FLOAT + DType_FLOAT, + DType_UINT16 }; return values; } inline const char * const *EnumNamesDType() { - static const char * const names[10] = { + static const char * const names[11] = { "UNKNOWN", "BOOL", "UINT8", @@ -126,13 +128,14 @@ inline const char * const *EnumNamesDType() { "INT32", "INT48", "FLOAT", + "UINT16", nullptr }; return names; } inline const char *EnumNameDType(DType e) { - if (flatbuffers::IsOutRange(e, DType_UNKNOWN, DType_FLOAT)) return ""; + if (flatbuffers::IsOutRange(e, DType_UNKNOWN, DType_UINT16)) return ""; const size_t index = static_cast<size_t>(e); return EnumNamesDType()[index]; } diff --git a/python/serializer/tosa_serializer.py b/python/serializer/tosa_serializer.py index b29f963..660619b 100644 --- a/python/serializer/tosa_serializer.py +++ b/python/serializer/tosa_serializer.py @@ -53,6 +53,7 @@ DTypeNames = [ "INT32", "INT48", "FLOAT", + "UINT16", ] ByteMask = np.uint64(0xFF) diff --git a/python/tosa/DType.py b/python/tosa/DType.py index 65432db..ad9048b 100644 --- a/python/tosa/DType.py +++ b/python/tosa/DType.py @@ -12,4 +12,5 @@ class DType(object): INT32 = 6 INT48 = 7 FLOAT = 8 + UINT16 = 9 diff --git a/schema/tosa.fbs b/schema/tosa.fbs index 33075ed..4267adf 100644 --- a/schema/tosa.fbs +++ b/schema/tosa.fbs @@ -30,6 +30,7 @@ enum DType:uint32 { INT32, INT48, FLOAT, + UINT16, } enum ResizeMode:uint32 { |