Searched refs:heatmapShape (Results 1 – 1 of 1) sorted by relevance
| /packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | HeatmapMaxKeypoint.cpp | 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()
[all …]
|