blob: 9fe06b8faad2ba7723c268e32bbe275d4e27b527 [file] [log] [blame]
#include "caffe2/core/context_gpu.h"
#include "caffe2/operators/tile_op.h"
namespace caffe2 {
namespace {
REGISTER_CUDA_OPERATOR(Tile, TileOp<float, CUDAContext>);
REGISTER_CUDA_OPERATOR(TileGradient, TileGradientOp<float, CUDAContext>);
} // namespace
} // namespace caffe2