Searched refs:internal_mem_max (Results 1 – 2 of 2) sorted by relevance
| /OK3568_Linux_fs/external/rknpu2/examples/rknn_internal_mem_reuse_demo/ |
| H A D | README.md | 23 internal_mem_max = rknn_create_mem(ctx_a, max_internal_size);
25 internal_mem_a = rknn_create_mem_from_fd(ctx_a, internal_mem_max->fd,
26 internal_mem_max->virt_addr, mem_size_a.total_internal_size, 0);
29 internal_mem_b = rknn_create_mem_from_fd(ctx_b, internal_mem_max->fd,
30 internal_mem_max->virt_addr, mem_size_a.total_internal_size, 0);
|
| /OK3568_Linux_fs/external/rknpu2/examples/rknn_internal_mem_reuse_demo/src/ |
| H A D | rknn_internal_mem_reuse_demo.cc | 348 rknn_tensor_mem* internal_mem_max = NULL; in main() local
457 internal_mem_max = rknn_create_mem(ctx[0], max_internal_size); in main()
460 internal_mem[n] = rknn_create_mem_from_fd(ctx[n], internal_mem_max->fd, in main()
461 … internal_mem_max->virt_addr, mem_size[n].total_internal_size, 0); in main()
598 if (internal_mem_max) { in main()
599 rknn_destroy_mem(ctx[0], internal_mem_max); in main()
|