Searched refs:topkData (Results 1 – 1 of 1) sorted by relevance
| /dports/misc/mnn/MNN-1.2.0/source/backend/cpu/ |
| H A D | CPUTopKV2.cpp | 112 float* topkData = outputData->host<float>(); in onExecute() local
117 float* rowTopkData = topkData + i * k; in onExecute()
132 int32_t* topkData = outputData->host<int32_t>(); in onExecute() local
136 int32_t* rowTopkData = topkData + i * k; in onExecute()
159 auto topkData = outputData->host<float>(); in onExecute() local
161 findTopK<float>(rowSize, numRows, inputData, k, indicesData, topkData); in onExecute()
164 auto topkData = outputData->host<int32_t>(); in onExecute() local
166 findTopK<int32_t>(rowSize, numRows, inputData, k, indicesData, topkData); in onExecute()
|