Sign in
android
/
platform
/
external
/
pytorch
/
a88463cd9a
/
.
/
caffe2
/
image
/
image_input_op_gpu.cc
blob: c69889c3f8127865d2eea6f1ba00c790fd387107 [
file
] [
log
] [
blame
]
#include
"caffe2/core/common_gpu.h"
#include
"caffe2/core/context_gpu.h"
#include
"caffe2/image/image_input_op.h"
namespace
caffe2
{
REGISTER_CUDA_OPERATOR
(
ImageInput
,
ImageInputOp
<
CUDAContext
>);
}
// namespace caffe2