colin
/
TIM-VX
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add support for Mish, SoftRelu and HardSigmoid 

Signed-off-by: Kainan Cha <kainan.cha@verisilicon.com>
This commit is contained in:
Kainan Cha 2021-03-31 12:00:02 +08:00
parent f92a5de68b
commit 90a52ea6c9
3 changed files with 10 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
include/tim/vx/ops/activations.h
View File

@ -42,6 +42,9 @@ DECLARE_NO_PARAMETER_ACTIVATION(Elu)
DECLARE_NO_PARAMETER_ACTIVATION(Tanh) DECLARE_NO_PARAMETER_ACTIVATION(Tanh)
DECLARE_NO_PARAMETER_ACTIVATION(Sigmoid) DECLARE_NO_PARAMETER_ACTIVATION(Sigmoid)
DECLARE_NO_PARAMETER_ACTIVATION(HardSwish) DECLARE_NO_PARAMETER_ACTIVATION(HardSwish)
DECLARE_NO_PARAMETER_ACTIVATION(Mish)
DECLARE_NO_PARAMETER_ACTIVATION(HardSigmoid)
DECLARE_NO_PARAMETER_ACTIVATION(SoftRelu)
#undef DEFINE_NO_PARAMETER_ACTIVATION #undef DEFINE_NO_PARAMETER_ACTIVATION

6
src/tim/vx/ops/README.md
View File

@ -40,7 +40,7 @@ Sigmoid|SIGMOID|Mapped
Tanh|TANH|Mapped Tanh|TANH|Mapped
Sqrt|SQRT|Mapped Sqrt|SQRT|Mapped
Rsqrt|RSQRT|Mapped Rsqrt|RSQRT|Mapped
||SOFTRELU|Unmapped SoftRelu|SOFTRELU|Unmapped
Div|DIVIDE|Mapped Div|DIVIDE|Mapped
||DROPOUT|Unmapped ||DROPOUT|Unmapped
||SHUFFLECHANNEL|Unmapped ||SHUFFLECHANNEL|Unmapped
@ -136,8 +136,8 @@ GatherNd|GATHER_ND|Mapped
||BATCHNORM_SINGLE|Unmapped ||BATCHNORM_SINGLE|Unmapped
||MOMENTS|Unmapped ||MOMENTS|Unmapped
||SQUEEZE|Unmapped ||SQUEEZE|Unmapped
||HARD_SIGMOID|Unmapped HardSigmoid|HARD_SIGMOID|Unmapped
||MISH|Unmapped Mish|MISH|Unmapped
||EXPAND_BROADCAST|Unmapped ||EXPAND_BROADCAST|Unmapped
||PRE_PROCESS_YUV444|Unmapped ||PRE_PROCESS_YUV444|Unmapped
||PRE_PROCESS_NV12|Unmapped ||PRE_PROCESS_NV12|Unmapped

4
src/tim/vx/ops/activations.cc
View File

@ -38,6 +38,10 @@ DEFINE_NO_PARAMETER_ACTIVATION(Relu1, VSI_NN_OP_RELU1)
DEFINE_NO_PARAMETER_ACTIVATION(Relu6, VSI_NN_OP_RELU6) DEFINE_NO_PARAMETER_ACTIVATION(Relu6, VSI_NN_OP_RELU6)
DEFINE_NO_PARAMETER_ACTIVATION(Elu, VSI_NN_OP_ELU) DEFINE_NO_PARAMETER_ACTIVATION(Elu, VSI_NN_OP_ELU)
DEFINE_NO_PARAMETER_ACTIVATION(Sigmoid, VSI_NN_OP_SIGMOID) DEFINE_NO_PARAMETER_ACTIVATION(Sigmoid, VSI_NN_OP_SIGMOID)
DEFINE_NO_PARAMETER_ACTIVATION(Mish, VSI_NN_OP_MISH)
DEFINE_NO_PARAMETER_ACTIVATION(HardSigmoid, VSI_NN_OP_HARD_SIGMOID)
DEFINE_NO_PARAMETER_ACTIVATION(SoftRelu, VSI_NN_OP_SOFTRELU)
#undef DEFINE_NO_PARAMETER_ACTIVATION #undef DEFINE_NO_PARAMETER_ACTIVATION