提交 0668b343 编写于 作者: M Megvii Engine Team 提交者: Xinran Xu

feat(dnn/arm_common): add arm_common int8 channel wise stride 1

GitOrigin-RevId: 679916a986ad6aff22821148179cb17e32b5bfc4
上级 6e9a104d
......@@ -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, ///<support non bias and identity
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册