| 146 | // ----------------------------------------------------------------------------------------------- |
| 147 | |
| 148 | InplaceConcatConv2::InplaceConcatConv2(Engine* engine, const ConcatConvDesc& desc) |
| 149 | : ConcatConv2(desc) |
| 150 | { |
| 151 | // Convolution 0: dst = conv(src0, weight0) + bias |
| 152 | conv0 = engine->newConv({src0Desc, weight0Desc, biasDesc, Activation::None, fusion, fastMath}); |
| 153 | |
| 154 | // Convolution 1: dst = activation(conv(src1, weight1) + dst) |
| 155 | // We use dst as bias |
| 156 | conv1 = engine->newConv({src1Desc, weight1Desc, dstDesc, activation, Fusion::None, fastMath}); |
| 157 | } |
| 158 | |
| 159 | bool InplaceConcatConv2::isSupported() const |
| 160 | { |