Lines Matching refs:kOutputTensor
41 constexpr uint32_t kOutputTensor = 0; variable
101 NN_RET_CHECK(!context->isOmittedOutput(kOutputTensor)); in prepare()
105 Shape output = context->getOutputShape(kOutputTensor); in prepare()
133 return context->setOutputShape(kOutputTensor, output); in prepare()
138 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in execute()
146 context->getOutputBuffer<float>(kOutputTensor), in execute()
147 context->getOutputShape(kOutputTensor)); in execute()
153 context->getOutputBuffer<_Float16>(kOutputTensor), in execute()
154 context->getOutputShape(kOutputTensor)); in execute()
160 context->getOutputBuffer<uint8_t>(kOutputTensor), in execute()
161 context->getOutputShape(kOutputTensor)); in execute()
167 context->getOutputBuffer<int8_t>(kOutputTensor), in execute()
168 context->getOutputShape(kOutputTensor)); in execute()