Lines Matching refs:roiShape
48 const Shape& roiShape, const int32_t* batchSplitData, in roiAlignNhwc() argument
64 uint32_t numRois = getSizeOfDimension(roiShape, 0); in roiAlignNhwc()
65 uint32_t roiInfoLength = getSizeOfDimension(roiShape, 1); in roiAlignNhwc()
173 const uint16_t* roiData, const Shape& roiShape, in roiAlignQuantNhwc() argument
191 uint32_t numRois = getSizeOfDimension(roiShape, 0); in roiAlignQuantNhwc()
192 uint32_t roiInfoLength = getSizeOfDimension(roiShape, 1); in roiAlignQuantNhwc()
306 const Shape& roiShape, const int32_t* batchSplitData, in roiAlign() argument
317 input.getNhwcBuffer(), input.getNhwcShape(), roiData, roiShape, batchSplitData, in roiAlign()
321 NN_RET_CHECK(roiAlignNhwc(input.getNhwcBuffer(), input.getNhwcShape(), roiData, roiShape, in roiAlign()
335 Shape roiShape = context->getInputShape(kRoiTensor); in prepare() local
338 NN_RET_CHECK_EQ(getNumberOfDimensions(roiShape), 2u); in prepare()
344 uint32_t numRois = getSizeOfDimension(roiShape, 0); in prepare()
350 NN_RET_CHECK_EQ(getSizeOfDimension(roiShape, 1), 4u); in prepare()
373 if (roiShape.type == OperandType::TENSOR_QUANT16_ASYMM) { in prepare()
374 NN_RET_CHECK_EQ(roiShape.scale, 0.125f); in prepare()
375 NN_RET_CHECK_EQ(roiShape.offset, 0); in prepare()