Lines Matching refs:kOutputTensor

93     Shape outputShape = context->getOutputShape(kOutputTensor);  in prepare()
107     return context->setOutputShape(kOutputTensor, outputShape);  in prepare()
112     if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true;  in execute()
121                                context->getOutputBuffer<_Float16>(kOutputTensor),  in execute()
122                                context->getOutputShape(kOutputTensor));  in execute()
130                                context->getOutputBuffer<float>(kOutputTensor),  in execute()
131                                context->getOutputShape(kOutputTensor));  in execute()
139                                context->getOutputBuffer<int32_t>(kOutputTensor),  in execute()
140                                context->getOutputShape(kOutputTensor));  in execute()
148                                context->getOutputBuffer<uint8_t>(kOutputTensor),  in execute()
149                                context->getOutputShape(kOutputTensor));  in execute()
157                                context->getOutputBuffer<int8_t>(kOutputTensor),  in execute()
158                                context->getOutputShape(kOutputTensor));  in execute()