Searched refs:inDepth (Results 1 – 3 of 3) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | RoiAlign.cpp | 71 uint32_t inDepth = getSizeOfDimension(inputShape, 3); in roiAlignNhwc() local
119 const T_Input* batchBase = inputData + batchId * inHeight * inWidth * inDepth; in roiAlignNhwc()
128 for (uint32_t k = 0; k < inDepth; k++) outPtr[k] = 0; in roiAlignNhwc()
156 uint32_t offsets[] = {y1 * inWidth * inDepth + x1 * inDepth, in roiAlignNhwc()
157 y1 * inWidth * inDepth + x2 * inDepth, in roiAlignNhwc()
158 y2 * inWidth * inDepth + x1 * inDepth, in roiAlignNhwc()
159 y2 * inWidth * inDepth + x2 * inDepth}; in roiAlignNhwc()
161 for (uint32_t k = 0; k < inDepth; k++) { in roiAlignNhwc()
172 for (uint32_t k = 0; k < inDepth; k++) in roiAlignNhwc()
174 outPtr += inDepth; in roiAlignNhwc()
[all …]
|
| D | RoiPooling.cpp | 67 uint32_t inDepth = getSizeOfDimension(inputShape, 3); in roiPoolingNhwc() local
106 const T_Input* batchBase = inputData + batchId * inHeight * inWidth * inDepth; in roiPoolingNhwc()
121 for (uint32_t k = 0; k < inDepth; k++) { in roiPoolingNhwc()
126 T_Input inputValue = batchBase[h * inWidth * inDepth + w * inDepth + k]; in roiPoolingNhwc()
135 outPtr += inDepth; in roiPoolingNhwc()
240 uint32_t inDepth = getSizeOfDimension(input, useNchw ? 1 : 3); in prepare() local
267 output.dimensions = {numRois, inDepth, static_cast<uint32_t>(outputHeight), in prepare()
271 static_cast<uint32_t>(outputWidth), inDepth}; in prepare()
|
| D | Conv2D.cpp | 139 uint32_t inDepth = getSizeOfDimension(inputShape, 3); \
148 im2colDim.sizes[0] = (int)inDepth * filterHeight * filterWidth; \
|