Home
last modified time | relevance | path

Searched refs:wRoiStart (Results 1 – 3 of 3) sorted by relevance

/frameworks/ml/nn/common/operations/
DRoiAlign.cpp99 T_Roi wRoiStart = roiInfo[0] * widthScale; in roiAlignNhwc() local
104 T_Roi roiWidth = std::max(static_cast<float>(wRoiEnd - wRoiStart), 1.0f); in roiAlignNhwc()
122 T_Roi wStart = wStepSize * j + wRoiStart; in roiAlignNhwc()
123 T_Roi wEnd = wStepSize * (j + 1) + wRoiStart; in roiAlignNhwc()
209 float wRoiStart = static_cast<float>(roiInfo[0]) * widthScale * 0.125f; in roiAlignQuantNhwc() local
220 NN_RET_CHECK(wRoiStart <= inWidth); in roiAlignQuantNhwc()
224 NN_RET_CHECK_LE(wRoiStart, wRoiEnd); in roiAlignQuantNhwc()
227 float roiWidth = std::max(wRoiEnd - wRoiStart, 1.0f); in roiAlignQuantNhwc()
251 float wStart = wStepSize * j + wRoiStart; in roiAlignQuantNhwc()
252 float wEnd = wStepSize * (j + 1) + wRoiStart; in roiAlignQuantNhwc()
DRoiPooling.cpp95 int32_t wRoiStart = std::round(static_cast<float>(roiInfo[0] * widthScale)); in roiPoolingNhwc() local
101 T_Roi roiWidth = static_cast<T_Roi>(std::max(wRoiEnd - wRoiStart + 1, 1)); in roiPoolingNhwc()
111 uint32_t wStart = std::floor(static_cast<float>(wStepSize * j + wRoiStart)); in roiPoolingNhwc()
112 uint32_t wEnd = std::ceil(static_cast<float>(wStepSize * (j + 1) + wRoiStart)); in roiPoolingNhwc()
DHeatmapMaxKeypoint.cpp145 float wRoiStart = boxInfoBase[0]; in heatmapMaxKeypointFloat32Nhwc() local
149 float roiWidth = wRoiEnd - wRoiStart; in heatmapMaxKeypointFloat32Nhwc()
156 outputKeypointBase[0] = wRelativePos * roiWidth + wRoiStart; in heatmapMaxKeypointFloat32Nhwc()