Lines Matching refs:roiShape
43 const Shape& roiShape, const int32_t* batchSplitData, in roiPoolingNhwc() argument
58 uint32_t numRois = getSizeOfDimension(roiShape, 0); in roiPoolingNhwc()
59 uint32_t roiInfoLength = getSizeOfDimension(roiShape, 1); in roiPoolingNhwc()
132 const Shape& roiShape, const int32_t* batchSplitData, in roiPooling() argument
139 NN_RET_CHECK(roiPoolingNhwc(input.getNhwcBuffer(), input.getNhwcShape(), roiData, roiShape, in roiPooling()
148 const uint16_t* roiData, const Shape& roiShape, in roiPooling() argument
153 std::vector<float> roi_float32(getNumberOfElements(roiShape)); in roiPooling()
154 convertQuantToFloat32(roiData, roiShape.scale, roiShape.offset, &roi_float32); in roiPooling()
155 NN_RET_CHECK(roiPooling(inputData, inputShape, roi_float32.data(), roiShape, batchSplitData, in roiPooling()
163 const uint16_t* roiData, const Shape& roiShape, in roiPooling() argument
168 std::vector<float> roi_float32(getNumberOfElements(roiShape)); in roiPooling()
169 convertQuantToFloat32(roiData, roiShape.scale, roiShape.offset, &roi_float32); in roiPooling()
170 NN_RET_CHECK(roiPooling(inputData, inputShape, roi_float32.data(), roiShape, batchSplitData, in roiPooling()
181 Shape roiShape = context->getInputShape(kRoiTensor); in prepare() local
184 NN_RET_CHECK_EQ(getNumberOfDimensions(roiShape), 2u); in prepare()
190 uint32_t numRois = getSizeOfDimension(roiShape, 0); in prepare()
191 NN_RET_CHECK_EQ(getSizeOfDimension(roiShape, 1), 4u); in prepare()
209 if (roiShape.type == OperandType::TENSOR_QUANT16_ASYMM) { in prepare()
210 NN_RET_CHECK_EQ(roiShape.scale, 0.125f); in prepare()
211 NN_RET_CHECK_EQ(roiShape.offset, 0); in prepare()