| 95 | } |
| 96 | |
| 97 | static VARP bottleNeckBlock(VARP x, INTS channels, int stride, int expansionRatio, int number) { |
| 98 | x = bottleNeck(x, {channels[0], channels[1]}, stride, expansionRatio); |
| 99 | for (int i = 1; i < number; ++i) { |
| 100 | x = bottleNeck(x, {channels[1], channels[1]}, 1, expansionRatio); |
| 101 | } |
| 102 | return x; |
| 103 | } |
| 104 | |
| 105 | VARP mobileNetV2Expr(int numClass) { |
| 106 | auto x = _Input({1, 3, 224, 224}, NC4HW4); |
no test coverage detected