blob: 5d96b99dbcbec9701d36918edd373396437e279e [file] [log] [blame]
#include "caffe2/core/context_gpu.h"
#include "caffe2/operators/scale_op.h"
namespace caffe2 {
REGISTER_CUDA_OPERATOR(Scale, ScaleOp<float, CUDAContext>);
} // namespace caffe2