Lines Matching refs:inDepth
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()
198 uint32_t inDepth = getSizeOfDimension(inputShape, 3); in roiAlignQuantNhwc() local
248 const T_Input* batchBase = inputData + batchId * inHeight * inWidth * inDepth; in roiAlignQuantNhwc()
256 std::vector<int32_t> outTemp(inDepth, 0); in roiAlignQuantNhwc()
282 uint32_t offsets[] = {y1 * inWidth * inDepth + x1 * inDepth, in roiAlignQuantNhwc()
283 y1 * inWidth * inDepth + x2 * inDepth, in roiAlignQuantNhwc()
284 y2 * inWidth * inDepth + x1 * inDepth, in roiAlignQuantNhwc()
285 y2 * inWidth * inDepth + x2 * inDepth}; in roiAlignQuantNhwc()
287 for (uint32_t k = 0; k < inDepth; k++) { in roiAlignQuantNhwc()
301 for (uint32_t k = 0; k < inDepth; k++) { in roiAlignQuantNhwc()
307 outPtr += inDepth; in roiAlignQuantNhwc()
395 uint32_t inDepth = getSizeOfDimension(input, useNchw ? 1 : 3); in prepare() local
401 NN_RET_CHECK_GT(inDepth, 0); in prepare()
433 output.dimensions = {numRois, inDepth, static_cast<uint32_t>(outputHeight), in prepare()
437 static_cast<uint32_t>(outputWidth), inDepth}; in prepare()