Function UpsampleDesc

core/upsample.cpp:6–17

#include "upsample.h"

OIDN_NAMESPACE_BEGIN

  Upsample::Upsample(const UpsampleDesc& desc)
    : UpsampleDesc(desc)
  {
    if (srcDesc.getRank() != 3)
      throw std::invalid_argument("invalid upsampling source shape");

    TensorDims dstDims{srcDesc.getC(), srcDesc.getH() * 2, srcDesc.getW() * 2};
    TensorDims dstPaddedDims{srcDesc.getPaddedC(), dstDims[1], dstDims[2]};
    dstDesc = {dstDims, dstPaddedDims, srcDesc.layout, srcDesc.dataType};
  }

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