Lines Matching refs:kOutputTensor

38 constexpr uint32_t kOutputTensor = 0;  variable
169         const float outputScale = context->getOutputShape(kOutputTensor).scale;  in validate()
220     Shape output = context->getOutputShape(kOutputTensor);  in prepare()
223     return context->setOutputShape(kOutputTensor, output);  in prepare()
228     if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true;  in execute()
238                                          context->getOutputBuffer<float>(kOutputTensor),  in execute()
239                                          context->getOutputShape(kOutputTensor));  in execute()
248                                          context->getOutputBuffer<_Float16>(kOutputTensor),  in execute()
249                                          context->getOutputShape(kOutputTensor));  in execute()
258                                         context->getOutputBuffer<uint8_t>(kOutputTensor),  in execute()
259                                         context->getOutputShape(kOutputTensor));  in execute()