src/operator/modulated_deformable_convolution.cu - master - githubmirror/incubator-mxnet

src/operator/modulated_deformable_convolution.cu

28 строк789 B

Zhenghui Jin

Port #20520 (#20609)

28 сен 2021, 21:38

28 сен 2021, 21:3880d72b5

/*!
 * Copyright (c) 2018 Microsoft
 * Licensed under The MIT License [see LICENSE for details]
 * \file modulated_deformable_convolution.cu
 * \brief
 * \author Yuwen Xiong, Haozhi Qi, Jifeng Dai, Xizhou Zhu, Han Hu
 */

#include "./modulated_deformable_convolution-inl.h"
#include <vector>

namespace mxnet {
namespace op {

template <>
Operator* CreateOp<gpu>(ModulatedDeformableConvolutionParam param,
                        int dtype,
                        std::vector<TShape>* in_shape,
                        std::vector<TShape>* out_shape,
                        Context ctx) {
  Operator* op = nullptr;
  MSHADOW_REAL_TYPE_SWITCH(
      dtype, DType, { op = new ModulatedDeformableConvolutionOp<gpu, DType>(param); })
  return op;
}

}  // namespace op
}  // namespace mxnet