Searched refs:outputScoreShape (Results 1 – 2 of 2) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | HeatmapMaxKeypoint.cpp | 90 float* outputScoreData, const Shape& outputScoreShape, in heatmapMaxKeypointFloat32Nhwc() argument
169 float* outputScoreData, const Shape& outputScoreShape, in heatmapMaxKeypointFloat32() argument
180 outputScoreData, outputScoreShape, outputKeypointData, in heatmapMaxKeypointFloat32()
186 uint8_t* outputScoreData, const Shape& outputScoreShape, in heatmapMaxKeypointQuant() argument
193 std::vector<float> outputScore_float32(getNumberOfElements(outputScoreShape)); in heatmapMaxKeypointQuant()
197 outputScore_float32.data(), outputScoreShape, outputKeypoint_float32.data(), in heatmapMaxKeypointQuant()
199 convertFloat32ToQuant(outputScore_float32, outputScoreShape.scale, outputScoreShape.offset, in heatmapMaxKeypointQuant()
208 int8_t* outputScoreData, const Shape& outputScoreShape, in heatmapMaxKeypointQuant() argument
215 std::vector<float> outputScore_float32(getNumberOfElements(outputScoreShape)); in heatmapMaxKeypointQuant()
219 outputScore_float32.data(), outputScoreShape, outputKeypoint_float32.data(), in heatmapMaxKeypointQuant()
[all …]
|
| D | GenerateProposals.cpp | 758 Shape outputScoreShape = context->getOutputShape(kOutputScoreTensor); in prepare() local
782 outputScoreShape.type = scoreShape.type; in prepare()
783 outputScoreShape.dimensions = {0}; in prepare()
784 outputScoreShape.scale = scoreShape.scale; in prepare()
785 outputScoreShape.offset = scoreShape.offset; in prepare()
786 NN_RET_CHECK(context->setOutputShape(kOutputScoreTensor, outputScoreShape)); in prepare()
1286 Shape outputScoreShape = context->getOutputShape(kOutputScoreTensor); in prepare() local
1316 outputScoreShape.type = scoreShape.type; in prepare()
1317 outputScoreShape.dimensions = {0}; in prepare()
1318 outputScoreShape.scale = scoreShape.scale; in prepare()
[all …]
|