Browse Source
feat(dnn/arm_common): add arm_common int8 channel wise stride 1
GitOrigin-RevId: 679916a986
tags/v0.4.0
Megvii Engine Team
Xinran Xu
5 years ago
1 changed files with 16 additions and 0 deletions
Unified View
Diff Options
+ 16
- 0
dnn/src/fallback/conv_bias/common.h
View File
| @@ -122,6 +122,22 @@ using BiasMode = ConvBiasForward::BiasMode; | |||||
| break; \ | 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 { | enum class PostprocessMode : uint8_t { | ||||
| FLOAT = 0, ///< support all biasmode and no_nonlinemode | FLOAT = 0, ///< support all biasmode and no_nonlinemode | ||||
| NO_PROCESS, ///<support non bias and identity | NO_PROCESS, ///<support non bias and identity | ||||