diff options
-rw-r--r-- | include/tosa_serialization_handler.h | 3 | ||||
-rw-r--r-- | src/tosa_serialization_handler.cpp | 8 |
2 files changed, 6 insertions, 5 deletions
diff --git a/include/tosa_serialization_handler.h b/include/tosa_serialization_handler.h index f50d7ad..5c53f57 100644 --- a/include/tosa_serialization_handler.h +++ b/include/tosa_serialization_handler.h @@ -421,7 +421,8 @@ public: static tosa_err_t ConvertI4toU8(const std::vector<int8_t>& in, std::vector<uint8_t>& out); static tosa_err_t ConvertBooltoU8(const std::vector<bool>& in, std::vector<uint8_t>& out); - static tosa_err_t ConvertU8toF16(const std::vector<uint8_t>& in, uint32_t out_size, std::vector<float>& out); + static tosa_err_t + ConvertU8toF16(const std::vector<uint8_t>& in, uint32_t out_size, std::vector<half_float::half>& out); static tosa_err_t ConvertU8toF32(const std::vector<uint8_t>& in, uint32_t out_size, std::vector<float>& out); static tosa_err_t ConvertU8toI64(const std::vector<uint8_t>& in, uint32_t out_size, std::vector<int64_t>& out); static tosa_err_t ConvertU8toI48(const std::vector<uint8_t>& in, uint32_t out_size, std::vector<int64_t>& out); diff --git a/src/tosa_serialization_handler.cpp b/src/tosa_serialization_handler.cpp index 453670f..749a3c8 100644 --- a/src/tosa_serialization_handler.cpp +++ b/src/tosa_serialization_handler.cpp @@ -896,8 +896,9 @@ tosa_err_t TosaSerializationHandler::ConvertBooltoU8(const std::vector<bool>& in return TOSA_OK; } -tosa_err_t - TosaSerializationHandler::ConvertU8toF16(const std::vector<uint8_t>& in, uint32_t out_size, std::vector<float>& out) +tosa_err_t TosaSerializationHandler::ConvertU8toF16(const std::vector<uint8_t>& in, + uint32_t out_size, + std::vector<half_float::half>& out) { // Note: fp16 values returned in fp32 type out.clear(); @@ -916,8 +917,7 @@ tosa_err_t // Reinterpret u16 byte as fp16 then convert to fp32 half_float::half val_f16 = *(half_float::half*)&val_u16; - float val_fp32 = half_float::half_cast<float, half_float::half>(val_f16); - out.push_back(val_fp32); + out.push_back(val_f16); } return TOSA_OK; } |