Lines Matching refs:kOutputTensor

34 constexpr uint32_t kOutputTensor = 0;  variable
109     Shape outputShape = context->getOutputShape(kOutputTensor);  in prepare()
122     return context->setOutputShape(kOutputTensor, outputShape);  in prepare()
127     if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true;  in execute()
136                                context->getOutputBuffer<_Float16>(kOutputTensor),  in execute()
137                                context->getOutputShape(kOutputTensor));  in execute()
145                                context->getOutputBuffer<float>(kOutputTensor),  in execute()
146                                context->getOutputShape(kOutputTensor));  in execute()
154                                context->getOutputBuffer<int32_t>(kOutputTensor),  in execute()
155                                context->getOutputShape(kOutputTensor));  in execute()
163                                context->getOutputBuffer<uint8_t>(kOutputTensor),  in execute()
164                                context->getOutputShape(kOutputTensor));  in execute()