80 inline bool heatmapMaxKeypointFloat32Nhwc(const float* heatmap, const Shape& heatmapShape,  in heatmapMaxKeypointFloat32Nhwc()  argument
88     uint32_t numBoxes = getSizeOfDimension(heatmapShape, 0);  in heatmapMaxKeypointFloat32Nhwc()
89     uint32_t heatmapSize = getSizeOfDimension(heatmapShape, 1);  in heatmapMaxKeypointFloat32Nhwc()
90     uint32_t numKeypoints = getSizeOfDimension(heatmapShape, 3);  in heatmapMaxKeypointFloat32Nhwc()
159 inline bool heatmapMaxKeypointFloat32(const float* heatmap, const Shape& heatmapShape,  in heatmapMaxKeypointFloat32()  argument
167         NN_RET_CHECK(convertNchwToNhwc(heatmap, heatmapShape, &heatmap_nhwc, &heatmapShape_nhwc));  in heatmapMaxKeypointFloat32()
170     const Shape& heatmapShape_tmp = layout ? heatmapShape_nhwc : heatmapShape;  in heatmapMaxKeypointFloat32()
176 inline bool heatmapMaxKeypointQuant(const uint8_t* heatmap, const Shape& heatmapShape,  in heatmapMaxKeypointQuant()  argument
181     std::vector<float> heatmap_float32(getNumberOfElements(heatmapShape));  in heatmapMaxKeypointQuant()
182     convertQuantToFloat32(heatmap, heatmapShape.scale, heatmapShape.offset, &heatmap_float32);  in heatmapMaxKeypointQuant()
188             heatmap_float32.data(), heatmapShape, boxes_float32.data(), boxesShape, layout,  in heatmapMaxKeypointQuant()
198 inline bool heatmapMaxKeypointQuant(const int8_t* heatmap, const Shape& heatmapShape,  in heatmapMaxKeypointQuant()  argument
203     std::vector<float> heatmap_float32(getNumberOfElements(heatmapShape));  in heatmapMaxKeypointQuant()
204     convertQuantToFloat32(heatmap, heatmapShape.scale, heatmapShape.offset, &heatmap_float32);  in heatmapMaxKeypointQuant()
210             heatmap_float32.data(), heatmapShape, boxes_float32.data(), boxesShape, layout,  in heatmapMaxKeypointQuant()
224     Shape heatmapShape = context->getInputShape(kHeatmapTensor);  in prepare()  local
226     NN_RET_CHECK_EQ(getNumberOfDimensions(heatmapShape), 4u);  in prepare()
229     uint32_t numBoxes = getSizeOfDimension(heatmapShape, 0);  in prepare()
230     uint32_t heatmapSize = getSizeOfDimension(heatmapShape, 2);  in prepare()
231     uint32_t numKeypoints = getSizeOfDimension(heatmapShape, layout ? 1 : 3);  in prepare()
233     NN_RET_CHECK_EQ(getSizeOfDimension(heatmapShape, layout ? 3 : 1), heatmapSize);  in prepare()
238     if (heatmapShape.type == OperandType::TENSOR_QUANT8_ASYMM ||  in prepare()
239         heatmapShape.type == OperandType::TENSOR_QUANT8_ASYMM_SIGNED) {  in prepare()
245     outputScore.type = heatmapShape.type;  in prepare()
254     if (heatmapShape.type == OperandType::TENSOR_QUANT8_ASYMM ||  in prepare()
255         heatmapShape.type == OperandType::TENSOR_QUANT8_ASYMM_SIGNED) {  in prepare()
267             const auto heatmapShape = context->getInputShape(kHeatmapTensor);  in execute()  local
274             std::vector<float> heatmap_float32(getNumberOfElements(heatmapShape));  in execute()
281                     heatmap_float32.data(), heatmapShape, boxes_float32.data(), boxesShape, layout,  in execute()