Home
last modified time | relevance | path

Searched refs:kOutputClassTensor (Results 1 – 2 of 2) sorted by relevance

/packages/modules/NeuralNetworks/common/cpu_operations/
DGenerateProposals.cpp536 Shape classesOutShape = context->getOutputShape(kOutputClassTensor); in boxWithNmsLimitWriteOutput()
538 NN_RET_CHECK(context->setOutputShape(kOutputClassTensor, classesOutShape)); in boxWithNmsLimitWriteOutput()
550 int32_t* classesOutPtr = context->getOutputBuffer<int32_t>(kOutputClassTensor); in boxWithNmsLimitWriteOutput()
671 Shape outputClassShape = context->getOutputShape(kOutputClassTensor); in prepare()
711 NN_RET_CHECK(context->setOutputShape(kOutputClassTensor, outputClassShape)); in prepare()
742 context->getOutputBuffer<int32_t>(kOutputClassTensor), in execute()
743 context->getOutputShape(kOutputClassTensor), in execute()
764 context->getOutputBuffer<int32_t>(kOutputClassTensor), in execute()
765 context->getOutputShape(kOutputClassTensor), in execute()
786 context->getOutputBuffer<int32_t>(kOutputClassTensor), in execute()
[all …]
/packages/modules/NeuralNetworks/common/types/operations/include/
DGenerateProposals.h57 constexpr uint32_t kOutputClassTensor = 2; variable
109 constexpr uint32_t kOutputClassTensor = 2; variable