Function ConvDesc

core/conv.cpp:6–49

#include "conv.h"

OIDN_NAMESPACE_BEGIN

  Conv::Conv(const ConvDesc& desc)
    : ConvDesc(desc)
  {
    if (srcDesc.getRank() != 3)
      throw std::invalid_argument("invalid convolution source shape");
    if (weightDesc.getRank() != 4 ||
        weightDesc.getI() != srcDesc.getC() ||
        weightDesc.getPaddedI() != srcDesc.getPaddedC())
      throw std::invalid_argument("invalid convolution weight shape");

    TensorDims dstDims;
    switch (fusion)
    {
    case Fusion::None:
      dstDims = {weightDesc.getO(), srcDesc.getH(), srcDesc.getW()};
      break;

    case Fusion::UpsampleSrc0:
      dstDims = {weightDesc.getO(), srcDesc.getH() * 2, srcDesc.getW() * 2};
      break;

    case Fusion::PoolDst:
      if (srcDesc.getH() % 2 != 0 || srcDesc.getW() % 2 != 0)
        throw std::invalid_argument("invalid pooling source shape");
      dstDims = {weightDesc.getO(), srcDesc.getH() / 2, srcDesc.getW() / 2};
      break;

    default:
      throw std::invalid_argument("unsupported convolution fusion");
    }

    TensorDims dstPaddedDims = dstDims;
    dstPaddedDims[0] = weightDesc.getPaddedO();

    dstDesc = {dstDims, dstPaddedDims, srcDesc.layout, srcDesc.dataType};

    if (!((biasDesc.getRank() == 1 && biasDesc.getX() == weightDesc.getO()
                                   && biasDesc.getPaddedX() == weightDesc.getPaddedO()) ||
          (biasDesc.getRank() == 3 && biasDesc.dims == dstDesc.dims
                                   && biasDesc.paddedDims == dstDesc.paddedDims)))
      throw std::invalid_argument("invalid convolution bias shape");
  }

  void Conv::setSrc(const Ref<Tensor>& src)
  {