| /OK3568_Linux_fs/external/rknpu2/examples/rknn_yolov5_android_apk_demo/app/src/main/cpp/ |
| H A D | yolo_image.cc | 60 rknn_tensor_mem *output_mems[3]; variable
190 output_mems[i] = rknn_create_mem(ctx, output_attrs[i].n_elems * sizeof(unsigned char)); in create()
191 memset(output_mems[i]->virt_addr, 0, output_attrs[i].n_elems * sizeof(unsigned char)); in create()
195 rknn_set_io_mem(ctx, output_mems[i], &(output_attrs[i])); in create()
217 rknn_destroy_mem(ctx, output_mems[i]); in destroy()
304 memcpy(y0, output_mems[0]->virt_addr, output_attrs[0].n_elems * sizeof(char)); in run_yolo()
305 memcpy(y1, output_mems[1]->virt_addr, output_attrs[1].n_elems * sizeof(char)); in run_yolo()
306 memcpy(y2, output_mems[2]->virt_addr, output_attrs[2].n_elems * sizeof(char)); in run_yolo()
334 fprintf(fp, "%d\n", *((int8_t *)(output_mems[i]->virt_addr) + i)); in run_yolo()
|
| /OK3568_Linux_fs/external/rknpu2/examples/rknn_common_test/src/ |
| H A D | main.cc | 242 rknn_tensor_mem* output_mems[io_num.n_output]; in main() local
247 output_mems[i] = rknn_create_mem(ctx, output_size); in main()
262 ret = rknn_set_io_mem(ctx, output_mems[i], &output_attrs[i]); in main()
287 float* buffer = (float*)output_mems[i]->virt_addr; in main()
302 rknn_destroy_mem(ctx, output_mems[i]); in main()
|
| /OK3568_Linux_fs/external/rknpu2/examples/rknn_api_demo/src/ |
| H A D | rknn_create_mem_demo.cpp | 280 rknn_tensor_mem* output_mems[io_num.n_output]; in main() local
285 output_mems[i] = rknn_create_mem(ctx, output_size); in main()
300 ret = rknn_set_io_mem(ctx, output_mems[i], &output_attrs[i]); in main()
325 float* buffer = (float*)output_mems[i]->virt_addr; in main()
340 rknn_destroy_mem(ctx, output_mems[i]); in main()
|
| H A D | rknn_create_mem_with_rga_demo.cpp | 286 rknn_tensor_mem* output_mems[io_num.n_output]; in main() local
291 output_mems[i] = rknn_create_mem(ctx, output_size); in main()
303 ret = rknn_set_io_mem(ctx, output_mems[i], &output_attrs[i]); in main()
328 float* buffer = (float*)output_mems[i]->virt_addr; in main()
343 rknn_destroy_mem(ctx, output_mems[i]); in main()
|
| H A D | rknn_with_mmz_demo.cpp | 303 rknn_tensor_mem* output_mems[io_num.n_output]; in main() local
305 …output_mems[i] = rknn_create_mem_from_phys(ctx, output_physs[i], output_virts[i], output_attrs[i].… in main()
318 ret = rknn_set_io_mem(ctx, output_mems[i], &output_attrs[i]); in main()
343 float* buffer = (float*)output_mems[i]->virt_addr; in main()
364 rknn_destroy_mem(ctx, output_mems[i]); in main()
|
| H A D | rknn_set_internal_mem_from_fd_demo.cpp | 364 rknn_tensor_mem* output_mems[io_num.n_output]; in main() local
366 …output_mems[i] = rknn_create_mem_from_fd(ctx, output_fds[i], output_virts[i], output_attrs[i].size… in main()
379 ret = rknn_set_io_mem(ctx, output_mems[i], &output_attrs[i]); in main()
404 float* buffer = (float*)output_mems[i]->virt_addr; in main()
427 rknn_destroy_mem(ctx, output_mems[i]); in main()
|
| /OK3568_Linux_fs/external/rknpu2/examples/librknn_api_android_demo/ |
| H A D | rknn_create_mem_demo.cpp | 280 rknn_tensor_mem* output_mems[io_num.n_output]; in main() local
285 output_mems[i] = rknn_create_mem(ctx, output_size); in main()
300 ret = rknn_set_io_mem(ctx, output_mems[i], &output_attrs[i]); in main()
325 float* buffer = (float*)output_mems[i]->virt_addr; in main()
340 rknn_destroy_mem(ctx, output_mems[i]); in main()
|
| /OK3568_Linux_fs/external/rknpu2/examples/rknn_internal_mem_reuse_demo/src/ |
| H A D | rknn_internal_mem_reuse_demo.cc | 347 rknn_tensor_mem* output_mems[TOTAL_RKNN_MODEL_NUM][MAX_OUTPUT_NUM]; in main() local
358 memset(output_mems, 0x00, sizeof(output_mems)); in main()
497 output_mems[n][i] = rknn_create_mem(ctx[n], output_attrs[n][i].n_elems * sizeof(float)); in main()
505 ret = rknn_set_io_mem(ctx[n], output_mems[n][i], &output_attrs[n][i]); in main()
581 float* buffer = (float*)output_mems[n][i]->virt_addr; in main()
611 if (output_mems[n][i]) { in main()
612 rknn_destroy_mem(ctx[n], output_mems[n][i]); in main()
|
| /OK3568_Linux_fs/external/rknpu2/examples/RV1106_RV1103/rknn_mobilenet_demo/src/ |
| H A D | main.cc | 398 rknn_tensor_mem *output_mems[io_num.n_output]; in main() local
401 output_mems[i] = rknn_create_mem(ctx, output_attrs[i].size_with_stride); in main()
416 ret = rknn_set_io_mem(ctx, output_mems[i], &output_attrs[i]); in main()
471 …NC1HWC2_int8_to_NCHW_float((int8_t *)output_mems[i]->virt_addr, (float *)output_mems_nchw[i], (int… in main()
476 int8_t *src = (int8_t *)output_mems[i]->virt_addr; in main()
510 rknn_destroy_mem(ctx, output_mems[i]); in main()
|
| H A D | main_nhwc.cc | 368 rknn_tensor_mem *output_mems[io_num.n_output]; in main() local
371 output_mems[i] = rknn_create_mem(ctx, output_attrs[i].n_elems * sizeof(float)); in main()
387 ret = rknn_set_io_mem(ctx, output_mems[i], &output_attrs[i]); in main()
420 float *buffer = (float *)output_mems[i]->virt_addr; in main()
435 rknn_destroy_mem(ctx, output_mems[i]); in main()
|
| /OK3568_Linux_fs/external/rknpu2/examples/RV1106_RV1103/rknn_yolov5_demo/src/ |
| H A D | main.cc | 370 rknn_tensor_mem *output_mems[io_num.n_output]; in main() local
373 output_mems[i] = rknn_create_mem(ctx, output_attrs[i].size_with_stride); in main()
388 ret = rknn_set_io_mem(ctx, output_mems[i], &output_attrs[i]); in main()
440 …NC1HWC2_int8_to_NCHW_int8((int8_t *)output_mems[i]->virt_addr, (int8_t *)output_mems_nchw[i], (int… in main()
489 rknn_destroy_mem(ctx, output_mems[i]); in main()
|
| /OK3568_Linux_fs/external/rknpu2/examples/rknn_dynamic_shape_input_demo/src/ |
| H A D | rknn_dynshape_inference_zero_copy.cc | 475 rknn_tensor_mem *output_mems[io_num.n_output]; in main() local
481 output_mems[i] = rknn_create_mem(ctx, output_size); in main()
596 ret = rknn_set_io_mem(ctx, output_mems[i], &cur_output_attrs[i]); in main()
627 save_npy<float>(output_path, (float*)output_mems[i]->virt_addr, &cur_output_attrs[i]); in main()
634 float *output = (float *)output_mems[i]->virt_addr; in main()
656 rknn_destroy_mem(ctx, output_mems[i]); in main()
|