Lines Matching refs:kOutputTensor
38 constexpr uint32_t kOutputTensor = 0; variable
121 Shape outputShape = context->getOutputShape(kOutputTensor); in prepare()
134 return context->setOutputShape(kOutputTensor, outputShape); in prepare()
139 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in execute()
148 context->getOutputBuffer<_Float16>(kOutputTensor), in execute()
149 context->getOutputShape(kOutputTensor)); in execute()
157 context->getOutputBuffer<float>(kOutputTensor), in execute()
158 context->getOutputShape(kOutputTensor)); in execute()
166 context->getOutputBuffer<int32_t>(kOutputTensor), in execute()
167 context->getOutputShape(kOutputTensor)); in execute()
175 context->getOutputBuffer<uint8_t>(kOutputTensor), in execute()
176 context->getOutputShape(kOutputTensor)); in execute()
184 context->getOutputBuffer<int8_t>(kOutputTensor), in execute()
185 context->getOutputShape(kOutputTensor)); in execute()