diff options
author | Tai Ly <tai.ly@arm.com> | 2023-04-04 20:34:05 +0000 |
---|---|---|
committer | Tai Ly <tai.ly@arm.com> | 2023-04-04 21:07:10 +0000 |
commit | 3ef34fb300e7f62bdb397c605ab6c3bd30682cf8 (patch) | |
tree | 0cd4e9c64fd393f6ba23b25f31b379dae7652049 /include | |
parent | dce6cebbeb6c45625c4ef8fafb5a7775319101c5 (diff) | |
download | serialization_lib-3ef34fb300e7f62bdb397c605ab6c3bd30682cf8.tar.gz |
Add readFromNpyFile and writeToNpyFile for double data
Signed-off-by: Tai Ly <tai.ly@arm.com>
Change-Id: Icc023cbe6aa8843cc37d25e740bc6ce05bb7abb2
Diffstat (limited to 'include')
-rw-r--r-- | include/numpy_utils.h | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/include/numpy_utils.h b/include/numpy_utils.h index 6a20eb3..8c2ed71 100644 --- a/include/numpy_utils.h +++ b/include/numpy_utils.h @@ -41,6 +41,8 @@ public: static NPError readFromNpyFile(const char* filename, const uint32_t elems, float* databuf); + static NPError readFromNpyFile(const char* filename, const uint32_t elems, double* databuf); + static NPError readFromNpyFile(const char* filename, const uint32_t elems, half_float::half* databuf); static NPError readFromNpyFile(const char* filename, const uint32_t elems, int32_t* databuf); @@ -68,6 +70,10 @@ public: static NPError writeToNpyFile(const char* filename, const uint32_t elems, const float* databuf); + static NPError writeToNpyFile(const char* filename, const std::vector<int32_t>& shape, const double* databuf); + + static NPError writeToNpyFile(const char* filename, const uint32_t elems, const double* databuf); + private: static NPError writeToNpyFileCommon(const char* filename, const char* dtype_str, |