Lines Matching refs:kOutputTensor

36 constexpr uint32_t kOutputTensor = 0;  variable
88     NN_RET_CHECK(!context->isOmittedOutput(kOutputTensor));  in prepare()
92     Shape output = context->getOutputShape(kOutputTensor);  in prepare()
120     return context->setOutputShape(kOutputTensor, output);  in prepare()
125     if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true;  in execute()
133                                     context->getOutputBuffer<float>(kOutputTensor),  in execute()
134                                     context->getOutputShape(kOutputTensor));  in execute()
140                                     context->getOutputBuffer<_Float16>(kOutputTensor),  in execute()
141                                     context->getOutputShape(kOutputTensor));  in execute()
147                                     context->getOutputBuffer<uint8_t>(kOutputTensor),  in execute()
148                                     context->getOutputShape(kOutputTensor));  in execute()