Searched refs:outputScoreShape (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | HeatmapMaxKeypoint.cpp | 161 float* outputScoreData, const Shape& outputScoreShape, in heatmapMaxKeypointFloat32() argument
172 outputScoreData, outputScoreShape, outputKeypointData, in heatmapMaxKeypointFloat32()
178 uint8_t* outputScoreData, const Shape& outputScoreShape, in heatmapMaxKeypointQuant() argument
185 std::vector<float> outputScore_float32(getNumberOfElements(outputScoreShape)); in heatmapMaxKeypointQuant()
189 outputScore_float32.data(), outputScoreShape, outputKeypoint_float32.data(), in heatmapMaxKeypointQuant()
191 convertFloat32ToQuant(outputScore_float32, outputScoreShape.scale, outputScoreShape.offset, in heatmapMaxKeypointQuant()
200 int8_t* outputScoreData, const Shape& outputScoreShape, in heatmapMaxKeypointQuant() argument
207 std::vector<float> outputScore_float32(getNumberOfElements(outputScoreShape)); in heatmapMaxKeypointQuant()
211 outputScore_float32.data(), outputScoreShape, outputKeypoint_float32.data(), in heatmapMaxKeypointQuant()
213 convertFloat32ToQuant(outputScore_float32, outputScoreShape.scale, outputScoreShape.offset, in heatmapMaxKeypointQuant()
[all …]
|
| D | GenerateProposals.cpp | 670 Shape outputScoreShape = context->getOutputShape(kOutputScoreTensor); in prepare() local
694 outputScoreShape.type = scoreShape.type; in prepare()
695 outputScoreShape.dimensions = {0}; in prepare()
696 outputScoreShape.scale = scoreShape.scale; in prepare()
697 outputScoreShape.offset = scoreShape.offset; in prepare()
698 NN_RET_CHECK(context->setOutputShape(kOutputScoreTensor, outputScoreShape)); in prepare()
1119 Shape outputScoreShape = context->getOutputShape(kOutputScoreTensor); in prepare() local
1149 outputScoreShape.type = scoreShape.type; in prepare()
1150 outputScoreShape.dimensions = {0}; in prepare()
1151 outputScoreShape.scale = scoreShape.scale; in prepare()
[all …]
|