| /OK3568_Linux_fs/external/rknpu2/examples/rknn_multiple_input_demo/src/ |
| H A D | main.cc | 113 rknn_input_output_num io_num; in main() local
114 ret = rknn_query(ctx, RKNN_QUERY_IN_OUT_NUM, &io_num, sizeof(io_num)); in main()
119 printf("model input num: %d, output num: %d\n", io_num.n_input, io_num.n_output); in main()
122 rknn_tensor_attr input_attrs[io_num.n_input]; in main()
123 memset(input_attrs, 0, io_num.n_input * sizeof(rknn_tensor_attr)); in main()
124 for (uint32_t i = 0; i < io_num.n_input; i++) { in main()
136 rknn_tensor_attr output_attrs[io_num.n_output]; in main()
137 memset(output_attrs, 0, io_num.n_output * sizeof(rknn_tensor_attr)); in main()
138 for (uint32_t i = 0; i < io_num.n_output; i++) { in main()
158 unsigned char* input_data[io_num.n_input]; in main()
[all …]
|
| /OK3568_Linux_fs/external/rknpu2/examples/rknn_dynamic_shape_input_demo/src/ |
| H A D | rknn_dynshape_inference.cc | 341 rknn_input_output_num io_num; in main() local
342 ret = rknn_query(ctx, RKNN_QUERY_IN_OUT_NUM, &io_num, sizeof(io_num)); in main()
348 if (io_num.n_input != input_paths_split.size()) in main()
353 printf("model input num: %d, output num: %d\n", io_num.n_input, io_num.n_output); in main()
357 rknn_tensor_attr input_attrs[io_num.n_input]; in main()
358 memset(input_attrs, 0, io_num.n_input * sizeof(rknn_tensor_attr)); in main()
359 for (uint32_t i = 0; i < io_num.n_input; i++) in main()
373 rknn_tensor_attr output_attrs[io_num.n_output]; in main()
374 memset(output_attrs, 0, io_num.n_output * sizeof(rknn_tensor_attr)); in main()
375 for (uint32_t i = 0; i < io_num.n_output; i++) in main()
[all …]
|
| H A D | rknn_dynshape_inference_zero_copy.cc | 341 rknn_input_output_num io_num; in main() local
342 ret = rknn_query(ctx, RKNN_QUERY_IN_OUT_NUM, &io_num, sizeof(io_num)); in main()
348 if (io_num.n_input != input_paths_split.size()) in main()
353 printf("model input num: %d, output num: %d\n", io_num.n_input, io_num.n_output); in main()
357 rknn_tensor_attr input_attrs[io_num.n_input]; in main()
358 memset(input_attrs, 0, io_num.n_input * sizeof(rknn_tensor_attr)); in main()
359 for (uint32_t i = 0; i < io_num.n_input; i++) in main()
373 rknn_tensor_attr output_attrs[io_num.n_output]; in main()
374 memset(output_attrs, 0, io_num.n_output * sizeof(rknn_tensor_attr)); in main()
375 for (uint32_t i = 0; i < io_num.n_output; i++) in main()
[all …]
|
| /OK3568_Linux_fs/external/rknpu2/examples/rknn_benchmark/src/ |
| H A D | rknn_benchmark.cpp | 307 rknn_input_output_num io_num; in main() local
308 ret = rknn_query(ctx, RKNN_QUERY_IN_OUT_NUM, &io_num, sizeof(io_num)); in main()
314 printf("model input num: %d, output num: %d\n", io_num.n_input, io_num.n_output); in main()
317 rknn_tensor_attr input_attrs[io_num.n_input]; in main()
318 memset(input_attrs, 0, io_num.n_input * sizeof(rknn_tensor_attr)); in main()
319 for (uint32_t i = 0; i < io_num.n_input; i++) { in main()
332 rknn_tensor_attr output_attrs[io_num.n_output]; in main()
333 memset(output_attrs, 0, io_num.n_output * sizeof(rknn_tensor_attr)); in main()
334 for (uint32_t i = 0; i < io_num.n_output; i++) { in main()
356 unsigned char* input_data[io_num.n_input]; in main()
[all …]
|
| /OK3568_Linux_fs/external/rknpu2/examples/rknn_api_demo/src/ |
| H A D | rknn_with_mmz_demo.cpp | 166 rknn_input_output_num io_num; in main() local
167 ret = rknn_query(ctx, RKNN_QUERY_IN_OUT_NUM, &io_num, sizeof(io_num)); in main()
172 printf("model input num: %d, output num: %d\n", io_num.n_input, io_num.n_output); in main()
175 rknn_tensor_attr input_attrs[io_num.n_input]; in main()
176 memset(input_attrs, 0, io_num.n_input * sizeof(rknn_tensor_attr)); in main()
177 for (uint32_t i = 0; i < io_num.n_input; i++) { in main()
189 rknn_tensor_attr output_attrs[io_num.n_output]; in main()
190 memset(output_attrs, 0, io_num.n_output * sizeof(rknn_tensor_attr)); in main()
191 for (uint32_t i = 0; i < io_num.n_output; i++) { in main()
245 MB_BLK output_mbs[io_num.n_output]; in main()
[all …]
|
| H A D | rknn_set_internal_mem_from_fd_demo.cpp | 175 rknn_input_output_num io_num; in main() local
176 ret = rknn_query(ctx, RKNN_QUERY_IN_OUT_NUM, &io_num, sizeof(io_num)); in main()
181 printf("model input num: %d, output num: %d\n", io_num.n_input, io_num.n_output); in main()
184 rknn_tensor_attr input_attrs[io_num.n_input]; in main()
185 memset(input_attrs, 0, io_num.n_input * sizeof(rknn_tensor_attr)); in main()
186 for (uint32_t i = 0; i < io_num.n_input; i++) { in main()
198 rknn_tensor_attr output_attrs[io_num.n_output]; in main()
199 memset(output_attrs, 0, io_num.n_output * sizeof(rknn_tensor_attr)); in main()
200 for (uint32_t i = 0; i < io_num.n_output; i++) { in main()
306 MB_BLK output_mbs[io_num.n_output]; in main()
[all …]
|
| H A D | rknn_create_mem_demo.cpp | 191 rknn_input_output_num io_num; in main() local
192 ret = rknn_query(ctx, RKNN_QUERY_IN_OUT_NUM, &io_num, sizeof(io_num)); in main()
197 printf("model input num: %d, output num: %d\n", io_num.n_input, io_num.n_output); in main()
200 rknn_tensor_attr input_attrs[io_num.n_input]; in main()
201 memset(input_attrs, 0, io_num.n_input * sizeof(rknn_tensor_attr)); in main()
202 for (uint32_t i = 0; i < io_num.n_input; i++) { in main()
214 rknn_tensor_attr output_attrs[io_num.n_output]; in main()
215 memset(output_attrs, 0, io_num.n_output * sizeof(rknn_tensor_attr)); in main()
216 for (uint32_t i = 0; i < io_num.n_output; i++) { in main()
280 rknn_tensor_mem* output_mems[io_num.n_output]; in main()
[all …]
|
| H A D | rknn_create_mem_with_rga_demo.cpp | 181 rknn_input_output_num io_num; in main() local
182 ret = rknn_query(ctx, RKNN_QUERY_IN_OUT_NUM, &io_num, sizeof(io_num)); in main()
187 printf("model input num: %d, output num: %d\n", io_num.n_input, io_num.n_output); in main()
190 rknn_tensor_attr input_attrs[io_num.n_input]; in main()
191 memset(input_attrs, 0, io_num.n_input * sizeof(rknn_tensor_attr)); in main()
192 for (uint32_t i = 0; i < io_num.n_input; i++) { in main()
204 rknn_tensor_attr output_attrs[io_num.n_output]; in main()
205 memset(output_attrs, 0, io_num.n_output * sizeof(rknn_tensor_attr)); in main()
206 for (uint32_t i = 0; i < io_num.n_output; i++) { in main()
286 rknn_tensor_mem* output_mems[io_num.n_output]; in main()
[all …]
|
| /OK3568_Linux_fs/external/rknpu2/doc/ |
| H A D | RKNN_Dynamic_Shape_Usage.md | 33 rknn_input_range dyn_range[io_num.n_input];
34 memset(dyn_range, 0, io_num.n_input * sizeof(rknn_input_range));
35 for (uint32_t i = 0; i < io_num.n_input; i++)
56 for (int i = 0; i < io_num.n_input; i++)
77 rknn_tensor_attr cur_input_attrs[io_num.n_input];
78 memset(cur_input_attrs, 0, io_num.n_input * sizeof(rknn_tensor_attr));
79 for (uint32_t i = 0; i < io_num.n_input; i++)
91 rknn_tensor_attr cur_output_attrs[io_num.n_output];
92 memset(cur_output_attrs, 0, io_num.n_output * sizeof(rknn_tensor_attr));
93 for (uint32_t i = 0; i < io_num.n_output; i++)
[all …]
|
| /OK3568_Linux_fs/external/rknpu2/examples/RV1106_RV1103/rknn_yolov5_demo/src/ |
| H A D | main.cc | 270 rknn_input_output_num io_num; in main() local
271 ret = rknn_query(ctx, RKNN_QUERY_IN_OUT_NUM, &io_num, sizeof(io_num)); in main()
277 printf("model input num: %d, output num: %d\n", io_num.n_input, io_num.n_output); in main()
280 rknn_tensor_attr input_attrs[io_num.n_input]; in main()
281 memset(input_attrs, 0, io_num.n_input * sizeof(rknn_tensor_attr)); in main()
282 for (uint32_t i = 0; i < io_num.n_input; i++) in main()
296 rknn_tensor_attr output_attrs[io_num.n_output]; in main()
297 memset(output_attrs, 0, io_num.n_output * sizeof(rknn_tensor_attr)); in main()
298 for (uint32_t i = 0; i < io_num.n_output; i++) in main()
370 rknn_tensor_mem *output_mems[io_num.n_output]; in main()
[all …]
|
| /OK3568_Linux_fs/external/rknpu2/examples/rknn_common_test/src/ |
| H A D | main.cc | 115 rknn_input_output_num io_num; in main() local
116 ret = rknn_query(ctx, RKNN_QUERY_IN_OUT_NUM, &io_num, sizeof(io_num)); in main()
121 printf("model input num: %d, output num: %d\n", io_num.n_input, io_num.n_output); in main()
124 rknn_tensor_attr input_attrs[io_num.n_input]; in main()
125 memset(input_attrs, 0, io_num.n_input * sizeof(rknn_tensor_attr)); in main()
126 for (uint32_t i = 0; i < io_num.n_input; i++) { in main()
138 rknn_tensor_attr output_attrs[io_num.n_output]; in main()
139 memset(output_attrs, 0, io_num.n_output * sizeof(rknn_tensor_attr)); in main()
140 for (uint32_t i = 0; i < io_num.n_output; i++) { in main()
242 rknn_tensor_mem* output_mems[io_num.n_output]; in main()
[all …]
|
| /OK3568_Linux_fs/external/rknpu2/examples/librknn_api_android_demo/ |
| H A D | rknn_create_mem_demo.cpp | 191 rknn_input_output_num io_num; in main() local
192 ret = rknn_query(ctx, RKNN_QUERY_IN_OUT_NUM, &io_num, sizeof(io_num)); in main()
197 printf("model input num: %d, output num: %d\n", io_num.n_input, io_num.n_output); in main()
200 rknn_tensor_attr input_attrs[io_num.n_input]; in main()
201 memset(input_attrs, 0, io_num.n_input * sizeof(rknn_tensor_attr)); in main()
202 for (uint32_t i = 0; i < io_num.n_input; i++) { in main()
214 rknn_tensor_attr output_attrs[io_num.n_output]; in main()
215 memset(output_attrs, 0, io_num.n_output * sizeof(rknn_tensor_attr)); in main()
216 for (uint32_t i = 0; i < io_num.n_output; i++) { in main()
280 rknn_tensor_mem* output_mems[io_num.n_output]; in main()
[all …]
|
| /OK3568_Linux_fs/external/rknpu2/examples/RV1106_RV1103/rknn_mobilenet_demo/src/ |
| H A D | main.cc | 298 rknn_input_output_num io_num; in main() local
299 ret = rknn_query(ctx, RKNN_QUERY_IN_OUT_NUM, &io_num, sizeof(io_num)); in main()
305 printf("model input num: %d, output num: %d\n", io_num.n_input, io_num.n_output); in main()
308 rknn_tensor_attr input_attrs[io_num.n_input]; in main()
309 memset(input_attrs, 0, io_num.n_input * sizeof(rknn_tensor_attr)); in main()
310 for (uint32_t i = 0; i < io_num.n_input; i++) in main()
324 rknn_tensor_attr output_attrs[io_num.n_output]; in main()
325 memset(output_attrs, 0, io_num.n_output * sizeof(rknn_tensor_attr)); in main()
326 for (uint32_t i = 0; i < io_num.n_output; i++) in main()
398 rknn_tensor_mem *output_mems[io_num.n_output]; in main()
[all …]
|
| H A D | main_nhwc.cc | 268 rknn_input_output_num io_num; in main() local
269 ret = rknn_query(ctx, RKNN_QUERY_IN_OUT_NUM, &io_num, sizeof(io_num)); in main()
275 printf("model input num: %d, output num: %d\n", io_num.n_input, io_num.n_output); in main()
278 rknn_tensor_attr input_attrs[io_num.n_input]; in main()
279 memset(input_attrs, 0, io_num.n_input * sizeof(rknn_tensor_attr)); in main()
280 for (uint32_t i = 0; i < io_num.n_input; i++) in main()
294 rknn_tensor_attr output_attrs[io_num.n_output]; in main()
295 memset(output_attrs, 0, io_num.n_output * sizeof(rknn_tensor_attr)); in main()
296 for (uint32_t i = 0; i < io_num.n_output; i++) in main()
368 rknn_tensor_mem *output_mems[io_num.n_output]; in main()
[all …]
|
| /OK3568_Linux_fs/external/rknpu2/examples/rknn_yolov5_demo/src/ |
| H A D | main.cc | 183 rknn_input_output_num io_num; in main() local
184 ret = rknn_query(ctx, RKNN_QUERY_IN_OUT_NUM, &io_num, sizeof(io_num)); in main()
189 printf("model input num: %d, output num: %d\n", io_num.n_input, io_num.n_output); in main()
191 rknn_tensor_attr input_attrs[io_num.n_input]; in main()
193 for (int i = 0; i < io_num.n_input; i++) { in main()
203 rknn_tensor_attr output_attrs[io_num.n_output]; in main()
205 for (int i = 0; i < io_num.n_output; i++) { in main()
263 rknn_inputs_set(ctx, io_num.n_input, inputs); in main()
265 rknn_output outputs[io_num.n_output]; in main()
267 for (int i = 0; i < io_num.n_output; i++) { in main()
[all …]
|
| H A D | main_video.cc | 43 rknn_input_output_num io_num; member
173 ret = rknn_query(ctx, RKNN_QUERY_IN_OUT_NUM, &app_ctx->io_num, sizeof(rknn_input_output_num)); in init_model()
178 …printf("model input num: %d, output num: %d\n", app_ctx->io_num.n_input, app_ctx->io_num.n_output); in init_model()
180 …rknn_tensor_attr* input_attrs = (rknn_tensor_attr*)malloc(app_ctx->io_num.n_input * sizeof(rknn_te… in init_model()
182 for (int i = 0; i < app_ctx->io_num.n_input; i++) { in init_model()
192 …rknn_tensor_attr* output_attrs = (rknn_tensor_attr*)malloc(app_ctx->io_num.n_output * sizeof(rknn_… in init_model()
194 for (int i = 0; i < app_ctx->io_num.n_output; i++) { in init_model()
285 rknn_inputs_set(ctx, app_ctx->io_num.n_input, inputs); in inference_model()
287 rknn_output outputs[app_ctx->io_num.n_output]; in inference_model()
289 for (int i = 0; i < app_ctx->io_num.n_output; i++) { in inference_model()
[all …]
|
| /OK3568_Linux_fs/external/rknpu2/examples/rknn_mobilenet_demo/src/ |
| H A D | main.cc | 145 rknn_input_output_num io_num; in main() local
146 ret = rknn_query(ctx, RKNN_QUERY_IN_OUT_NUM, &io_num, sizeof(io_num)); in main()
151 printf("model input num: %d, output num: %d\n", io_num.n_input, io_num.n_output); in main()
154 rknn_tensor_attr input_attrs[io_num.n_input]; in main()
156 for (int i = 0; i < io_num.n_input; i++) { in main()
167 rknn_tensor_attr output_attrs[io_num.n_output]; in main()
169 for (int i = 0; i < io_num.n_output; i++) { in main()
188 ret = rknn_inputs_set(ctx, io_num.n_input, inputs); in main()
213 for (int i = 0; i < io_num.n_output; i++) { in main()
|
| /OK3568_Linux_fs/external/rknpu2/examples/rknn_internal_mem_reuse_demo/src/ |
| H A D | rknn_internal_mem_reuse_demo.cc | 341 rknn_input_output_num io_num[TOTAL_RKNN_MODEL_NUM]; in main() local
400 ret = rknn_query(ctx[n], RKNN_QUERY_IN_OUT_NUM, &io_num[n], sizeof(io_num[n])); in main()
405 printf("model input num: %d, output num: %d\n", io_num[n].n_input, io_num[n].n_output); in main()
407 if (io_num[n].n_output > MAX_OUTPUT_NUM) { in main()
413 memset(input_attrs[n], 0, io_num[n].n_input * sizeof(rknn_tensor_attr)); in main()
414 for (uint32_t i = 0; i < io_num[n].n_input; i++) { in main()
426 memset(output_attrs[n], 0, io_num[n].n_output * sizeof(rknn_tensor_attr)); in main()
427 for (uint32_t i = 0; i < io_num[n].n_output; i++) { in main()
496 for (uint32_t i = 0; i < io_num[n].n_output; ++i) { in main()
501 for (uint32_t i = 0; i < io_num[n].n_output; ++i) { in main()
[all …]
|
| /OK3568_Linux_fs/external/rknpu2/examples/rknn_yolov5_android_apk_demo/app/src/main/cpp/ |
| H A D | yolo_image.cc | 112 rknn_input_output_num io_num; in create() local
115 ret = rknn_query(ctx, cmd, &io_num, sizeof(io_num)); in create()
120 n_input = io_num.n_input; in create()
121 n_output = io_num.n_output; in create()
|
| /OK3568_Linux_fs/kernel/kernel/ |
| H A D | resource.c | 1566 unsigned int io_start, io_num; in reserve_setup() local
1572 if (get_option(&str, &io_num) == 0) in reserve_setup()
1590 res->end = io_start + io_num - 1; in reserve_setup()
|