Sign in
android
/
platform
/
external
/
pytorch
/
aec09eeb3a570bb79c642e8ae540b3981e7d69cd
/
.
/
caffe2
/
operators
/
log_op_gpu.cc
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