From 0668b343f0d4e47c1a5c18dc1635a87b8f31227b Mon Sep 17 00:00:00 2001 From: Megvii Engine Team Date: Tue, 7 Apr 2020 21:54:27 +0800 Subject: [PATCH] feat(dnn/arm_common): add arm_common int8 channel wise stride 1 GitOrigin-RevId: 679916a986ad6aff22821148179cb17e32b5bfc4 --- dnn/src/fallback/conv_bias/common.h | 16 ++++++++++++++++ 1 file changed, 16 insertions(+) diff --git a/dnn/src/fallback/conv_bias/common.h b/dnn/src/fallback/conv_bias/common.h index aac5ce72c..29fdae8c0 100644 --- a/dnn/src/fallback/conv_bias/common.h +++ b/dnn/src/fallback/conv_bias/common.h @@ -122,6 +122,22 @@ using BiasMode = ConvBiasForward::BiasMode; break; \ } +#define DISPATCH_FILTER_CHANNEL_WISE(filter, kern, arg...) \ + switch (filter) { \ + case 2: \ + kern(2, ##arg); \ + break; \ + case 3: \ + kern(3, ##arg); \ + break; \ + case 5: \ + kern(5, ##arg); \ + break; \ + default: \ + megdnn_assert(0); \ + break; \ + } + enum class PostprocessMode : uint8_t { FLOAT = 0, ///< support all biasmode and no_nonlinemode NO_PROCESS, ///