Searched refs:kOutputClassTensor (Results 1 – 2 of 2) sorted by relevance
537 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 …]
57 constexpr uint32_t kOutputClassTensor = 2; variable109 constexpr uint32_t kOutputClassTensor = 2; variable