BYTE_SIZE_MAP

Function

Provides the data type size.

Format

const std::unordered_map<InferDataType, size_t> BYTE_SIZE_MAP = {
    {InferDataType::TYPE_INVALID, 0},
    {InferDataType::TYPE_BOOL, sizeof(bool)},
    {InferDataType::TYPE_UINT8, sizeof(uint8_t)},
    {InferDataType::TYPE_UINT16, sizeof(uint16_t)},
    {InferDataType::TYPE_UINT32, sizeof(uint32_t)},
    {InferDataType::TYPE_UINT64, sizeof(uint64_t)},
    {InferDataType::TYPE_INT8, sizeof(int8_t)},
    {InferDataType::TYPE_INT16, sizeof(int16_t)},
    {InferDataType::TYPE_INT32, sizeof(int32_t)},
    {InferDataType::TYPE_INT64, sizeof(int64_t)},
    {InferDataType::TYPE_FP16, sizeof(int16_t)},    // The float16 type may not be supported.
    {InferDataType::TYPE_FP32, sizeof(float)},
    {InferDataType::TYPE_FP64, sizeof(double)},
    {InferDataType::TYPE_STRING, 0},                // The length is uncertain.
    {InferDataType::TYPE_BF16, sizeof(int16_t)},    // The bfloat16 type may not be supported.
};