accumulate_op.cu « operators « caffe2 - platform/upstream/pytorch - Domain: Machine Learning / ML Framework; Licenses: BSD-3-Clause;

index : platform/upstream/pytorch

Domain: Machine Learning / ML Framework; Licenses: BSD-3-Clause;

Geunsik Lim <geunsik.lim@samsung.com>, HyoungJoo Ahn <hello.ahn@samsung.com>, Jijoong Moon <jijoong.moon@samsung.com>, MyungJoo Ham <myungjoo.ham@samsung.com>, Parichay Kapoor <pk.kapoor@samsung.com>

summary refs log tree commit diff

path: root/caffe2/operators/accumulate_op.cu

blob: 96f0d0c5ba7d46dbe26aa5c43fda34cb192a4e4d (plain)

#include "caffe2/core/context_gpu.h"
#include "caffe2/operators/accumulate_op.h"

namespace caffe2 {
REGISTER_CUDA_OPERATOR(Accumulate, AccumulateOp<float, CUDAContext>);
}  // namespace caffe2