Lines Matching refs:kRoiDim
242 const uint32_t kRoiDim = 4; in prepare() local
244 uint32_t numClasses = getSizeOfDimension(bboxDeltasShape, 1) / kRoiDim; in prepare()
248 NN_RET_CHECK_EQ(getSizeOfDimension(roiShape, 1), kRoiDim); in prepare()
250 NN_RET_CHECK_EQ(getSizeOfDimension(bboxDeltasShape, 1), kRoiDim * numClasses); in prepare()
262 outputShape.dimensions = {numRois, numClasses * kRoiDim}; in prepare()
504 const uint32_t kRoiDim = 4; in boxWithNmsLimitFloat32Compute() local
524 const float* roi = roiBase + i * kRoiDim; in boxWithNmsLimitFloat32Compute()
533 [&roiBase](uint32_t ind) { return roiBase + ind * kRoiDim; }, kernel, &result); in boxWithNmsLimitFloat32Compute()
544 roiBase += batchSplitIn->at(b) * numClasses * kRoiDim; in boxWithNmsLimitFloat32Compute()
569 const uint32_t kRoiDim = 4; in boxWithNmsLimitWriteOutput() local
605 memcpy(roiOutPtr, roiBase + index * kRoiDim, kRoiDim * sizeof(T_Roi)); in boxWithNmsLimitWriteOutput()
606 roiOutPtr += kRoiDim; in boxWithNmsLimitWriteOutput()
611 roiBase += batchSplitIn[b] * numClasses * kRoiDim; in boxWithNmsLimitWriteOutput()
768 const uint32_t kRoiDim = 4; in prepare() local
772 NN_RET_CHECK(getSizeOfDimension(roiShape, 1) == kRoiDim * numClasses); in prepare()
935 const uint32_t kRoiDim = 4; in filterBoxes() local
938 const float* roiInfo = roiBase + (*select)[j] * kRoiDim; in filterBoxes()
961 const uint32_t kRoiDim = 4; in generateProposalsNhwcFloat32Compute() local
969 uint32_t roiBufferSize = batchSize * kRoiDim; in generateProposalsNhwcFloat32Compute()
983 for (uint32_t a = 0; a < numAnchors; a++, roiBase += kRoiDim, anchorsBase += kRoiDim) { in generateProposalsNhwcFloat32Compute()
997 tempRoiShape.dimensions = {batchSize, kRoiDim}; in generateProposalsNhwcFloat32Compute()
999 tempBBoxDeltasShape.dimensions = {batchSize, kRoiDim}; in generateProposalsNhwcFloat32Compute()
1034 return roiTransformedBuffer.data() + ind * kRoiDim; in generateProposalsNhwcFloat32Compute()
1042 roiOutData->insert(roiOutData->end(), roiTransformedBuffer.begin() + i * kRoiDim, in generateProposalsNhwcFloat32Compute()
1043 roiTransformedBuffer.begin() + (i + 1) * kRoiDim); in generateProposalsNhwcFloat32Compute()
1295 const uint32_t kRoiDim = 4; in prepare() local
1304 NN_RET_CHECK_EQ(getSizeOfDimension(bboxDeltasShape, useNchw ? 1 : 3), numAnchors * kRoiDim); in prepare()
1308 NN_RET_CHECK_EQ(getSizeOfDimension(anchorsShape, 1), kRoiDim); in prepare()
1448 const uint32_t kRoiDim = 4; in detectionPostprocessFloat32() local
1465 std::vector<float> roiBuffer(numAnchors * kRoiDim); in detectionPostprocessFloat32()
1474 roiBuffer[a * kRoiDim] = yCtr - hHalf; in detectionPostprocessFloat32()
1475 roiBuffer[a * kRoiDim + 1] = xCtr - wHalf; in detectionPostprocessFloat32()
1476 roiBuffer[a * kRoiDim + 2] = yCtr + hHalf; in detectionPostprocessFloat32()
1477 roiBuffer[a * kRoiDim + 3] = xCtr + wHalf; in detectionPostprocessFloat32()
1478 anchorBase += kRoiDim; in detectionPostprocessFloat32()
1488 return roiBuffer.data() + (ind / numClasses) * kRoiDim; in detectionPostprocessFloat32()
1494 memcpy(roiOutBase + i * kRoiDim, &roiBuffer[(ind / numClasses) * kRoiDim], in detectionPostprocessFloat32()
1495 kRoiDim * sizeof(float)); in detectionPostprocessFloat32()
1514 [&roiBuffer](uint32_t ind) { return roiBuffer.data() + ind * kRoiDim; }, in detectionPostprocessFloat32()
1530 memcpy(roiOutPtr, &roiBuffer[i * kRoiDim], kRoiDim * sizeof(float)); in detectionPostprocessFloat32()
1531 roiOutPtr += kRoiDim; in detectionPostprocessFloat32()
1539 roiOutBase += numOutDetection * kRoiDim; in detectionPostprocessFloat32()
1619 const uint32_t kRoiDim = 4; in prepare() local
1632 NN_RET_CHECK_EQ(getSizeOfDimension(anchorsShape, 1), kRoiDim); in prepare()