Sign in
android
/
platform
/
external
/
pytorch
/
9c2067cc49
/
.
/
caffe2
/
operators
/
accumulate_op.cu
blob: d042c13ea224280c6d77a18f945fbd5d468c78ed [
file
] [
log
] [
blame
]
#include
"caffe2/core/context_gpu.h"
#include
"caffe2/operators/accumulate_op.h"
namespace
caffe2
{
namespace
{
REGISTER_CUDA_OPERATOR
(
Accumulate
,
AccumulateOp
<
float
,
CUDAContext
>);
}
// namespace
}
// namespace caffe2