Home
last modified time | relevance | path

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

/OK3568_Linux_fs/external/rknpu2/examples/RV1106_RV1103/rknn_yolov5_demo/src/
H A Dmain.cc427 int8_t *output_mems_nchw[io_num.n_output]; in main() local
431 output_mems_nchw[i] = (int8_t *)malloc(size); in main()
440 …NC1HWC2_int8_to_NCHW_int8((int8_t *)output_mems[i]->virt_addr, (int8_t *)output_mems_nchw[i], (int… in main()
471 …post_process((int8_t *)output_mems_nchw[0], (int8_t *)output_mems_nchw[1], (int8_t *)output_mems_n… in main()
490 free(output_mems_nchw[i]); in main()
/OK3568_Linux_fs/external/rknpu2/examples/RV1106_RV1103/rknn_mobilenet_demo/src/
H A Dmain.cc455 float *output_mems_nchw[io_num.n_output]; in main() local
459 output_mems_nchw[i] = (float *)malloc(size); in main()
471 …NC1HWC2_int8_to_NCHW_float((int8_t *)output_mems[i]->virt_addr, (float *)output_mems_nchw[i], (int… in main()
477 float *dst = output_mems_nchw[i]; in main()
495 float *buffer = (float *)output_mems_nchw[i]; in main()
511 free(output_mems_nchw[i]); in main()