diff options
author | Kevin Cheng <kevin.cheng@arm.com> | 2021-07-28 13:42:29 -0700 |
---|---|---|
committer | Kevin Cheng <kevin.cheng@arm.com> | 2021-08-02 11:11:08 -0700 |
commit | 3ce563449c1e607b016b82c5dbb6e33883f846a5 (patch) | |
tree | 1af52ab625cebe46632c27631005d423d79b7529 /include | |
parent | 82dbb32980a58889bef28b7ad653c30694364195 (diff) | |
download | serialization_lib-3ce563449c1e607b016b82c5dbb6e33883f846a5.tar.gz |
Support I4 weights pack/unpack.
Signed-off-by: Kevin Cheng <kevin.cheng@arm.com>
Change-Id: Ia7d2bfaa473c8a92c71f075c86aca0a275245f83
Diffstat (limited to 'include')
-rw-r--r-- | include/tosa_serialization_handler.h | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/include/tosa_serialization_handler.h b/include/tosa_serialization_handler.h index db9481b..73254dd 100644 --- a/include/tosa_serialization_handler.h +++ b/include/tosa_serialization_handler.h @@ -291,6 +291,7 @@ public: static tosa_err_t ConvertI32toU8(const std::vector<int32_t>& in, std::vector<uint8_t>& out); static tosa_err_t ConvertI16toU8(const std::vector<int16_t>& in, std::vector<uint8_t>& out); static tosa_err_t ConvertI8toU8(const std::vector<int8_t>& in, std::vector<uint8_t>& out); + 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 ConvertU8toF32(const std::vector<uint8_t>& in, uint32_t out_size, std::vector<float>& out); @@ -298,6 +299,7 @@ public: static tosa_err_t ConvertU8toI32(const std::vector<uint8_t>& in, uint32_t out_size, std::vector<int32_t>& out); static tosa_err_t ConvertU8toI16(const std::vector<uint8_t>& in, uint32_t out_size, std::vector<int16_t>& out); static tosa_err_t ConvertU8toI8(const std::vector<uint8_t>& in, uint32_t out_size, std::vector<int8_t>& out); + static tosa_err_t ConvertU8toI4(const std::vector<uint8_t>& in, uint32_t out_size, std::vector<int8_t>& out); static tosa_err_t ConvertU8toBool(const std::vector<uint8_t>& in, uint32_t out_size, std::vector<bool>& out); // version |