Lines Matching refs:inDepth
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()
188 uint32_t inDepth = getSizeOfDimension(inputShape, 3); in roiAlignQuantNhwc() local
238 const T_Input* batchBase = inputData + batchId * inHeight * inWidth * inDepth; in roiAlignQuantNhwc()
246 std::vector<int32_t> outTemp(inDepth, 0); in roiAlignQuantNhwc()
272 uint32_t offsets[] = {y1 * inWidth * inDepth + x1 * inDepth, in roiAlignQuantNhwc()
273 y1 * inWidth * inDepth + x2 * inDepth, in roiAlignQuantNhwc()
274 y2 * inWidth * inDepth + x1 * inDepth, in roiAlignQuantNhwc()
275 y2 * inWidth * inDepth + x2 * inDepth}; in roiAlignQuantNhwc()
277 for (uint32_t k = 0; k < inDepth; k++) { in roiAlignQuantNhwc()
291 for (uint32_t k = 0; k < inDepth; k++) { in roiAlignQuantNhwc()
297 outPtr += inDepth; in roiAlignQuantNhwc()
343 uint32_t inDepth = getSizeOfDimension(input, useNchw ? 1 : 3); in prepare() local
349 NN_RET_CHECK_GT(inDepth, 0u); in prepare()
381 output.dimensions = {numRois, inDepth, static_cast<uint32_t>(outputHeight), in prepare()
385 static_cast<uint32_t>(outputWidth), inDepth}; in prepare()