From 90a52ea6c993c256146f8c36670f7dd5e5da878d Mon Sep 17 00:00:00 2001 From: Kainan Cha Date: Wed, 31 Mar 2021 12:00:02 +0800 Subject: [PATCH] Add support for Mish, SoftRelu and HardSigmoid Signed-off-by: Kainan Cha --- include/tim/vx/ops/activations.h | 3 +++ src/tim/vx/ops/README.md | 6 +++--- src/tim/vx/ops/activations.cc | 4 ++++ 3 files changed, 10 insertions(+), 3 deletions(-) diff --git a/include/tim/vx/ops/activations.h b/include/tim/vx/ops/activations.h index 3640e29..09580e9 100644 --- a/include/tim/vx/ops/activations.h +++ b/include/tim/vx/ops/activations.h @@ -42,6 +42,9 @@ DECLARE_NO_PARAMETER_ACTIVATION(Elu) DECLARE_NO_PARAMETER_ACTIVATION(Tanh) DECLARE_NO_PARAMETER_ACTIVATION(Sigmoid) 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 diff --git a/src/tim/vx/ops/README.md b/src/tim/vx/ops/README.md index 9586ca2..9ad5078 100644 --- a/src/tim/vx/ops/README.md +++ b/src/tim/vx/ops/README.md @@ -40,7 +40,7 @@ Sigmoid|SIGMOID|Mapped Tanh|TANH|Mapped Sqrt|SQRT|Mapped Rsqrt|RSQRT|Mapped -||SOFTRELU|Unmapped +SoftRelu|SOFTRELU|Unmapped Div|DIVIDE|Mapped ||DROPOUT|Unmapped ||SHUFFLECHANNEL|Unmapped @@ -136,8 +136,8 @@ GatherNd|GATHER_ND|Mapped ||BATCHNORM_SINGLE|Unmapped ||MOMENTS|Unmapped ||SQUEEZE|Unmapped -||HARD_SIGMOID|Unmapped -||MISH|Unmapped +HardSigmoid|HARD_SIGMOID|Unmapped +Mish|MISH|Unmapped ||EXPAND_BROADCAST|Unmapped ||PRE_PROCESS_YUV444|Unmapped ||PRE_PROCESS_NV12|Unmapped diff --git a/src/tim/vx/ops/activations.cc b/src/tim/vx/ops/activations.cc index 24353f5..b6b03c5 100644 --- a/src/tim/vx/ops/activations.cc +++ b/src/tim/vx/ops/activations.cc @@ -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(Elu, VSI_NN_OP_ELU) 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