Searched refs:kOutputScoreTensor (Results 1 – 2 of 2) sorted by relevance
44 constexpr uint32_t kOutputScoreTensor = 0; variable283 Shape outputScore = context->getOutputShape(kOutputScoreTensor); in prepare()286 NN_RET_CHECK(context->setOutputShape(kOutputScoreTensor, outputScore)); in prepare()309 auto outputScoreData = context->getOutputBuffer<_Float16>(kOutputScoreTensor); in execute()310 const auto outputScoreShape = context->getOutputShape(kOutputScoreTensor); in execute()332 context->getOutputBuffer<float>(kOutputScoreTensor), in execute()333 context->getOutputShape(kOutputScoreTensor), in execute()344 context->getOutputBuffer<uint8_t>(kOutputScoreTensor), in execute()345 context->getOutputShape(kOutputScoreTensor), in execute()355 context->getOutputBuffer<int8_t>(kOutputScoreTensor), in execute()[all …]
351 constexpr uint32_t kOutputScoreTensor = 0; variable579 Shape scoresOutShape = context->getOutputShape(kOutputScoreTensor); in boxWithNmsLimitWriteOutput()763 Shape outputScoreShape = context->getOutputShape(kOutputScoreTensor); in prepare()833 context->getOutputShape(kOutputScoreTensor), in execute()933 constexpr uint32_t kOutputScoreTensor = 0; variable1097 Shape scoresOutShape = context->getOutputShape(kOutputScoreTensor); in generateProposalsFloat32()1149 Shape scoresOutShape = context->getOutputShape(kOutputScoreTensor); in generateProposalsFloat16()1201 Shape scoresOutShape = context->getOutputShape(kOutputScoreTensor); in generateProposalsQuant()1443 constexpr uint32_t kOutputScoreTensor = 0; variable1715 context->getOutputShape(kOutputScoreTensor), in execute()[all …]