Searched refs:heatmapShape (Results 1 – 1 of 1) sorted by relevance
| /aosp12/packages/modules/NeuralNetworks/common/operations/ |
| H A D | HeatmapMaxKeypoint.cpp | 97 uint32_t numBoxes = getSizeOfDimension(heatmapShape, 0); in heatmapMaxKeypointFloat32Nhwc()
98 uint32_t heatmapSize = getSizeOfDimension(heatmapShape, 1); in heatmapMaxKeypointFloat32Nhwc()
99 uint32_t numKeypoints = getSizeOfDimension(heatmapShape, 3); in heatmapMaxKeypointFloat32Nhwc()
191 convertQuantToFloat32(heatmap, heatmapShape.scale, heatmapShape.offset, &heatmap_float32); in heatmapMaxKeypointQuant()
213 convertQuantToFloat32(heatmap, heatmapShape.scale, heatmapShape.offset, &heatmap_float32); in heatmapMaxKeypointQuant()
263 Shape heatmapShape = context->getInputShape(kHeatmapTensor); in prepare() local
265 NN_RET_CHECK_EQ(getNumberOfDimensions(heatmapShape), 4); in prepare()
268 uint32_t numBoxes = getSizeOfDimension(heatmapShape, 0); in prepare()
269 uint32_t heatmapSize = getSizeOfDimension(heatmapShape, 2); in prepare()
277 if (heatmapShape.type == OperandType::TENSOR_QUANT8_ASYMM || in prepare()
[all …]
|