Home
last modified time | relevance | path

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

/OK3568_Linux_fs/external/rknpu2/examples/rknn_internal_mem_reuse_demo/src/
H A Drknn_internal_mem_reuse_demo.cc315 #define TOTAL_RKNN_MODEL_NUM 2 macro
337 char *model_path[TOTAL_RKNN_MODEL_NUM]; in main()
338 char *input_path[TOTAL_RKNN_MODEL_NUM]; in main()
339 rknn_context ctx[TOTAL_RKNN_MODEL_NUM]; in main()
340 rknn_mem_size mem_size[TOTAL_RKNN_MODEL_NUM]; in main()
341 rknn_input_output_num io_num[TOTAL_RKNN_MODEL_NUM]; in main()
342 rknn_tensor_mem* internal_mem[TOTAL_RKNN_MODEL_NUM]; in main()
343 rknn_tensor_mem *weight_mems[TOTAL_RKNN_MODEL_NUM]; in main()
344 rknn_tensor_attr input_attrs[TOTAL_RKNN_MODEL_NUM][1]; // this demo only support one input in main()
345 rknn_tensor_attr output_attrs[TOTAL_RKNN_MODEL_NUM][MAX_OUTPUT_NUM]; in main()
[all …]