Activation.cpp - OpenGrok cross reference for /frameworks/ml/nn/common/operations/Activation.cpp

Lines Matching refs:kOutputTensor
47 constexpr uint32_t kOutputTensor = 0;  variable
410                 auto outputShape = context->getOutputShape(kOutputTensor);  in prepare()
430     return context->setOutputShape(kOutputTensor, output);  in prepare()
435     if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true;  in executeRelu()
440                              context->getOutputBuffer<_Float16>(kOutputTensor),  in executeRelu()
441                              context->getOutputShape(kOutputTensor));  in executeRelu()
445                              context->getOutputBuffer<float>(kOutputTensor),  in executeRelu()
446                              context->getOutputShape(kOutputTensor));  in executeRelu()
450                               context->getOutputBuffer<uint8_t>(kOutputTensor),  in executeRelu()
451                               context->getOutputShape(kOutputTensor));  in executeRelu()
455                                     context->getOutputBuffer<int8_t>(kOutputTensor),  in executeRelu()
456                                     context->getOutputShape(kOutputTensor));  in executeRelu()
464     if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true;  in executeRelu1()
469                               context->getOutputBuffer<_Float16>(kOutputTensor),  in executeRelu1()
470                               context->getOutputShape(kOutputTensor));  in executeRelu1()
474                               context->getOutputBuffer<float>(kOutputTensor),  in executeRelu1()
475                               context->getOutputShape(kOutputTensor));  in executeRelu1()
479                                context->getOutputBuffer<uint8_t>(kOutputTensor),  in executeRelu1()
480                                context->getOutputShape(kOutputTensor));  in executeRelu1()
484                                      context->getOutputBuffer<int8_t>(kOutputTensor),  in executeRelu1()
485                                      context->getOutputShape(kOutputTensor));  in executeRelu1()
493     if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true;  in executeRelu6()
498                               context->getOutputBuffer<_Float16>(kOutputTensor),  in executeRelu6()
499                               context->getOutputShape(kOutputTensor));  in executeRelu6()
503                               context->getOutputBuffer<float>(kOutputTensor),  in executeRelu6()
504                               context->getOutputShape(kOutputTensor));  in executeRelu6()
508                                context->getOutputBuffer<uint8_t>(kOutputTensor),  in executeRelu6()
509                                context->getOutputShape(kOutputTensor));  in executeRelu6()
513                                      context->getOutputBuffer<int8_t>(kOutputTensor),  in executeRelu6()
514                                      context->getOutputShape(kOutputTensor));  in executeRelu6()
522     if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true;  in executeLogistic()
527                                  context->getOutputBuffer<_Float16>(kOutputTensor),  in executeLogistic()
528                                  context->getOutputShape(kOutputTensor));  in executeLogistic()
532                                  context->getOutputBuffer<float>(kOutputTensor),  in executeLogistic()
533                                  context->getOutputShape(kOutputTensor));  in executeLogistic()
537                                   context->getOutputBuffer<uint8_t>(kOutputTensor),  in executeLogistic()
538                                   context->getOutputShape(kOutputTensor));  in executeLogistic()
542                                         context->getOutputBuffer<int8_t>(kOutputTensor),  in executeLogistic()
543                                         context->getOutputShape(kOutputTensor));  in executeLogistic()
551     if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true;  in executeTanh()
556                                context->getOutputBuffer<_Float16>(kOutputTensor),  in executeTanh()
557                                context->getOutputShape(kOutputTensor));  in executeTanh()
561                                context->getOutputBuffer<float>(kOutputTensor),  in executeTanh()
562                                context->getOutputShape(kOutputTensor));  in executeTanh()
566                               context->getOutputBuffer<uint8_t>(kOutputTensor),  in executeTanh()
567                               context->getOutputShape(kOutputTensor));  in executeTanh()
571                                     context->getOutputBuffer<int8_t>(kOutputTensor),  in executeTanh()
572                                     context->getOutputShape(kOutputTensor));  in executeTanh()
580     if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true;  in executeHardSwish()
584             const Shape& outputShape = context->getOutputShape(kOutputTensor);  in executeHardSwish()
591             convertFloat32ToFloat16(outputFloat, context->getOutputBuffer<_Float16>(kOutputTensor));  in executeHardSwish()
598                     convertShapeToTflshape(context->getOutputShape(kOutputTensor)),  in executeHardSwish()
599                     context->getOutputBuffer<float>(kOutputTensor));  in executeHardSwish()
605                                   context->getOutputBuffer<uint8_t>(kOutputTensor),  in executeHardSwish()
606                                   context->getOutputShape(kOutputTensor));  in executeHardSwish()
610                                   context->getOutputBuffer<int8_t>(kOutputTensor),  in executeHardSwish()
611                                   context->getOutputShape(kOutputTensor));  in executeHardSwish()