24.02.1
|
#include <arm_gemm.hpp>
Public Types | |
enum | Type { None, ReLU, BoundedReLU } |
Public Member Functions | |
Activation (Type type=Type::None, float p1=0.0f, float p2=0.0f) | |
Data Fields | |
Type | type |
float | param1 |
float | param2 |
Definition at line 121 of file arm_gemm.hpp.
|
strong |
Enumerator | |
---|---|
None | |
ReLU | |
BoundedReLU |
Definition at line 123 of file arm_gemm.hpp.
|
inline |
float param1 |
Definition at line 131 of file arm_gemm.hpp.
Referenced by arm_gemm::activator(), arm_compute::assembly_utils::map_to_arm_gemm_activation(), and arm_gemm::MergeResults().
float param2 |
Definition at line 132 of file arm_gemm.hpp.
Referenced by arm_compute::assembly_utils::map_to_arm_gemm_activation().
Type type |
Definition at line 130 of file arm_gemm.hpp.
Referenced by arm_gemm::activator(), CpuDepthwiseConv2dAssemblyDispatch::is_activation_supported(), CpuGemmAssemblyDispatch::is_activation_supported(), arm_compute::assembly_utils::map_to_arm_gemm_activation(), and arm_gemm::MergeResults().