Sign in
android
/
platform
/
external
/
pytorch
/
e8e3980e6593200d32b0288e7d6e2b384b2db849
/
.
/
caffe2
/
operators
/
data_couple_gpu.cu
blob: 81736d20556d853e1d3432c2972f5b059388d086 [
file
] [
log
] [
blame
]
#include
"caffe2/core/context_gpu.h"
#include
"caffe2/operators/data_couple.h"
namespace
caffe2
{
REGISTER_CUDA_OPERATOR
(
DataCouple
,
DataCoupleOp
<
CUDAContext
>);
}