Home
last modified time | relevance | path

Searched refs:input_attrs (Results 1 – 19 of 19) sorted by relevance

/OK3568_Linux_fs/external/rknpu2/examples/rknn_common_test/src/
H A Dmain.cc124 rknn_tensor_attr input_attrs[io_num.n_input]; in main() local
125 memset(input_attrs, 0, io_num.n_input * sizeof(rknn_tensor_attr)); in main()
127 input_attrs[i].index = i; in main()
129 ret = rknn_query(ctx, RKNN_QUERY_INPUT_ATTR, &(input_attrs[i]), sizeof(rknn_tensor_attr)); in main()
134 dump_tensor_attr(&input_attrs[i]); in main()
169 switch (input_attrs[0].fmt) { in main()
171 req_height = input_attrs[0].dims[1]; in main()
172 req_width = input_attrs[0].dims[2]; in main()
173 req_channel = input_attrs[0].dims[3]; in main()
176 req_height = input_attrs[0].dims[2]; in main()
[all …]
/OK3568_Linux_fs/external/rknpu2/examples/rknn_yolov5_android_apk_demo/app/src/main/cpp/
H A Dyolo_image.cc56 rknn_tensor_attr input_attrs[1]; variable
124 memset(input_attrs, 0, n_input * sizeof(rknn_tensor_attr)); in create()
126 input_attrs[i].index = i; in create()
128 ret = rknn_query(ctx, cmd, &(input_attrs[i]), sizeof(rknn_tensor_attr)); in create()
135 if (RKNN_TENSOR_NHWC == input_attrs[0].fmt) { in create()
136 m_in_height = input_attrs[0].dims[1]; in create()
137 m_in_width = input_attrs[0].dims[2]; in create()
138 m_in_channel = input_attrs[0].dims[3]; in create()
139 } else if (RKNN_TENSOR_NCHW == input_attrs[0].fmt) { in create()
140 m_in_height = input_attrs[0].dims[2]; in create()
[all …]
/OK3568_Linux_fs/external/rknpu2/examples/RV1106_RV1103/rknn_yolov5_demo/src/
H A Dmain.cc280 rknn_tensor_attr input_attrs[io_num.n_input]; in main() local
281 memset(input_attrs, 0, io_num.n_input * sizeof(rknn_tensor_attr)); in main()
284 input_attrs[i].index = i; in main()
286 ret = rknn_query(ctx, RKNN_QUERY_INPUT_ATTR, &(input_attrs[i]), sizeof(rknn_tensor_attr)); in main()
292 dump_tensor_attr(&input_attrs[i]); in main()
326 input_data = load_image(input_path, &input_attrs[0], &img_height, &img_width); in main()
337 input_attrs[0].type = input_type; in main()
339 input_attrs[0].fmt = input_layout; in main()
341 input_mems[0] = rknn_create_mem(ctx, input_attrs[0].size_with_stride); in main()
344 int width = input_attrs[0].dims[2]; in main()
[all …]
/OK3568_Linux_fs/external/rknpu2/examples/rknn_api_demo/src/
H A Drknn_create_mem_demo.cpp200 rknn_tensor_attr input_attrs[io_num.n_input]; in main() local
201 memset(input_attrs, 0, io_num.n_input * sizeof(rknn_tensor_attr)); in main()
203 input_attrs[i].index = i; in main()
205 ret = rknn_query(ctx, RKNN_QUERY_INPUT_ATTR, &(input_attrs[i]), sizeof(rknn_tensor_attr)); in main()
210 dump_tensor_attr(&input_attrs[i]); in main()
241 input_data = load_image(input_path, &input_attrs[0]); in main()
251 input_attrs[0].type = input_type; in main()
253 input_attrs[0].fmt = input_layout; in main()
255 input_mems[0] = rknn_create_mem(ctx, input_attrs[0].size_with_stride); in main()
258 int width = input_attrs[0].dims[2]; in main()
[all …]
H A Drknn_create_mem_with_rga_demo.cpp190 rknn_tensor_attr input_attrs[io_num.n_input]; in main() local
191 memset(input_attrs, 0, io_num.n_input * sizeof(rknn_tensor_attr)); in main()
193 input_attrs[i].index = i; in main()
195 ret = rknn_query(ctx, RKNN_QUERY_INPUT_ATTR, &(input_attrs[i]), sizeof(rknn_tensor_attr)); in main()
200 dump_tensor_attr(&input_attrs[i]); in main()
237 switch (input_attrs[0].fmt) { in main()
239 model_in_height = input_attrs[0].dims[1]; in main()
240 model_in_width = input_attrs[0].dims[2]; in main()
241 req_channel = input_attrs[0].dims[3]; in main()
244 model_in_height = input_attrs[0].dims[2]; in main()
[all …]
H A Drknn_with_mmz_demo.cpp175 rknn_tensor_attr input_attrs[io_num.n_input]; in main() local
176 memset(input_attrs, 0, io_num.n_input * sizeof(rknn_tensor_attr)); in main()
178 input_attrs[i].index = i; in main()
180 ret = rknn_query(ctx, RKNN_QUERY_INPUT_ATTR, &(input_attrs[i]), sizeof(rknn_tensor_attr)); in main()
185 dump_tensor_attr(&input_attrs[i]); in main()
216 input_data = load_image(input_path, &input_attrs[0]); in main()
226 int input_size = input_attrs[0].size_with_stride; in main()
275 input_attrs[0].type = input_type; in main()
277 input_attrs[0].fmt = input_layout; in main()
279 …input_mems[0] = rknn_create_mem_from_phys(ctx, input_phys, input_virt, input_attrs[0].size_with_st… in main()
[all …]
H A Drknn_set_internal_mem_from_fd_demo.cpp184 rknn_tensor_attr input_attrs[io_num.n_input]; in main() local
185 memset(input_attrs, 0, io_num.n_input * sizeof(rknn_tensor_attr)); in main()
187 input_attrs[i].index = i; in main()
189 ret = rknn_query(ctx, RKNN_QUERY_INPUT_ATTR, &(input_attrs[i]), sizeof(rknn_tensor_attr)); in main()
194 dump_tensor_attr(&input_attrs[i]); in main()
225 input_data = load_image(input_path, &input_attrs[0]); in main()
287 int input_size = input_attrs[0].size_with_stride; in main()
336 input_attrs[0].type = input_type; in main()
338 input_attrs[0].fmt = input_layout; in main()
340 …input_mems[0] = rknn_create_mem_from_fd(ctx, input_fd, input_virt, input_attrs[0].size_with_stride… in main()
[all …]
/OK3568_Linux_fs/external/rknpu2/examples/librknn_api_android_demo/
H A Drknn_create_mem_demo.cpp200 rknn_tensor_attr input_attrs[io_num.n_input]; in main() local
201 memset(input_attrs, 0, io_num.n_input * sizeof(rknn_tensor_attr)); in main()
203 input_attrs[i].index = i; in main()
205 ret = rknn_query(ctx, RKNN_QUERY_INPUT_ATTR, &(input_attrs[i]), sizeof(rknn_tensor_attr)); in main()
210 dump_tensor_attr(&input_attrs[i]); in main()
241 input_data = load_image(input_path, &input_attrs[0]); in main()
251 input_attrs[0].type = input_type; in main()
253 input_attrs[0].fmt = input_layout; in main()
255 input_mems[0] = rknn_create_mem(ctx, input_attrs[0].size_with_stride); in main()
258 int width = input_attrs[0].dims[2]; in main()
[all …]
/OK3568_Linux_fs/external/rknpu2/examples/RV1106_RV1103/rknn_mobilenet_demo/src/
H A Dmain_nhwc.cc278 rknn_tensor_attr input_attrs[io_num.n_input]; in main() local
279 memset(input_attrs, 0, io_num.n_input * sizeof(rknn_tensor_attr)); in main()
282 input_attrs[i].index = i; in main()
284 ret = rknn_query(ctx, RKNN_QUERY_INPUT_ATTR, &(input_attrs[i]), sizeof(rknn_tensor_attr)); in main()
290 dump_tensor_attr(&input_attrs[i]); in main()
324 input_data = load_image(input_path, &input_attrs[0]); in main()
335 input_attrs[0].type = input_type; in main()
337 input_attrs[0].fmt = input_layout; in main()
339 input_mems[0] = rknn_create_mem(ctx, input_attrs[0].size_with_stride); in main()
342 int width = input_attrs[0].dims[2]; in main()
[all …]
H A Dmain.cc308 rknn_tensor_attr input_attrs[io_num.n_input]; in main() local
309 memset(input_attrs, 0, io_num.n_input * sizeof(rknn_tensor_attr)); in main()
312 input_attrs[i].index = i; in main()
314 ret = rknn_query(ctx, RKNN_QUERY_INPUT_ATTR, &(input_attrs[i]), sizeof(rknn_tensor_attr)); in main()
320 dump_tensor_attr(&input_attrs[i]); in main()
354 input_data = load_image(input_path, &input_attrs[0]); in main()
365 input_attrs[0].type = input_type; in main()
367 input_attrs[0].fmt = input_layout; in main()
369 input_mems[0] = rknn_create_mem(ctx, input_attrs[0].size_with_stride); in main()
372 int width = input_attrs[0].dims[2]; in main()
[all …]
/OK3568_Linux_fs/external/rknpu2/examples/rknn_dynamic_shape_input_demo/src/
H A Drknn_dynshape_inference_zero_copy.cc357 rknn_tensor_attr input_attrs[io_num.n_input]; in main() local
358 memset(input_attrs, 0, io_num.n_input * sizeof(rknn_tensor_attr)); in main()
361 input_attrs[i].index = i; in main()
363 ret = rknn_query(ctx, RKNN_QUERY_INPUT_ATTR, &(input_attrs[i]), sizeof(rknn_tensor_attr)); in main()
369 dump_tensor_attr(&input_attrs[i]); in main()
411 input_size[i] = input_attrs[i].n_elems * sizeof(uint8_t); in main()
428 … input_data[i] = load_npy(input_paths_split[i].c_str(), &input_attrs[i], &input_size[i]); in main()
429 if (input_size[i] != input_attrs[i].n_elems * sizeof(uint8_t)) in main()
431 …[%d] npy size error, read: %d, but requried: %lu\n", i, input_size[i], input_attrs[i].n_elems * si… in main()
438 …int height = input_attrs[i].fmt == RKNN_TENSOR_NHWC ? input_attrs[i].dims[1] : input_attrs[i].dims… in main()
[all …]
H A Drknn_dynshape_inference.cc357 rknn_tensor_attr input_attrs[io_num.n_input]; in main() local
358 memset(input_attrs, 0, io_num.n_input * sizeof(rknn_tensor_attr)); in main()
361 input_attrs[i].index = i; in main()
363 ret = rknn_query(ctx, RKNN_QUERY_INPUT_ATTR, &(input_attrs[i]), sizeof(rknn_tensor_attr)); in main()
369 dump_tensor_attr(&input_attrs[i]); in main()
411 input_size[i] = input_attrs[i].n_elems * sizeof(uint8_t); in main()
428 … input_data[i] = load_npy(input_paths_split[i].c_str(), &input_attrs[i], &input_size[i]); in main()
429 if (input_size[i] != input_attrs[i].n_elems * sizeof(uint8_t)) in main()
431 …[%d] npy size error, read: %d, but requried: %lu\n", i, input_size[i], input_attrs[i].n_elems * si… in main()
438 …int height = input_attrs[i].fmt == RKNN_TENSOR_NHWC ? input_attrs[i].dims[1] : input_attrs[i].dims… in main()
[all …]
/OK3568_Linux_fs/external/rknpu2/examples/rknn_yolov5_demo/src/
H A Dmain.cc191 rknn_tensor_attr input_attrs[io_num.n_input]; in main() local
192 memset(input_attrs, 0, sizeof(input_attrs)); in main()
194 input_attrs[i].index = i; in main()
195 …ret = rknn_query(ctx, RKNN_QUERY_INPUT_ATTR, &(input_attrs[i]), sizeof(rknn_tenso… in main()
200 dump_tensor_attr(&(input_attrs[i])); in main()
214 if (input_attrs[0].fmt == RKNN_TENSOR_NCHW) { in main()
216 channel = input_attrs[0].dims[1]; in main()
217 height = input_attrs[0].dims[2]; in main()
218 width = input_attrs[0].dims[3]; in main()
221 height = input_attrs[0].dims[1]; in main()
[all …]
H A Dmain_video.cc44 rknn_tensor_attr* input_attrs; member
180 …rknn_tensor_attr* input_attrs = (rknn_tensor_attr*)malloc(app_ctx->io_num.n_input * sizeof(rknn_te… in init_model() local
181 memset(input_attrs, 0, sizeof(input_attrs)); in init_model()
183 input_attrs[i].index = i; in init_model()
184 ret = rknn_query(ctx, RKNN_QUERY_INPUT_ATTR, &(input_attrs[i]), sizeof(rknn_tensor_attr)); in init_model()
189 dump_tensor_attr(&(input_attrs[i])); in init_model()
204 app_ctx->input_attrs = input_attrs; in init_model()
208 if (input_attrs[0].fmt == RKNN_TENSOR_NCHW) { in init_model()
210 app_ctx->model_channel = input_attrs[0].dims[1]; in init_model()
211 app_ctx->model_height = input_attrs[0].dims[2]; in init_model()
[all …]
/OK3568_Linux_fs/external/rknpu2/examples/rknn_internal_mem_reuse_demo/src/
H A Drknn_internal_mem_reuse_demo.cc344 rknn_tensor_attr input_attrs[TOTAL_RKNN_MODEL_NUM][1]; // this demo only support one input in main() local
413 memset(input_attrs[n], 0, io_num[n].n_input * sizeof(rknn_tensor_attr)); in main()
415 input_attrs[n][i].index = i; in main()
417 … ret = rknn_query(ctx[n], RKNN_QUERY_INPUT_ATTR, &(input_attrs[n][i]), sizeof(rknn_tensor_attr)); in main()
422 dump_tensor_attr(&input_attrs[n][i]); in main()
481 input_attrs[n][0].type = input_type; in main()
483 input_attrs[n][0].fmt = input_layout; in main()
485 input_mems[n][0] = rknn_create_mem(ctx[n], input_attrs[n][0].size_with_stride); in main()
488 ret = rknn_set_io_mem(ctx[n], input_mems[n][0], &input_attrs[n][0]); in main()
519 input_data = load_npy(input_path[n], &input_attrs[n][0], (int*)&input_type, &input_size); in main()
[all …]
/OK3568_Linux_fs/external/rknpu2/examples/rknn_multiple_input_demo/src/
H A Dmain.cc122 rknn_tensor_attr input_attrs[io_num.n_input]; in main() local
123 memset(input_attrs, 0, io_num.n_input * sizeof(rknn_tensor_attr)); in main()
125 input_attrs[i].index = i; in main()
127 ret = rknn_query(ctx, RKNN_QUERY_INPUT_ATTR, &(input_attrs[i]), sizeof(rknn_tensor_attr)); in main()
132 dump_tensor_attr(&input_attrs[i]); in main()
166 input_size[i] = input_attrs[i].n_elems * sizeof(uint8_t); in main()
174 input_data[i] = new unsigned char[input_attrs[i].size]; in main()
184 fread(input_data[i], input_attrs[i].size, 1, fp); in main()
/OK3568_Linux_fs/external/rknpu2/examples/rknn_mobilenet_demo/src/
H A Dmain.cc154 rknn_tensor_attr input_attrs[io_num.n_input]; in main() local
155 memset(input_attrs, 0, sizeof(input_attrs)); in main()
157 input_attrs[i].index = i; in main()
158 …ret = rknn_query(ctx, RKNN_QUERY_INPUT_ATTR, &(input_attrs[i]), sizeof(rknn_tenso… in main()
163 dump_tensor_attr(&(input_attrs[i])); in main()
/OK3568_Linux_fs/external/rknpu2/examples/rknn_benchmark/src/
H A Drknn_benchmark.cpp317 rknn_tensor_attr input_attrs[io_num.n_input]; in main() local
318 memset(input_attrs, 0, io_num.n_input * sizeof(rknn_tensor_attr)); in main()
320 input_attrs[i].index = i; in main()
322 ret = rknn_query(ctx, RKNN_QUERY_INPUT_ATTR, &(input_attrs[i]), sizeof(rknn_tensor_attr)); in main()
328 dump_tensor_attr(&input_attrs[i]); in main()
367 input_size[i] = input_attrs[i].n_elems * sizeof(uint8_t); in main()
378 …input_data[i] = load_npy(input_paths_split[i].c_str(), &input_attrs[i], &input_type[i], &input_siz… in main()
381 input_data[i] = load_image(input_paths_split[i].c_str(), &input_attrs[i]); in main()
/OK3568_Linux_fs/external/rknpu2/doc/
H A DRKNN_Dynamic_Shape_Usage.md58 for (int j = 0; j < input_attrs[i].n_dims; ++j)
60 input_attrs[i].dims[j] = shape_range[i].dyn_range[s][j];
63 ret = rknn_set_input_shape(ctx, &input_attrs[i]);
72 其中，shape_num是支持的形状个数，shape_range[i]是第i个输入的rknn_input_range结构体，input_attrs[i]是第i个输入的rknn_tensor_attr…