Lines Matching refs:kOutputTensor
40 constexpr uint32_t kOutputTensor = 0; variable
126 Shape outputShape = context->getOutputShape(kOutputTensor); in prepare()
139 return context->setOutputShape(kOutputTensor, outputShape); in prepare()
144 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in execute()
153 context->getOutputBuffer<_Float16>(kOutputTensor), in execute()
154 context->getOutputShape(kOutputTensor)); in execute()
162 context->getOutputBuffer<float>(kOutputTensor), in execute()
163 context->getOutputShape(kOutputTensor)); in execute()
171 context->getOutputBuffer<int32_t>(kOutputTensor), in execute()
172 context->getOutputShape(kOutputTensor)); in execute()
180 context->getOutputBuffer<uint8_t>(kOutputTensor), in execute()
181 context->getOutputShape(kOutputTensor)); in execute()
189 context->getOutputBuffer<int8_t>(kOutputTensor), in execute()
190 context->getOutputShape(kOutputTensor)); in execute()