Lines Matching refs:shape
32 : shape(_shape) in NpyArray()
38 for (size_t i = 0; i < shape.size(); i++) in NpyArray()
39 num_vals *= shape[i]; in NpyArray()
44 : shape(0) in NpyArray()
72 std::vector<size_t> shape; member
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…
87 void parse_npy_header(unsigned char* buffer, size_t& word_size, std::vector<size_t>& shape, boo…
111 int npy_save(std::string fname, const T* data, const std::vector<size_t> shape, std::string mode = …
137 if (true_data_shape.size() != shape.size()) {
139 assert(true_data_shape.size() != shape.size());
142 for (size_t i = 1; i < shape.size(); i++) {
143 if (shape[i] != true_data_shape[i]) {
145 assert(shape[i] == true_data_shape[i]);
148 true_data_shape[0] += shape[0];
151 true_data_shape = shape;
155 …size_t nels = std::accumulate(shape.begin(), shape.end(), 1, std::multiplies<size_t>(…
166 …d npz_save(std::string zipname, std::string fname, const T* data, const std::vector<size_t>& shape,
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>());
268 std::vector<size_t> shape; variable
269 shape.push_back(data.size());
270 npy_save(fname, &data[0], shape, mode);
276 std::vector<size_t> shape; variable
277 shape.push_back(data.size());
278 npz_save(zipname, fname, &data[0], shape, mode);
282 std::vector<char> create_npy_header(const std::vector<size_t>& shape) in create_npy_header() argument
295 dict += std::to_string(shape[0]); in create_npy_header()
296 for (size_t i = 1; i < shape.size(); i++) { in create_npy_header()
298 dict += std::to_string(shape[i]); in create_npy_header()
300 if (shape.size() == 1) in create_npy_header()