Method PreConcatConvCHW

core/concat_conv.cpp:119–132

  // -----------------------------------------------------------------------------------------------

  PreConcatConvCHW::PreConcatConvCHW(Engine* engine, const ConcatConvDesc& desc)
    : ConcatConv(desc)
  {
    if (src0Desc.layout == TensorLayout::hwc)
      throw std::invalid_argument("unsupported concat+conv source layout");
    if (fusion != Fusion::None && fusion != Fusion::PoolDst) // only post-ops supported
      throw std::invalid_argument("unsupported concat+conv fusion");

    TensorDims srcDims{src0Desc.getC() + src1Desc.getC(), src0Desc.getH(), src0Desc.getW()};
    TensorDims srcPaddedDims{src0Desc.getPaddedC() + src1Desc.getPaddedC(), src0Desc.getH(), src0Desc.getW()};
    srcDesc = {srcDims, srcPaddedDims, src0Desc.layout, src0Desc.dataType};

    conv = engine->newConv({srcDesc, weightDesc, biasDesc, activation, fusion, fastMath});
  }

  void PreConcatConvCHW::updateSrc()
  {