plugin/torch/torch_module.cu

/*!
 * Copyright (c) 2015 by Contributors
 * \file activation.cc
 * \brief activation op
 * \author Bing Xu
*/
#include "./torch_module-inl.h"
#include "../../src/operator/mshadow_op.h"

namespace mxnet {
namespace op {
template<>
Operator *CreateOp<gpu>(TorchModuleParam param, TorchState* torchState) {
  return new TorchModuleOp<gpu>(param, torchState);
}

}  // namespace op
}  // namespace mxnet