3rdparty/cnpy/cnpy.h

31 …NpyArray(const std::vector<size_t>& _shape, size_t _word_size, bool _fortran_order, std::string _t…  in NpyArray()
40     data_holder = std::shared_ptr<std::vector<char>>(new std::vector<char>(num_vals * word_size));  in NpyArray()
63   std::vector<T> as_vec() const  in as_vec()
66     return std::vector<T>(p, p + num_vals);  in as_vec()
71   std::shared_ptr<std::vector<char>> data_holder;
72   std::vector<size_t>                shape;
76   std::string                        typeName;
79 using npz_t = std::map<std::string, NpyArray>;
82 char map_type(const std::type_info& t);
84 std::vector<char> create_npy_header(const std::vector<size_t>& shape);
85 void              parse_npy_header(FILE* fp, size_t& word_size, std::vector<size_t>& shape, bool& f…
86                                    std::string& typeName);
87 void     parse_npy_header(unsigned char* buffer, size_t& word_size, std::vector<size_t>& shape, boo…
88                           std::string& typeName);
90 npz_t    npz_load(std::string fname);
91 NpyArray npz_load(std::string fname, std::string varname);
92 NpyArray npy_load(std::string fname);
95 std::vector<char>& operator+=(std::vector<char>& lhs, const T rhs)
106 std::vector<char>& operator+=(std::vector<char>& lhs, const std::string rhs);
108 std::vector<char>& operator+=(std::vector<char>& lhs, const char* rhs);
111 int npy_save(std::string fname, const T* data, const std::vector<size_t> shape, std::string mode = …
113   std::ofstream ofs(fname, std::ios::out);
119   std::vector<size_t> true_data_shape; // if appending, the shape of existing + new data
128     std::string typeName;
133 …std::cout << "libnpy error: " << fname << " has word size " << word_size << " but npy_save appendi…
138 …std::cout << "libnpy error: npy_save attempting to append misdimensioned data to " << fname << "\n…
144 …    std::cout << "libnpy error: npy_save attempting to append misshaped data to " << fname << "\n";
154   std::vector<char> header = create_npy_header<T>(true_data_shape);
155 …size_t            nels   = std::accumulate(shape.begin(), shape.end(), 1, std::multiplies<size_t>(…
166 void npz_save(std::string zipname, std::string fname, const T* data, const std::vector<size_t>& sha…
167               std::string mode = "w")
176   std::vector<char> global_header;
193       throw std::runtime_error("npz_save: header read error while adding to existing zip");
200   std::vector<char> npy_header = create_npy_header<T>(shape);
202   size_t nels   = std::accumulate(shape.begin(), shape.end(), 1, std::multiplies<size_t>());
214   std::vector<char> local_header;
243   std::vector<char> footer;
266 void npy_save(std::string fname, const std::vector<T> data, std::string mode = "w")
268   std::vector<size_t> shape;
274 void npz_save(std::string zipname, std::string fname, const std::vector<T> data, std::string mode =…
276   std::vector<size_t> shape;
282 std::vector<char> create_npy_header(const std::vector<size_t>& shape)  in create_npy_header()
285   std::vector<char> dict;  in create_npy_header()
288   if (std::string(tpye_name) == "N4rknn7float16E") {  in create_npy_header()
293   dict += std::to_string(sizeof(T));  in create_npy_header()
295   dict += std::to_string(shape[0]);  in create_npy_header()
298     dict += std::to_string(shape[i]);  in create_npy_header()
308   std::vector<char> header;  in create_npy_header()