Searched refs:npy_data (Results 1 – 4 of 4) sorted by relevance
| /OK3568_Linux_fs/external/rknpu2/examples/rknn_internal_mem_reuse_demo/src/ |
| H A D | rknn_internal_mem_reuse_demo.cc | 233 cnpy_array npy_data; in load_npy() local
236 if (cnpy_open(input_path, writable, &npy_data) != CNPY_SUCCESS) { in load_npy()
241 int data_bytes = npy_data.raw_data_size - npy_data.data_begin; in load_npy()
242 cnpy_dtype dtype = npy_data.dtype; in load_npy()
255 int start = npy_data.n_dim == 4 ? 0 : 1; in load_npy()
256 for (size_t i = 0; i < npy_data.n_dim && i < 4; ++i) { in load_npy()
257 npy_shape[start + i] = npy_data.dims[i]; in load_npy()
277 memcpy(data, npy_data.raw_data + npy_data.data_begin, data_bytes); in load_npy()
292 cnpy_array npy_data; in save_npy() local
298 &npy_data) != CNPY_SUCCESS) { in save_npy()
[all …]
|
| /OK3568_Linux_fs/external/rknpu2/examples/rknn_benchmark/src/ |
| H A D | rknn_benchmark.cpp | 113 NpyArray npy_data = npy_load(input_path); in load_npy() local
115 int type_bytes = npy_data.word_size; in load_npy()
116 std::string typeName = npy_data.typeName; in load_npy()
137 int start = npy_data.shape.size() == 4 ? 0 : 1; in load_npy()
138 for (size_t i = 0; i < npy_data.shape.size() && i < 4; ++i) { in load_npy()
139 npy_shape[start + i] = npy_data.shape[i]; in load_npy()
153 unsigned char* data = (unsigned char*)malloc(npy_data.num_bytes()); in load_npy()
159 memcpy(data, npy_data.data<unsigned char>(), npy_data.num_bytes()); in load_npy()
161 *input_size = npy_data.num_bytes(); in load_npy()
|
| /OK3568_Linux_fs/external/rknpu2/examples/rknn_dynamic_shape_input_demo/src/ |
| H A D | rknn_dynshape_inference.cc | 120 NpyArray npy_data = npy_load(input_path); in load_npy() local
122 int type_bytes = npy_data.word_size; in load_npy()
123 std::string typeName = npy_data.typeName; in load_npy()
136 int start = npy_data.shape.size() == 4 ? 0 : 1; in load_npy()
137 for (size_t i = 0; i < npy_data.shape.size() && i < 4; ++i) in load_npy()
139 npy_shape[start + i] = npy_data.shape[i]; in load_npy()
154 unsigned char *data = (unsigned char *)malloc(npy_data.num_bytes()); in load_npy()
161 memcpy(data, npy_data.data<unsigned char>(), npy_data.num_bytes()); in load_npy()
163 *input_size = npy_data.num_bytes(); in load_npy()
|
| H A D | rknn_dynshape_inference_zero_copy.cc | 120 NpyArray npy_data = npy_load(input_path); in load_npy() local
122 int type_bytes = npy_data.word_size; in load_npy()
123 std::string typeName = npy_data.typeName; in load_npy()
136 int start = npy_data.shape.size() == 4 ? 0 : 1; in load_npy()
137 for (size_t i = 0; i < npy_data.shape.size() && i < 4; ++i) in load_npy()
139 npy_shape[start + i] = npy_data.shape[i]; in load_npy()
154 unsigned char *data = (unsigned char *)malloc(npy_data.num_bytes()); in load_npy()
161 memcpy(data, npy_data.data<unsigned char>(), npy_data.num_bytes()); in load_npy()
163 *input_size = npy_data.num_bytes(); in load_npy()
|