Home
last modified time | relevance | path

Searched defs:outputScoreShape (Results 1 – 2 of 2) sorted by relevance

/packages/modules/NeuralNetworks/common/cpu_operations/
DHeatmapMaxKeypoint.cpp161 float* outputScoreData, const Shape& outputScoreShape, in heatmapMaxKeypointFloat32()
178 uint8_t* outputScoreData, const Shape& outputScoreShape, in heatmapMaxKeypointQuant()
200 int8_t* outputScoreData, const Shape& outputScoreShape, in heatmapMaxKeypointQuant()
271 const auto outputScoreShape = context->getOutputShape(kOutputScoreTensor); in execute() local
DGenerateProposals.cpp670 Shape outputScoreShape = context->getOutputShape(kOutputScoreTensor); in prepare() local
1119 Shape outputScoreShape = context->getOutputShape(kOutputScoreTensor); in prepare() local
1393 Shape outputScoreShape = context->getOutputShape(kOutputScoreTensor); in prepare() local