Home
last modified time | relevance | path

Searched refs:pMaxClass (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 static int rknn_GetTopN(float *pfProb, float *pfMaxProb, uint32_t *pMaxClass, uint32_t outputCount,… in rknn_GetTopN() argument
49 pMaxClass[i] = -1; in rknn_GetTopN()
56 …if ((i == *(pMaxClass + 0)) || (i == *(pMaxClass + 1)) || (i == *(pMaxClass + 2)) || (i == *(pMaxC… in rknn_GetTopN()
57 (i == *(pMaxClass + 4))) in rknn_GetTopN()
66 *(pMaxClass + j) = i; in rknn_GetTopN()
74 …ic int rknn_GetTopN_int8(int8_t *pProb, float scale, int zp, float *pfMaxProb, uint32_t *pMaxClass, in rknn_GetTopN_int8() argument
83 pMaxClass[i] = -1; in rknn_GetTopN_int8()
90 …if ((i == *(pMaxClass + 0)) || (i == *(pMaxClass + 1)) || (i == *(pMaxClass + 2)) || (i == *(pMaxC… in rknn_GetTopN_int8()
91 (i == *(pMaxClass + 4))) in rknn_GetTopN_int8()
100 *(pMaxClass + j) = i; in rknn_GetTopN_int8()
H A Dmain.cc41 static int rknn_GetTopN(float *pfProb, float *pfMaxProb, uint32_t *pMaxClass, uint32_t outputCount,… in rknn_GetTopN() argument
49 pMaxClass[i] = -1; in rknn_GetTopN()
56 …if ((i == *(pMaxClass + 0)) || (i == *(pMaxClass + 1)) || (i == *(pMaxClass + 2)) || (i == *(pMaxC… in rknn_GetTopN()
57 (i == *(pMaxClass + 4))) in rknn_GetTopN()
66 *(pMaxClass + j) = i; in rknn_GetTopN()
74 …ic int rknn_GetTopN_int8(int8_t *pProb, float scale, int zp, float *pfMaxProb, uint32_t *pMaxClass, in rknn_GetTopN_int8() argument
83 pMaxClass[i] = -1; in rknn_GetTopN_int8()
90 …if ((i == *(pMaxClass + 0)) || (i == *(pMaxClass + 1)) || (i == *(pMaxClass + 2)) || (i == *(pMaxC… in rknn_GetTopN_int8()
91 (i == *(pMaxClass + 4))) in rknn_GetTopN_int8()
100 *(pMaxClass + j) = i; in rknn_GetTopN_int8()
/OK3568_Linux_fs/external/rknpu2/examples/rknn_mobilenet_demo/src/
H A Dmain.cc70 static int rknn_GetTop(float* pfProb, float* pfMaxProb, uint32_t* pMaxClass, uint32_t outputCount, … in rknn_GetTop() argument
79 memset(pMaxClass, 0xff, sizeof(float) * topNum); in rknn_GetTop()
83 …if ((i == *(pMaxClass + 0)) || (i == *(pMaxClass + 1)) || (i == *(pMaxClass + 2)) || (i == *(pMaxC… in rknn_GetTop()
84 (i == *(pMaxClass + 4))) { in rknn_GetTop()
90 *(pMaxClass + j) = i; in rknn_GetTop()
/OK3568_Linux_fs/external/rknpu2/examples/rknn_multiple_input_demo/src/
H A Dmain.cc33 static int rknn_GetTopN(float* pfProb, float* pfMaxProb, uint32_t* pMaxClass, uint32_t outputCount,… in rknn_GetTopN() argument
40 pMaxClass[i] = -1; in rknn_GetTopN()
45 …if ((i == *(pMaxClass + 0)) || (i == *(pMaxClass + 1)) || (i == *(pMaxClass + 2)) || (i == *(pMaxC… in rknn_GetTopN()
46 (i == *(pMaxClass + 4))) { in rknn_GetTopN()
52 *(pMaxClass + j) = i; in rknn_GetTopN()
/OK3568_Linux_fs/external/rknpu2/examples/rknn_common_test/src/
H A Dmain.cc42 static int rknn_GetTopN(float* pfProb, float* pfMaxProb, uint32_t* pMaxClass, uint32_t outputCount,… in rknn_GetTopN() argument
49 pMaxClass[i] = -1; in rknn_GetTopN()
54 …if ((i == *(pMaxClass + 0)) || (i == *(pMaxClass + 1)) || (i == *(pMaxClass + 2)) || (i == *(pMaxC… in rknn_GetTopN()
55 (i == *(pMaxClass + 4))) { in rknn_GetTopN()
61 *(pMaxClass + j) = i; in rknn_GetTopN()
/OK3568_Linux_fs/external/rknpu2/examples/rknn_api_demo/src/
H A Drknn_create_mem_demo.cpp41 static int rknn_GetTopN(float* pfProb, float* pfMaxProb, uint32_t* pMaxClass, uint32_t outputCount,… in rknn_GetTopN() argument
48 pMaxClass[i] = -1; in rknn_GetTopN()
53 …if ((i == *(pMaxClass + 0)) || (i == *(pMaxClass + 1)) || (i == *(pMaxClass + 2)) || (i == *(pMaxC… in rknn_GetTopN()
54 (i == *(pMaxClass + 4))) { in rknn_GetTopN()
60 *(pMaxClass + j) = i; in rknn_GetTopN()
H A Drknn_create_mem_with_rga_demo.cpp47 static int rknn_GetTopN(float* pfProb, float* pfMaxProb, uint32_t* pMaxClass, uint32_t outputCount,… in rknn_GetTopN() argument
54 pMaxClass[i] = -1; in rknn_GetTopN()
59 …if ((i == *(pMaxClass + 0)) || (i == *(pMaxClass + 1)) || (i == *(pMaxClass + 2)) || (i == *(pMaxC… in rknn_GetTopN()
60 (i == *(pMaxClass + 4))) { in rknn_GetTopN()
66 *(pMaxClass + j) = i; in rknn_GetTopN()
H A Drknn_with_mmz_demo.cpp42 static int rknn_GetTopN(float* pfProb, float* pfMaxProb, uint32_t* pMaxClass, uint32_t outputCount,… in rknn_GetTopN() argument
49 pMaxClass[i] = -1; in rknn_GetTopN()
54 …if ((i == *(pMaxClass + 0)) || (i == *(pMaxClass + 1)) || (i == *(pMaxClass + 2)) || (i == *(pMaxC… in rknn_GetTopN()
55 (i == *(pMaxClass + 4))) { in rknn_GetTopN()
61 *(pMaxClass + j) = i; in rknn_GetTopN()
H A Drknn_set_internal_mem_from_fd_demo.cpp42 static int rknn_GetTopN(float* pfProb, float* pfMaxProb, uint32_t* pMaxClass, uint32_t outputCount,… in rknn_GetTopN() argument
49 pMaxClass[i] = -1; in rknn_GetTopN()
54 …if ((i == *(pMaxClass + 0)) || (i == *(pMaxClass + 1)) || (i == *(pMaxClass + 2)) || (i == *(pMaxC… in rknn_GetTopN()
55 (i == *(pMaxClass + 4))) { in rknn_GetTopN()
61 *(pMaxClass + j) = i; in rknn_GetTopN()
/OK3568_Linux_fs/external/rknpu2/examples/librknn_api_android_demo/
H A Drknn_create_mem_demo.cpp41 static int rknn_GetTopN(float* pfProb, float* pfMaxProb, uint32_t* pMaxClass, uint32_t outputCount,… in rknn_GetTopN() argument
48 pMaxClass[i] = -1; in rknn_GetTopN()
53 …if ((i == *(pMaxClass + 0)) || (i == *(pMaxClass + 1)) || (i == *(pMaxClass + 2)) || (i == *(pMaxC… in rknn_GetTopN()
54 (i == *(pMaxClass + 4))) { in rknn_GetTopN()
60 *(pMaxClass + j) = i; in rknn_GetTopN()
/OK3568_Linux_fs/external/rknpu2/examples/rknn_benchmark/src/
H A Drknn_benchmark.cpp45 static int rknn_GetTopN(float* pfProb, float* pfMaxProb, uint32_t* pMaxClass, uint32_t outputCount,… in rknn_GetTopN() argument
52 pMaxClass[i] = -1; in rknn_GetTopN()
57 …if ((i == *(pMaxClass + 0)) || (i == *(pMaxClass + 1)) || (i == *(pMaxClass + 2)) || (i == *(pMaxC… in rknn_GetTopN()
58 (i == *(pMaxClass + 4))) { in rknn_GetTopN()
64 *(pMaxClass + j) = i; in rknn_GetTopN()
/OK3568_Linux_fs/external/rknpu2/examples/rknn_internal_mem_reuse_demo/src/
H A Drknn_internal_mem_reuse_demo.cc50 static int rknn_GetTopN(float* pfProb, float* pfMaxProb, uint32_t* pMaxClass, uint32_t outputCount,… in rknn_GetTopN() argument
57 pMaxClass[i] = -1; in rknn_GetTopN()
62 …if ((i == *(pMaxClass + 0)) || (i == *(pMaxClass + 1)) || (i == *(pMaxClass + 2)) || (i == *(pMaxC… in rknn_GetTopN()
63 (i == *(pMaxClass + 4))) { in rknn_GetTopN()
69 *(pMaxClass + j) = i; in rknn_GetTopN()
/OK3568_Linux_fs/external/rknpu2/examples/rknn_dynamic_shape_input_demo/src/
H A Drknn_dynshape_inference.cc40 static int rknn_GetTopN(float *pfProb, float *pfMaxProb, uint32_t *pMaxClass, uint32_t outputCount,… in rknn_GetTopN() argument
48 pMaxClass[i] = -1; in rknn_GetTopN()
55 …if ((i == *(pMaxClass + 0)) || (i == *(pMaxClass + 1)) || (i == *(pMaxClass + 2)) || (i == *(pMaxC… in rknn_GetTopN()
56 (i == *(pMaxClass + 4))) in rknn_GetTopN()
64 *(pMaxClass + j) = i; in rknn_GetTopN()
H A Drknn_dynshape_inference_zero_copy.cc40 static int rknn_GetTopN(float *pfProb, float *pfMaxProb, uint32_t *pMaxClass, uint32_t outputCount,… in rknn_GetTopN() argument
48 pMaxClass[i] = -1; in rknn_GetTopN()
55 …if ((i == *(pMaxClass + 0)) || (i == *(pMaxClass + 1)) || (i == *(pMaxClass + 2)) || (i == *(pMaxC… in rknn_GetTopN()
56 (i == *(pMaxClass + 4))) in rknn_GetTopN()
64 *(pMaxClass + j) = i; in rknn_GetTopN()