| 28 | } |
| 29 | |
| 30 | static VARP residualBlock(VARP x, INTS channels, int stride, int number) { |
| 31 | x = residual(x, {channels[0], channels[1]}, stride); |
| 32 | for (int i = 1; i < number; ++i) { |
| 33 | x = residual(x, {channels[1], channels[1]}, 1); |
| 34 | } |
| 35 | return x; |
| 36 | } |
| 37 | |
| 38 | static VARP bottleNeck(VARP x, INTS channels, int stride) { |
| 39 | int inputChannel = x->getInfo()->dim[1], narrowChannel = channels[1], outputChannel = channels[2]; |
no test coverage detected