Lines Matching refs:kOutputTensor
42 constexpr uint32_t kOutputTensor = 0; variable
105 NN_RET_CHECK(!context->isOmittedOutput(kOutputTensor)); in prepare()
109 Shape output = context->getOutputShape(kOutputTensor); in prepare()
137 return context->setOutputShape(kOutputTensor, output); in prepare()
142 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in execute()
150 context->getOutputBuffer<float>(kOutputTensor), in execute()
151 context->getOutputShape(kOutputTensor)); in execute()
157 context->getOutputBuffer<_Float16>(kOutputTensor), in execute()
158 context->getOutputShape(kOutputTensor)); in execute()
164 context->getOutputBuffer<uint8_t>(kOutputTensor), in execute()
165 context->getOutputShape(kOutputTensor)); in execute()
171 context->getOutputBuffer<int8_t>(kOutputTensor), in execute()
172 context->getOutputShape(kOutputTensor)); in execute()