Home
last modified time | relevance | path

Searched refs:kOutputScoreTensor (Results 1 – 4 of 4) sorted by relevance

/packages/modules/NeuralNetworks/common/cpu_operations/
DHeatmapMaxKeypoint.cpp244 Shape outputScore = context->getOutputShape(kOutputScoreTensor); in prepare()
247 NN_RET_CHECK(context->setOutputShape(kOutputScoreTensor, outputScore)); in prepare()
270 auto outputScoreData = context->getOutputBuffer<_Float16>(kOutputScoreTensor); in execute()
271 const auto outputScoreShape = context->getOutputShape(kOutputScoreTensor); in execute()
293 context->getOutputBuffer<float>(kOutputScoreTensor), in execute()
294 context->getOutputShape(kOutputScoreTensor), in execute()
305 context->getOutputBuffer<uint8_t>(kOutputScoreTensor), in execute()
306 context->getOutputShape(kOutputScoreTensor), in execute()
316 context->getOutputBuffer<int8_t>(kOutputScoreTensor), in execute()
317 context->getOutputShape(kOutputScoreTensor), in execute()
DGenerateProposals.cpp529 Shape scoresOutShape = context->getOutputShape(kOutputScoreTensor); in boxWithNmsLimitWriteOutput()
531 NN_RET_CHECK(context->setOutputShape(kOutputScoreTensor, scoresOutShape)); in boxWithNmsLimitWriteOutput()
549 T_Score* scoresOutPtr = context->getOutputBuffer<T_Score>(kOutputScoreTensor); in boxWithNmsLimitWriteOutput()
670 Shape outputScoreShape = context->getOutputShape(kOutputScoreTensor); in prepare()
698 NN_RET_CHECK(context->setOutputShape(kOutputScoreTensor, outputScoreShape)); in prepare()
739 context->getOutputBuffer<_Float16>(kOutputScoreTensor), in execute()
740 context->getOutputShape(kOutputScoreTensor), in execute()
761 context->getOutputBuffer<float>(kOutputScoreTensor), in execute()
762 context->getOutputShape(kOutputScoreTensor), in execute()
783 context->getOutputBuffer<uint8_t>(kOutputScoreTensor), in execute()
[all …]
/packages/modules/NeuralNetworks/common/types/operations/include/
DGenerateProposals.h55 constexpr uint32_t kOutputScoreTensor = 0; variable
80 constexpr uint32_t kOutputScoreTensor = 0; variable
107 constexpr uint32_t kOutputScoreTensor = 0; variable
DHeatmapMaxKeypoint.h32 constexpr uint32_t kOutputScoreTensor = 0; variable