Searched refs:inDepth (Results 1 – 3 of 3) sorted by relevance
| /packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | RoiAlign.cpp | 61 uint32_t inDepth = getSizeOfDimension(inputShape, 3); in roiAlignNhwc() local
109 const T_Input* batchBase = inputData + batchId * inHeight * inWidth * inDepth; in roiAlignNhwc()
118 for (uint32_t k = 0; k < inDepth; k++) outPtr[k] = 0; 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()
151 for (uint32_t k = 0; k < inDepth; k++) { in roiAlignNhwc()
162 for (uint32_t k = 0; k < inDepth; k++) in roiAlignNhwc()
164 outPtr += inDepth; in roiAlignNhwc()
[all …]
|
| D | RoiPooling.cpp | 55 uint32_t inDepth = getSizeOfDimension(inputShape, 3); in roiPoolingNhwc() local
94 const T_Input* batchBase = inputData + batchId * inHeight * inWidth * inDepth; in roiPoolingNhwc()
109 for (uint32_t k = 0; k < inDepth; k++) { in roiPoolingNhwc()
114 T_Input inputValue = batchBase[h * inWidth * inDepth + w * inDepth + k]; in roiPoolingNhwc()
123 outPtr += inDepth; in roiPoolingNhwc()
189 uint32_t inDepth = getSizeOfDimension(input, useNchw ? 1 : 3); in prepare() local
216 output.dimensions = {numRois, inDepth, static_cast<uint32_t>(outputHeight), in prepare()
220 static_cast<uint32_t>(outputWidth), inDepth}; in prepare()
|
| D | Conv2D.cpp | 137 uint32_t inDepth = getSizeOfDimension(inputShape, 3); \
146 im2colDim.sizes[0] = (int)inDepth * filterHeight * filterWidth; \
|