| 301 | } |
| 302 | |
| 303 | SymbolVar mgb::bottleneck( |
| 304 | Network& network, SymbolVar f, size_t input_channels, size_t channels, size_t t, |
| 305 | size_t stride, DType out_dtype) { |
| 306 | size_t in_channels = f.node()->shape()[1]; |
| 307 | SymbolVar x = f; |
| 308 | if (t != 1) { |
| 309 | x = network.add_conv( |
| 310 | f, input_channels * t, {1, 1}, out_dtype, true, {1, 1}, {0, 0}); |
| 311 | } |
| 312 | x = network.add_group_conv( |
| 313 | x, input_channels * t, input_channels * t, {3, 3}, out_dtype, true, |
| 314 | {stride, stride}, {1, 1}); |
| 315 | x = network.add_conv(x, channels, {1, 1}, out_dtype, false, {1, 1}, {0, 0}); |
| 316 | if (stride == 1 && in_channels == channels) |
| 317 | x = f + x; |
| 318 | return x; |
| 319 | } |
| 320 | |
| 321 | SymbolVar mgb::bottleneck_group( |
| 322 | Network& network, SymbolVar f, size_t input_channels, size_t channels, |
nothing calls this directly
no test coverage detected