ONE - On-device Neural Engine
All Data Structures Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Modules Pages
onert::backend::acl_common::AclActivationBuilder< T_Tensor, T_ActivationLayer, T_ExecFunction > Class Template Reference

#include <AclActivationBuilder.h>

Static Public Member Functions

static std::unique_ptr< exec::IFunctiongenerate (ir::Activation code, T_Tensor *ifm_alloc)
 

Detailed Description

template<typename T_Tensor, typename T_ActivationLayer, typename T_ExecFunction>
class onert::backend::acl_common::AclActivationBuilder< T_Tensor, T_ActivationLayer, T_ExecFunction >

Definition at line 32 of file AclActivationBuilder.h.

Member Function Documentation

◆ generate()

template<typename T_Tensor , typename T_ActivationLayer , typename T_ExecFunction >
std::unique_ptr< exec::IFunction > onert::backend::acl_common::AclActivationBuilder< T_Tensor, T_ActivationLayer, T_ExecFunction >::generate ( ir::Activation  code,
T_Tensor *  ifm_alloc 
)
static

Definition at line 89 of file AclActivationBuilder.h.

91{
92 switch (code)
93 {
95 {
96 return std::make_unique<exec::NopFunction>();
97 }
99 {
100 return generateReLU(ifm_alloc);
101 }
103 {
104 return generateReLU1(ifm_alloc);
105 }
107 {
108 return generateReLU6(ifm_alloc);
109 }
110 default:
111 {
112 throw std::runtime_error("Not supported, yet");
113 }
114 }
115}

References onert::ir::NONE, onert::ir::RELU, onert::ir::RELU1, and onert::ir::RELU6.


The documentation for this class was generated from the following file: