Activation.cpp - OpenGrok cross reference for /packages/modules/NeuralNetworks/common/operations/Activation.cpp

Lines Matching refs:kOutputTensor
47 constexpr uint32_t kOutputTensor = 0;  variable
420                 auto outputShape = context->getOutputShape(kOutputTensor);  in prepare()
440     return context->setOutputShape(kOutputTensor, output);  in prepare()
445     if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true;  in executeRelu()
450                              context->getOutputBuffer<_Float16>(kOutputTensor),  in executeRelu()
451                              context->getOutputShape(kOutputTensor));  in executeRelu()
455                              context->getOutputBuffer<float>(kOutputTensor),  in executeRelu()
456                              context->getOutputShape(kOutputTensor));  in executeRelu()
460                               context->getOutputBuffer<uint8_t>(kOutputTensor),  in executeRelu()
461                               context->getOutputShape(kOutputTensor));  in executeRelu()
465                                     context->getOutputBuffer<int8_t>(kOutputTensor),  in executeRelu()
466                                     context->getOutputShape(kOutputTensor));  in executeRelu()
474     if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true;  in executeRelu1()
479                               context->getOutputBuffer<_Float16>(kOutputTensor),  in executeRelu1()
480                               context->getOutputShape(kOutputTensor));  in executeRelu1()
484                               context->getOutputBuffer<float>(kOutputTensor),  in executeRelu1()
485                               context->getOutputShape(kOutputTensor));  in executeRelu1()
489                                context->getOutputBuffer<uint8_t>(kOutputTensor),  in executeRelu1()
490                                context->getOutputShape(kOutputTensor));  in executeRelu1()
494                                      context->getOutputBuffer<int8_t>(kOutputTensor),  in executeRelu1()
495                                      context->getOutputShape(kOutputTensor));  in executeRelu1()
503     if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true;  in executeRelu6()
508                               context->getOutputBuffer<_Float16>(kOutputTensor),  in executeRelu6()
509                               context->getOutputShape(kOutputTensor));  in executeRelu6()
513                               context->getOutputBuffer<float>(kOutputTensor),  in executeRelu6()
514                               context->getOutputShape(kOutputTensor));  in executeRelu6()
518                                context->getOutputBuffer<uint8_t>(kOutputTensor),  in executeRelu6()
519                                context->getOutputShape(kOutputTensor));  in executeRelu6()
523                                      context->getOutputBuffer<int8_t>(kOutputTensor),  in executeRelu6()
524                                      context->getOutputShape(kOutputTensor));  in executeRelu6()
532     if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true;  in executeLogistic()
537                                  context->getOutputBuffer<_Float16>(kOutputTensor),  in executeLogistic()
538                                  context->getOutputShape(kOutputTensor));  in executeLogistic()
542                                  context->getOutputBuffer<float>(kOutputTensor),  in executeLogistic()
543                                  context->getOutputShape(kOutputTensor));  in executeLogistic()
547                                   context->getOutputBuffer<uint8_t>(kOutputTensor),  in executeLogistic()
548                                   context->getOutputShape(kOutputTensor));  in executeLogistic()
552                                         context->getOutputBuffer<int8_t>(kOutputTensor),  in executeLogistic()
553                                         context->getOutputShape(kOutputTensor));  in executeLogistic()
561     if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true;  in executeTanh()
566                                context->getOutputBuffer<_Float16>(kOutputTensor),  in executeTanh()
567                                context->getOutputShape(kOutputTensor));  in executeTanh()
571                                context->getOutputBuffer<float>(kOutputTensor),  in executeTanh()
572                                context->getOutputShape(kOutputTensor));  in executeTanh()
576                               context->getOutputBuffer<uint8_t>(kOutputTensor),  in executeTanh()
577                               context->getOutputShape(kOutputTensor));  in executeTanh()
581                                     context->getOutputBuffer<int8_t>(kOutputTensor),  in executeTanh()
582                                     context->getOutputShape(kOutputTensor));  in executeTanh()
590     if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true;  in executeHardSwish()
594             const Shape& outputShape = context->getOutputShape(kOutputTensor);  in executeHardSwish()
601             convertFloat32ToFloat16(outputFloat, context->getOutputBuffer<_Float16>(kOutputTensor));  in executeHardSwish()
608                     convertShapeToTflshape(context->getOutputShape(kOutputTensor)),  in executeHardSwish()
609                     context->getOutputBuffer<float>(kOutputTensor));  in executeHardSwish()
615                                   context->getOutputBuffer<uint8_t>(kOutputTensor),  in executeHardSwish()
616                                   context->getOutputShape(kOutputTensor));  in executeHardSwish()
620                                   context->getOutputBuffer<int8_t>(kOutputTensor),  in executeHardSwish()
621                                   context->getOutputShape(kOutputTensor));  in executeHardSwish()