Searched refs:inWidth (Results 1 – 4 of 4) sorted by relevance
/packages/modules/NeuralNetworks/common/cpu_operations/ |
D | RoiAlign.cpp | 60 uint32_t inWidth = getSizeOfDimension(inputShape, 2); in roiAlignNhwc() local 82 NN_RET_CHECK(roiInfo[0] * widthScale <= inWidth); in roiAlignNhwc() 84 NN_RET_CHECK(roiInfo[2] * widthScale <= inWidth); in roiAlignNhwc() 109 const T_Input* batchBase = inputData + batchId * inHeight * inWidth * inDepth; in roiAlignNhwc() 135 if (x1 >= inWidth - 1) { in roiAlignNhwc() 136 x1 = x2 = inWidth - 1; in roiAlignNhwc() 146 uint32_t offsets[] = {y1 * inWidth * inDepth + x1 * inDepth, in roiAlignNhwc() 147 y1 * inWidth * inDepth + x2 * inDepth, in roiAlignNhwc() 148 y2 * inWidth * inDepth + x1 * inDepth, in roiAlignNhwc() 149 y2 * inWidth * inDepth + x2 * inDepth}; in roiAlignNhwc() [all …]
|
D | ResizeImageOps.cpp | 65 const int inWidth = getSizeOfDimension(inputShape, 2); in resizeNearestNeighbor() local 71 const float widthScale = calculateResizeScale(inWidth, outWidth, alignCorners); in resizeNearestNeighbor() 87 inWidth - 1); in resizeNearestNeighbor() 91 std::copy_n(inputData + b * inHeight * inWidth * channels + in resizeNearestNeighbor() 92 inY * inWidth * channels + inX * channels, in resizeNearestNeighbor() 182 uint32_t inWidth = getSizeOfDimension(input, useNchw ? 3 : 2); in prepare() local 185 NN_RET_CHECK_GT(inWidth, 0u); in prepare() 196 width = std::floor(static_cast<float>(inWidth) * in prepare() 203 static_cast<float>(inWidth) * in prepare()
|
D | RoiPooling.cpp | 54 uint32_t inWidth = getSizeOfDimension(inputShape, 2); in roiPoolingNhwc() local 76 NN_RET_CHECK(roiInfo[0] * widthScale <= inWidth); in roiPoolingNhwc() 78 NN_RET_CHECK(roiInfo[2] * widthScale <= inWidth); in roiPoolingNhwc() 94 const T_Input* batchBase = inputData + batchId * inHeight * inWidth * inDepth; in roiPoolingNhwc() 104 wStart = std::min(wStart, inWidth); in roiPoolingNhwc() 105 wEnd = std::min(wEnd, inWidth); in roiPoolingNhwc() 114 T_Input inputValue = batchBase[h * inWidth * inDepth + w * inDepth + k]; in roiPoolingNhwc() 188 [[maybe_unused]] uint32_t inWidth = getSizeOfDimension(input, useNchw ? 3 : 2); in prepare() local
|
/packages/modules/NeuralNetworks/common/include/ |
D | OperationsExecutionUtils.h | 163 inline PaddingScheme getPaddingScheme(int32_t inWidth, int32_t inHeight, int32_t strideWidth, in getPaddingScheme() argument 175 calculateExplicitPadding(inWidth, strideWidth, filterWidth, kPaddingSame, &expectedPaddingLeft, in getPaddingScheme()
|