diff options
Diffstat (limited to 'include/tosa_generated.h')
-rw-r--r-- | include/tosa_generated.h | 13 |
1 files changed, 8 insertions, 5 deletions
diff --git a/include/tosa_generated.h b/include/tosa_generated.h index 2b9d0ea..f36ed37 100644 --- a/include/tosa_generated.h +++ b/include/tosa_generated.h @@ -95,11 +95,12 @@ enum DType : uint32_t { DType_FP32 = 8, DType_UINT16 = 9, DType_FP16 = 10, + DType_BF16 = 11, DType_MIN = DType_UNKNOWN, - DType_MAX = DType_FP16 + DType_MAX = DType_BF16 }; -inline const DType (&EnumValuesDType())[11] { +inline const DType (&EnumValuesDType())[12] { static const DType values[] = { DType_UNKNOWN, DType_BOOL, @@ -111,13 +112,14 @@ inline const DType (&EnumValuesDType())[11] { DType_INT48, DType_FP32, DType_UINT16, - DType_FP16 + DType_FP16, + DType_BF16 }; return values; } inline const char * const *EnumNamesDType() { - static const char * const names[12] = { + static const char * const names[13] = { "UNKNOWN", "BOOL", "UINT8", @@ -129,13 +131,14 @@ inline const char * const *EnumNamesDType() { "FP32", "UINT16", "FP16", + "BF16", nullptr }; return names; } inline const char *EnumNameDType(DType e) { - if (flatbuffers::IsOutRange(e, DType_UNKNOWN, DType_FP16)) return ""; + if (flatbuffers::IsOutRange(e, DType_UNKNOWN, DType_BF16)) return ""; const size_t index = static_cast<size_t>(e); return EnumNamesDType()[index]; } |