下载
中文
注册

BYTE_SIZE_MAP

枚举类功能

提供数据类型大小。

枚举类格式

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)},    // float16 类型不一定支持
    {InferDataType::TYPE_FP32, sizeof(float)},
    {InferDataType::TYPE_FP64, sizeof(double)},
    {InferDataType::TYPE_STRING, 0},                // 长度不确定
    {InferDataType::TYPE_BF16, sizeof(int16_t)},    // bfloat16 类型不一定支持
};