Home
last modified time | relevance | path

Searched refs:rknn_tensor_mem (Results 1 – 24 of 24) sorted by relevance

/OK3568_Linux_fs/external/rknpu2/examples/librknn_api_android_demo/include/
H A Drknn_api.h326 } rknn_tensor_mem; typedef
565 rknn_tensor_mem* rknn_create_mem_from_phys(rknn_context ctx, uint64_t phys_addr, void *virt_addr, u…
581 rknn_tensor_mem* rknn_create_mem_from_fd(rknn_context ctx, int32_t fd, void *virt_addr, uint32_t si…
595 rknn_tensor_mem* rknn_create_mem_from_mb_blk(rknn_context ctx, void *mb_blk, int32_t offset);
608 rknn_tensor_mem* rknn_create_mem(rknn_context ctx, uint32_t size);
621 int rknn_destroy_mem(rknn_context ctx, rknn_tensor_mem *mem);
634 int rknn_set_weight_mem(rknn_context ctx, rknn_tensor_mem *mem);
647 int rknn_set_internal_mem(rknn_context ctx, rknn_tensor_mem *mem);
661 int rknn_set_io_mem(rknn_context ctx, rknn_tensor_mem *mem, rknn_tensor_attr *attr);
/OK3568_Linux_fs/external/rknpu2/runtime/RK3588/Android/librknn_api/include/
H A Drknn_api.h344 } rknn_tensor_mem; typedef
583 rknn_tensor_mem* rknn_create_mem_from_phys(rknn_context ctx, uint64_t phys_addr, void *virt_addr, u…
599 rknn_tensor_mem* rknn_create_mem_from_fd(rknn_context ctx, int32_t fd, void *virt_addr, uint32_t si…
613 rknn_tensor_mem* rknn_create_mem_from_mb_blk(rknn_context ctx, void *mb_blk, int32_t offset);
626 rknn_tensor_mem* rknn_create_mem(rknn_context ctx, uint32_t size);
639 int rknn_destroy_mem(rknn_context ctx, rknn_tensor_mem *mem);
652 int rknn_set_weight_mem(rknn_context ctx, rknn_tensor_mem *mem);
665 int rknn_set_internal_mem(rknn_context ctx, rknn_tensor_mem *mem);
679 int rknn_set_io_mem(rknn_context ctx, rknn_tensor_mem *mem, rknn_tensor_attr *attr);
H A Drknn_matmul_api.h214 int rknn_matmul_set_io_mem(rknn_matmul_ctx ctx, rknn_tensor_mem* mem, rknn_matmul_tensor_attr* attr…
/OK3568_Linux_fs/external/rknpu2/runtime/RK356X/Linux/librknn_api/include/
H A Drknn_api.h344 } rknn_tensor_mem; typedef
583 rknn_tensor_mem* rknn_create_mem_from_phys(rknn_context ctx, uint64_t phys_addr, void *virt_addr, u…
599 rknn_tensor_mem* rknn_create_mem_from_fd(rknn_context ctx, int32_t fd, void *virt_addr, uint32_t si…
613 rknn_tensor_mem* rknn_create_mem_from_mb_blk(rknn_context ctx, void *mb_blk, int32_t offset);
626 rknn_tensor_mem* rknn_create_mem(rknn_context ctx, uint32_t size);
639 int rknn_destroy_mem(rknn_context ctx, rknn_tensor_mem *mem);
652 int rknn_set_weight_mem(rknn_context ctx, rknn_tensor_mem *mem);
665 int rknn_set_internal_mem(rknn_context ctx, rknn_tensor_mem *mem);
679 int rknn_set_io_mem(rknn_context ctx, rknn_tensor_mem *mem, rknn_tensor_attr *attr);
H A Drknn_matmul_api.h143 int rknn_matmul_set_io_mem(rknn_matmul_ctx ctx, rknn_tensor_mem* mem, rknn_matmul_tensor_attr* attr…
/OK3568_Linux_fs/external/rknpu2/runtime/RK3588/Linux/librknn_api/include/
H A Drknn_api.h344 } rknn_tensor_mem; typedef
583 rknn_tensor_mem* rknn_create_mem_from_phys(rknn_context ctx, uint64_t phys_addr, void *virt_addr, u…
599 rknn_tensor_mem* rknn_create_mem_from_fd(rknn_context ctx, int32_t fd, void *virt_addr, uint32_t si…
613 rknn_tensor_mem* rknn_create_mem_from_mb_blk(rknn_context ctx, void *mb_blk, int32_t offset);
626 rknn_tensor_mem* rknn_create_mem(rknn_context ctx, uint32_t size);
639 int rknn_destroy_mem(rknn_context ctx, rknn_tensor_mem *mem);
652 int rknn_set_weight_mem(rknn_context ctx, rknn_tensor_mem *mem);
665 int rknn_set_internal_mem(rknn_context ctx, rknn_tensor_mem *mem);
679 int rknn_set_io_mem(rknn_context ctx, rknn_tensor_mem *mem, rknn_tensor_attr *attr);
H A Drknn_matmul_api.h214 int rknn_matmul_set_io_mem(rknn_matmul_ctx ctx, rknn_tensor_mem* mem, rknn_matmul_tensor_attr* attr…
/OK3568_Linux_fs/external/rknpu2/runtime/RK356X/Android/librknn_api/include/
H A Drknn_api.h344 } rknn_tensor_mem; typedef
583 rknn_tensor_mem* rknn_create_mem_from_phys(rknn_context ctx, uint64_t phys_addr, void *virt_addr, u…
599 rknn_tensor_mem* rknn_create_mem_from_fd(rknn_context ctx, int32_t fd, void *virt_addr, uint32_t si…
613 rknn_tensor_mem* rknn_create_mem_from_mb_blk(rknn_context ctx, void *mb_blk, int32_t offset);
626 rknn_tensor_mem* rknn_create_mem(rknn_context ctx, uint32_t size);
639 int rknn_destroy_mem(rknn_context ctx, rknn_tensor_mem *mem);
652 int rknn_set_weight_mem(rknn_context ctx, rknn_tensor_mem *mem);
665 int rknn_set_internal_mem(rknn_context ctx, rknn_tensor_mem *mem);
679 int rknn_set_io_mem(rknn_context ctx, rknn_tensor_mem *mem, rknn_tensor_attr *attr);
H A Drknn_matmul_api.h143 int rknn_matmul_set_io_mem(rknn_matmul_ctx ctx, rknn_tensor_mem* mem, rknn_matmul_tensor_attr* attr…
/OK3568_Linux_fs/external/rknpu2/runtime/RV1106/Linux/librknn_api/include/
H A Drknn_api.h344 } rknn_tensor_mem; typedef
583 rknn_tensor_mem* rknn_create_mem_from_phys(rknn_context ctx, uint64_t phys_addr, void *virt_addr, u…
599 rknn_tensor_mem* rknn_create_mem_from_fd(rknn_context ctx, int32_t fd, void *virt_addr, uint32_t si…
613 rknn_tensor_mem* rknn_create_mem_from_mb_blk(rknn_context ctx, void *mb_blk, int32_t offset);
626 rknn_tensor_mem* rknn_create_mem(rknn_context ctx, uint32_t size);
639 int rknn_destroy_mem(rknn_context ctx, rknn_tensor_mem *mem);
652 int rknn_set_weight_mem(rknn_context ctx, rknn_tensor_mem *mem);
665 int rknn_set_internal_mem(rknn_context ctx, rknn_tensor_mem *mem);
679 int rknn_set_io_mem(rknn_context ctx, rknn_tensor_mem *mem, rknn_tensor_attr *attr);
/OK3568_Linux_fs/external/rknpu2/examples/rknn_yolov5_android_apk_demo/app/src/main/cpp/
H A Drknn_api.h344 } rknn_tensor_mem; typedef
583 rknn_tensor_mem* rknn_create_mem_from_phys(rknn_context ctx, uint64_t phys_addr, void *virt_addr, u…
599 rknn_tensor_mem* rknn_create_mem_from_fd(rknn_context ctx, int32_t fd, void *virt_addr, uint32_t si…
613 rknn_tensor_mem* rknn_create_mem_from_mb_blk(rknn_context ctx, void *mb_blk, int32_t offset);
626 rknn_tensor_mem* rknn_create_mem(rknn_context ctx, uint32_t size);
639 int rknn_destroy_mem(rknn_context ctx, rknn_tensor_mem *mem);
652 int rknn_set_weight_mem(rknn_context ctx, rknn_tensor_mem *mem);
665 int rknn_set_internal_mem(rknn_context ctx, rknn_tensor_mem *mem);
679 int rknn_set_io_mem(rknn_context ctx, rknn_tensor_mem *mem, rknn_tensor_attr *attr);
H A Dyolo_image.cc59 rknn_tensor_mem *input_mems[1];
60 rknn_tensor_mem *output_mems[3];
/OK3568_Linux_fs/external/rknpu2/examples/rknn_matmul_api_demo/src/
H A Drknn_matmul_demo.cpp55 static void dump_matmul_tensor(rknn_tensor_mem* tensor, rknn_matmul_tensor_attr* attr) in dump_matmul_tensor()
178 rknn_tensor_mem* A = rknn_create_mem(ctx, io_attr.A.size); in main()
186 rknn_tensor_mem* B = rknn_create_mem(ctx, io_attr.B.size); in main()
215 rknn_tensor_mem* C = rknn_create_mem(ctx, io_attr.C.size); in main()
/OK3568_Linux_fs/external/rknpu2/examples/rknn_api_demo/src/
H A Drknn_set_internal_mem_from_fd_demo.cpp235 rknn_tensor_mem* weight_mem; in main()
261 rknn_tensor_mem* internal_mem; in main()
333 rknn_tensor_mem* input_mems[1]; in main()
364 rknn_tensor_mem* output_mems[io_num.n_output]; in main()
H A Drknn_create_mem_demo.cpp248 rknn_tensor_mem* input_mems[1]; in main()
280 rknn_tensor_mem* output_mems[io_num.n_output]; in main()
H A Drknn_create_mem_with_rga_demo.cpp262 rknn_tensor_mem* input_mems[1]; in main()
286 rknn_tensor_mem* output_mems[io_num.n_output]; in main()
H A Drknn_with_mmz_demo.cpp272 rknn_tensor_mem* input_mems[1]; in main()
303 rknn_tensor_mem* output_mems[io_num.n_output]; in main()
/OK3568_Linux_fs/external/rknpu2/examples/rknn_internal_mem_reuse_demo/src/
H A Drknn_internal_mem_reuse_demo.cc342 rknn_tensor_mem* internal_mem[TOTAL_RKNN_MODEL_NUM]; in main()
343 rknn_tensor_mem *weight_mems[TOTAL_RKNN_MODEL_NUM]; in main()
346 rknn_tensor_mem* input_mems[TOTAL_RKNN_MODEL_NUM][1]; // this demo only support one input in main()
347 rknn_tensor_mem* output_mems[TOTAL_RKNN_MODEL_NUM][MAX_OUTPUT_NUM]; in main()
348 rknn_tensor_mem* internal_mem_max = NULL; in main()
/OK3568_Linux_fs/external/rknpu2/examples/rknn_common_test/src/
H A Dmain.cc210 rknn_tensor_mem* input_mems[1]; in main()
242 rknn_tensor_mem* output_mems[io_num.n_output]; in main()
/OK3568_Linux_fs/external/rknpu2/examples/librknn_api_android_demo/
H A Drknn_create_mem_demo.cpp248 rknn_tensor_mem* input_mems[1]; in main()
280 rknn_tensor_mem* output_mems[io_num.n_output]; in main()
/OK3568_Linux_fs/external/rknpu2/examples/RV1106_RV1103/rknn_mobilenet_demo/src/
H A Dmain_nhwc.cc332 rknn_tensor_mem *input_mems[1]; in main()
368 rknn_tensor_mem *output_mems[io_num.n_output]; in main()
H A Dmain.cc362 rknn_tensor_mem *input_mems[1]; in main()
398 rknn_tensor_mem *output_mems[io_num.n_output]; in main()
/OK3568_Linux_fs/external/rknpu2/examples/RV1106_RV1103/rknn_yolov5_demo/src/
H A Dmain.cc334 rknn_tensor_mem *input_mems[1]; in main()
370 rknn_tensor_mem *output_mems[io_num.n_output]; in main()
/OK3568_Linux_fs/external/rknpu2/examples/rknn_dynamic_shape_input_demo/src/
H A Drknn_dynshape_inference_zero_copy.cc462 rknn_tensor_mem *input_mems[io_num.n_input]; in main()
475 rknn_tensor_mem *output_mems[io_num.n_output]; in main()