| /OK3568_Linux_fs/external/rknpu2/examples/rknn_internal_mem_reuse_demo/src/ |
| H A D | rknn_internal_mem_reuse_demo.cc | 346 rknn_tensor_mem* input_mems[TOTAL_RKNN_MODEL_NUM][1]; // this demo only support one input in main() local
357 memset(input_mems, 0x00, sizeof(input_mems)); 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()
539 …memcpy((char *)(input_mems[n][0]->virt_addr) + input_mems[n][0]->offset, input_data, input_attrs[n… in main()
545 uint8_t* dst_ptr = (uint8_t*)input_mems[n][0]->virt_addr+input_mems[n][0]->offset; in main()
606 if (input_mems[n][0]) { in main()
607 rknn_destroy_mem(ctx[n], input_mems[n][0]); in main()
|
| /OK3568_Linux_fs/external/rknpu2/examples/rknn_common_test/src/ |
| H A D | main.cc | 210 rknn_tensor_mem* input_mems[1]; in main() local
217 input_mems[0] = rknn_create_mem(ctx, input_attrs[0].size_with_stride); in main()
224 …memcpy(input_mems[0]->virt_addr, input_data, width * input_attrs[0].dims[1] * input_attrs[0].dims[… in main()
230 uint8_t* dst_ptr = (uint8_t*)input_mems[0]->virt_addr; in main()
251 ret = rknn_set_io_mem(ctx, input_mems[0], &input_attrs[0]); in main()
300 rknn_destroy_mem(ctx, input_mems[0]); in main()
|
| /OK3568_Linux_fs/external/rknpu2/examples/rknn_api_demo/src/ |
| H A D | rknn_create_mem_demo.cpp | 248 rknn_tensor_mem* input_mems[1]; in main() local
255 input_mems[0] = rknn_create_mem(ctx, input_attrs[0].size_with_stride); in main()
262 …memcpy(input_mems[0]->virt_addr, input_data, width * input_attrs[0].dims[1] * input_attrs[0].dims[… in main()
268 uint8_t* dst_ptr = (uint8_t*)input_mems[0]->virt_addr; in main()
289 ret = rknn_set_io_mem(ctx, input_mems[0], &input_attrs[0]); in main()
338 rknn_destroy_mem(ctx, input_mems[0]); in main()
|
| H A D | rknn_with_mmz_demo.cpp | 272 rknn_tensor_mem* input_mems[1]; in main() local
279 …input_mems[0] = rknn_create_mem_from_phys(ctx, input_phys, input_virt, input_attrs[0].size_with_st… in main()
285 …memcpy(input_mems[0]->virt_addr, input_data, width * input_attrs[0].dims[1] * input_attrs[0].dims[… in main()
291 uint8_t* dst_ptr = (uint8_t*)input_mems[0]->virt_addr; in main()
309 ret = rknn_set_io_mem(ctx, input_mems[0], &input_attrs[0]); in main()
362 rknn_destroy_mem(ctx, input_mems[0]); in main()
|
| H A D | rknn_create_mem_with_rga_demo.cpp | 262 rknn_tensor_mem* input_mems[1]; in main() local
268 input_mems[0] = rknn_create_mem(ctx, input_attrs[0].size_with_stride); in main()
274 …dst = wrapbuffer_fd_t(input_mems[0]->fd, model_in_width, model_in_height, wstride, hstride, in main()
295 ret = rknn_set_io_mem(ctx, input_mems[0], &input_attrs[0]); in main()
341 rknn_destroy_mem(ctx, input_mems[0]); in main()
|
| H A D | rknn_set_internal_mem_from_fd_demo.cpp | 333 rknn_tensor_mem* input_mems[1]; in main() local
340 …input_mems[0] = rknn_create_mem_from_fd(ctx, input_fd, input_virt, input_attrs[0].size_with_stride… in main()
346 …memcpy(input_mems[0]->virt_addr, input_data, width * input_attrs[0].dims[1] * input_attrs[0].dims[… in main()
352 uint8_t* dst_ptr = (uint8_t*)input_mems[0]->virt_addr; in main()
370 ret = rknn_set_io_mem(ctx, input_mems[0], &input_attrs[0]); in main()
425 rknn_destroy_mem(ctx, input_mems[0]); in main()
|
| /OK3568_Linux_fs/external/rknpu2/examples/librknn_api_android_demo/ |
| H A D | rknn_create_mem_demo.cpp | 248 rknn_tensor_mem* input_mems[1]; in main() local
255 input_mems[0] = rknn_create_mem(ctx, input_attrs[0].size_with_stride); in main()
262 …memcpy(input_mems[0]->virt_addr, input_data, width * input_attrs[0].dims[1] * input_attrs[0].dims[… in main()
268 uint8_t* dst_ptr = (uint8_t*)input_mems[0]->virt_addr; in main()
289 ret = rknn_set_io_mem(ctx, input_mems[0], &input_attrs[0]); in main()
338 rknn_destroy_mem(ctx, input_mems[0]); in main()
|
| /OK3568_Linux_fs/external/rknpu2/examples/RV1106_RV1103/rknn_mobilenet_demo/src/ |
| H A D | main_nhwc.cc | 332 rknn_tensor_mem *input_mems[1]; in main() local
339 input_mems[0] = rknn_create_mem(ctx, input_attrs[0].size_with_stride); in main()
347 …memcpy(input_mems[0]->virt_addr, input_data, width * input_attrs[0].dims[1] * input_attrs[0].dims[… in main()
355 uint8_t *dst_ptr = (uint8_t *)input_mems[0]->virt_addr; in main()
375 ret = rknn_set_io_mem(ctx, input_mems[0], &input_attrs[0]); in main()
432 rknn_destroy_mem(ctx, input_mems[0]); in main()
|
| H A D | main.cc | 362 rknn_tensor_mem *input_mems[1]; in main() local
369 input_mems[0] = rknn_create_mem(ctx, input_attrs[0].size_with_stride); in main()
377 …memcpy(input_mems[0]->virt_addr, input_data, width * input_attrs[0].dims[1] * input_attrs[0].dims[… in main()
385 uint8_t *dst_ptr = (uint8_t *)input_mems[0]->virt_addr; in main()
405 ret = rknn_set_io_mem(ctx, input_mems[0], &input_attrs[0]); in main()
507 rknn_destroy_mem(ctx, input_mems[0]); in main()
|
| /OK3568_Linux_fs/external/rknpu2/examples/rknn_yolov5_android_apk_demo/app/src/main/cpp/ |
| H A D | yolo_image.cc | 59 rknn_tensor_mem *input_mems[1]; variable
171 input_mems[0] = rknn_create_mem(ctx, input_attrs[0].size_with_stride * sizeof(char)); in create()
172 memset(input_mems[0]->virt_addr, 0, input_attrs[0].size_with_stride * sizeof(char)); in create()
182 rknn_set_io_mem(ctx, input_mems[0], &(input_attrs[0])); in create()
184 g_rga_dst = wrapbuffer_virtualaddr((void *)input_mems[0]->virt_addr, m_in_width, m_in_height, in create()
214 rknn_destroy_mem(ctx, input_mems[i]); in destroy()
|
| /OK3568_Linux_fs/external/rknpu2/examples/RV1106_RV1103/rknn_yolov5_demo/src/ |
| H A D | main.cc | 334 rknn_tensor_mem *input_mems[1]; in main() local
341 input_mems[0] = rknn_create_mem(ctx, input_attrs[0].size_with_stride); in main()
349 …memcpy(input_mems[0]->virt_addr, input_data, width * input_attrs[0].dims[1] * input_attrs[0].dims[… in main()
357 uint8_t *dst_ptr = (uint8_t *)input_mems[0]->virt_addr; in main()
377 ret = rknn_set_io_mem(ctx, input_mems[0], &input_attrs[0]); in main()
486 rknn_destroy_mem(ctx, input_mems[0]); in main()
|
| /OK3568_Linux_fs/external/rknpu2/examples/rknn_dynamic_shape_input_demo/src/ |
| H A D | rknn_dynshape_inference_zero_copy.cc | 462 rknn_tensor_mem *input_mems[io_num.n_input]; in main() local
471 input_mems[i] = rknn_create_mem(ctx, input_attrs[i].size_with_stride); in main()
553 memcpy(input_mems[i]->virt_addr, resize_imgs[i].data, input_size); in main()
561 uint8_t *dst_ptr = (uint8_t *)input_mems[i]->virt_addr; in main()
581 ret = rknn_set_io_mem(ctx, input_mems[i], &cur_input_attrs[i]); in main()
652 rknn_destroy_mem(ctx, input_mems[i]); in main()
|