Lines Matching refs:kOutputClassTensor

536     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()
787                                         context->getOutputShape(kOutputClassTensor),  in execute()
808                                         context->getOutputBuffer<int32_t>(kOutputClassTensor),  in execute()
809                                         context->getOutputShape(kOutputClassTensor),  in execute()
1394     Shape outputClassShape = context->getOutputShape(kOutputClassTensor);  in prepare()
1453     NN_RET_CHECK(context->setOutputShape(kOutputClassTensor, outputClassShape));  in prepare()
1487                     context->getOutputBuffer<int32_t>(kOutputClassTensor),  in execute()
1488                     context->getOutputShape(kOutputClassTensor),  in execute()
1515                     context->getOutputBuffer<int32_t>(kOutputClassTensor),  in execute()
1516                     context->getOutputShape(kOutputClassTensor),  in execute()