Sign in
android
/
platform
/
external
/
pytorch
/
d95e68c8cc
/
.
/
caffe2
/
video
/
video_input_op_gpu.cc
blob: b9e60e63094a133932de6829d37d77e1c0b929e3 [
file
] [
log
] [
blame
]
#include
<caffe2/core/common_gpu.h>
#include
<caffe2/core/context_gpu.h>
#include
<caffe2/video/video_input_op.h>
namespace
caffe2
{
REGISTER_CUDA_OPERATOR
(
VideoInput
,
VideoInputOp
<
CUDAContext
>);
}
// namespace caffe2