Home
last modified time | relevance | path

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

/aosp_15_r20/packages/modules/NeuralNetworks/common/cpu_operations/
DGenerateProposals.cpp537 Shape classesOutShape = context->getOutputShape(kOutputClassTensor); in boxWithNmsLimitWriteOutput()
539 NN_RET_CHECK(context->setOutputShape(kOutputClassTensor, classesOutShape)); in boxWithNmsLimitWriteOutput()
551 int32_t* classesOutPtr = context->getOutputBuffer<int32_t>(kOutputClassTensor); in boxWithNmsLimitWriteOutput()
672 Shape outputClassShape = context->getOutputShape(kOutputClassTensor); in prepare()
712 NN_RET_CHECK(context->setOutputShape(kOutputClassTensor, outputClassShape)); in prepare()
743 context->getOutputBuffer<int32_t>(kOutputClassTensor), in execute()
744 context->getOutputShape(kOutputClassTensor), in execute()
765 context->getOutputBuffer<int32_t>(kOutputClassTensor), in execute()
766 context->getOutputShape(kOutputClassTensor), in execute()
787 context->getOutputBuffer<int32_t>(kOutputClassTensor), in execute()
[all …]
/aosp_15_r20/packages/modules/NeuralNetworks/common/types/operations/include/
DGenerateProposals.h57 constexpr uint32_t kOutputClassTensor = 2; variable
109 constexpr uint32_t kOutputClassTensor = 2; variable