Home
last modified time | relevance | path

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

/OK3568_Linux_fs/external/rknpu2/examples/RV1106_RV1103/rknn_mobilenet_demo/src/
H A Dmain_nhwc.cc41 …etTopN(float *pfProb, float *pfMaxProb, uint32_t *pMaxClass, uint32_t outputCount, uint32_t topNum) in rknn_GetTopN() argument
44 uint32_t top_count = outputCount > topNum ? topNum : outputCount; in rknn_GetTopN()
46 for (i = 0; i < topNum; ++i) in rknn_GetTopN()
75 uint32_t outputCount, uint32_t topNum) in rknn_GetTopN_int8() argument
78 uint32_t top_count = outputCount > topNum ? topNum : outputCount; in rknn_GetTopN_int8()
80 for (i = 0; i < topNum; ++i) in rknn_GetTopN_int8()
411 uint32_t topNum = 5; in main() local
414 uint32_t MaxClass[topNum]; in main()
415 float fMaxProb[topNum]; in main()
418 int top_count = sz > topNum ? topNum : sz; in main()
[all …]
H A Dmain.cc41 …etTopN(float *pfProb, float *pfMaxProb, uint32_t *pMaxClass, uint32_t outputCount, uint32_t topNum) in rknn_GetTopN() argument
44 uint32_t top_count = outputCount > topNum ? topNum : outputCount; in rknn_GetTopN()
46 for (i = 0; i < topNum; ++i) in rknn_GetTopN()
75 uint32_t outputCount, uint32_t topNum) in rknn_GetTopN_int8() argument
78 uint32_t top_count = outputCount > topNum ? topNum : outputCount; in rknn_GetTopN_int8()
80 for (i = 0; i < topNum; ++i) in rknn_GetTopN_int8()
486 uint32_t topNum = 5; in main() local
489 uint32_t MaxClass[topNum]; in main()
490 float fMaxProb[topNum]; in main()
493 int top_count = sz > topNum ? topNum : sz; in main()
[all …]
/OK3568_Linux_fs/external/rknpu2/examples/rknn_multiple_input_demo/src/
H A Dmain.cc33 …etTopN(float* pfProb, float* pfMaxProb, uint32_t* pMaxClass, uint32_t outputCount, uint32_t topNum) in rknn_GetTopN() argument
36 uint32_t top_count = outputCount > topNum ? topNum : outputCount; in rknn_GetTopN()
38 for (i = 0; i < topNum; ++i) { in rknn_GetTopN()
231 uint32_t topNum = 5; in main() local
233 uint32_t MaxClass[topNum]; in main()
234 float fMaxProb[topNum]; in main()
237 int top_count = sz > topNum ? topNum : sz; in main()
239 rknn_GetTopN(buffer, fMaxProb, MaxClass, sz, topNum); in main()
/OK3568_Linux_fs/external/rknpu2/examples/rknn_common_test/src/
H A Dmain.cc42 …etTopN(float* pfProb, float* pfMaxProb, uint32_t* pMaxClass, uint32_t outputCount, uint32_t topNum) in rknn_GetTopN() argument
45 uint32_t top_count = outputCount > topNum ? topNum : outputCount; in rknn_GetTopN()
47 for (i = 0; i < topNum; ++i) { in rknn_GetTopN()
283 uint32_t topNum = 5; in main() local
285 uint32_t MaxClass[topNum]; in main()
286 float fMaxProb[topNum]; in main()
289 int top_count = sz > topNum ? topNum : sz; in main()
291 rknn_GetTopN(buffer, fMaxProb, MaxClass, sz, topNum); in main()
/OK3568_Linux_fs/external/rknpu2/examples/rknn_api_demo/src/
H A Drknn_create_mem_demo.cpp41 …etTopN(float* pfProb, float* pfMaxProb, uint32_t* pMaxClass, uint32_t outputCount, uint32_t topNum) in rknn_GetTopN() argument
44 uint32_t top_count = outputCount > topNum ? topNum : outputCount; in rknn_GetTopN()
46 for (i = 0; i < topNum; ++i) { in rknn_GetTopN()
321 uint32_t topNum = 5; in main() local
323 uint32_t MaxClass[topNum]; in main()
324 float fMaxProb[topNum]; in main()
327 int top_count = sz > topNum ? topNum : sz; in main()
329 rknn_GetTopN(buffer, fMaxProb, MaxClass, sz, topNum); in main()
H A Drknn_create_mem_with_rga_demo.cpp47 …etTopN(float* pfProb, float* pfMaxProb, uint32_t* pMaxClass, uint32_t outputCount, uint32_t topNum) in rknn_GetTopN() argument
50 uint32_t top_count = outputCount > topNum ? topNum : outputCount; in rknn_GetTopN()
52 for (i = 0; i < topNum; ++i) { in rknn_GetTopN()
324 uint32_t topNum = 5; in main() local
326 uint32_t MaxClass[topNum]; in main()
327 float fMaxProb[topNum]; in main()
330 int top_count = sz > topNum ? topNum : sz; in main()
332 rknn_GetTopN(buffer, fMaxProb, MaxClass, sz, topNum); in main()
H A Drknn_with_mmz_demo.cpp42 …etTopN(float* pfProb, float* pfMaxProb, uint32_t* pMaxClass, uint32_t outputCount, uint32_t topNum) in rknn_GetTopN() argument
45 uint32_t top_count = outputCount > topNum ? topNum : outputCount; in rknn_GetTopN()
47 for (i = 0; i < topNum; ++i) { in rknn_GetTopN()
339 uint32_t topNum = 5; in main() local
341 uint32_t MaxClass[topNum]; in main()
342 float fMaxProb[topNum]; in main()
345 int top_count = sz > topNum ? topNum : sz; in main()
347 rknn_GetTopN(buffer, fMaxProb, MaxClass, sz, topNum); in main()
H A Drknn_set_internal_mem_from_fd_demo.cpp42 …etTopN(float* pfProb, float* pfMaxProb, uint32_t* pMaxClass, uint32_t outputCount, uint32_t topNum) in rknn_GetTopN() argument
45 uint32_t top_count = outputCount > topNum ? topNum : outputCount; in rknn_GetTopN()
47 for (i = 0; i < topNum; ++i) { in rknn_GetTopN()
400 uint32_t topNum = 5; in main() local
402 uint32_t MaxClass[topNum]; in main()
403 float fMaxProb[topNum]; in main()
406 int top_count = sz > topNum ? topNum : sz; in main()
408 rknn_GetTopN(buffer, fMaxProb, MaxClass, sz, topNum); in main()
/OK3568_Linux_fs/external/rknpu2/examples/librknn_api_android_demo/
H A Drknn_create_mem_demo.cpp41 …etTopN(float* pfProb, float* pfMaxProb, uint32_t* pMaxClass, uint32_t outputCount, uint32_t topNum) in rknn_GetTopN() argument
44 uint32_t top_count = outputCount > topNum ? topNum : outputCount; in rknn_GetTopN()
46 for (i = 0; i < topNum; ++i) { in rknn_GetTopN()
321 uint32_t topNum = 5; in main() local
323 uint32_t MaxClass[topNum]; in main()
324 float fMaxProb[topNum]; in main()
327 int top_count = sz > topNum ? topNum : sz; in main()
329 rknn_GetTopN(buffer, fMaxProb, MaxClass, sz, topNum); in main()
/OK3568_Linux_fs/external/rknpu2/examples/rknn_benchmark/src/
H A Drknn_benchmark.cpp45 …etTopN(float* pfProb, float* pfMaxProb, uint32_t* pMaxClass, uint32_t outputCount, uint32_t topNum) in rknn_GetTopN() argument
48 uint32_t top_count = outputCount > topNum ? topNum : outputCount; in rknn_GetTopN()
50 for (i = 0; i < topNum; ++i) { in rknn_GetTopN()
260 uint32_t topNum = 5; in main() local
469 uint32_t MaxClass[topNum]; in main()
470 float fMaxProb[topNum]; in main()
473 int top_count = sz > topNum ? topNum : sz; in main()
475 rknn_GetTopN(buffer, fMaxProb, MaxClass, sz, topNum); in main()
/OK3568_Linux_fs/external/rknpu2/examples/rknn_internal_mem_reuse_demo/src/
H A Drknn_internal_mem_reuse_demo.cc50 …etTopN(float* pfProb, float* pfMaxProb, uint32_t* pMaxClass, uint32_t outputCount, uint32_t topNum) in rknn_GetTopN() argument
53 uint32_t top_count = outputCount > topNum ? topNum : outputCount; in rknn_GetTopN()
55 for (i = 0; i < topNum; ++i) { in rknn_GetTopN()
576 uint32_t topNum = 5; in main() local
578 uint32_t MaxClass[topNum]; in main()
579 float fMaxProb[topNum]; in main()
583 int top_count = sz > topNum ? topNum : sz; in main()
585 rknn_GetTopN(buffer, fMaxProb, MaxClass, sz, topNum); in main()
/OK3568_Linux_fs/external/rknpu2/examples/rknn_mobilenet_demo/src/
H A Dmain.cc70 …GetTop(float* pfProb, float* pfMaxProb, uint32_t* pMaxClass, uint32_t outputCount, uint32_t topNum) in rknn_GetTop() argument
75 if (topNum > MAX_TOP_NUM) in rknn_GetTop()
78 memset(pfMaxProb, 0, sizeof(float) * topNum); in rknn_GetTop()
79 memset(pMaxClass, 0xff, sizeof(float) * topNum); in rknn_GetTop()
81 for (j = 0; j < topNum; j++) { in rknn_GetTop()
/OK3568_Linux_fs/external/rknpu2/examples/rknn_dynamic_shape_input_demo/src/
H A Drknn_dynshape_inference.cc40 …etTopN(float *pfProb, float *pfMaxProb, uint32_t *pMaxClass, uint32_t outputCount, uint32_t topNum) in rknn_GetTopN() argument
43 uint32_t top_count = outputCount > topNum ? topNum : outputCount; in rknn_GetTopN()
45 for (i = 0; i < topNum; ++i) in rknn_GetTopN()
H A Drknn_dynshape_inference_zero_copy.cc40 …etTopN(float *pfProb, float *pfMaxProb, uint32_t *pMaxClass, uint32_t outputCount, uint32_t topNum) in rknn_GetTopN() argument
43 uint32_t top_count = outputCount > topNum ? topNum : outputCount; in rknn_GetTopN()
45 for (i = 0; i < topNum; ++i) in rknn_GetTopN()