ONE - On-device Neural Engine
Loading...
Searching...
No Matches
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 36 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 93 of file AclActivationBuilder.h.

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

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: