blob: dbf920d8afc1430660ac200963cb3769c7732f87 [file] [log] [blame]
#include "caffe2/operators/log_op.h"
#include "caffe2/core/context_gpu.h"
namespace caffe2 {
REGISTER_CUDA_OPERATOR(
Log,
UnaryElementwiseOp<
TensorTypes<float>,
CUDAContext,
LogFunctor<CUDAContext>>);
} // namespace caffe2