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.cpp528 Shape scoresOutShape = context->getOutputShape(kOutputScoreTensor); in boxWithNmsLimitWriteOutput()
530 NN_RET_CHECK(context->setOutputShape(kOutputScoreTensor, scoresOutShape)); in boxWithNmsLimitWriteOutput()
548 T_Score* scoresOutPtr = context->getOutputBuffer<T_Score>(kOutputScoreTensor); in boxWithNmsLimitWriteOutput()
669 Shape outputScoreShape = context->getOutputShape(kOutputScoreTensor); in prepare()
697 NN_RET_CHECK(context->setOutputShape(kOutputScoreTensor, outputScoreShape)); in prepare()
738 context->getOutputBuffer<_Float16>(kOutputScoreTensor), in execute()
739 context->getOutputShape(kOutputScoreTensor), in execute()
760 context->getOutputBuffer<float>(kOutputScoreTensor), in execute()
761 context->getOutputShape(kOutputScoreTensor), in execute()
782 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