Caffe2 - C++ API
A deep learning, cross platform ML framework
counter_ops_gpu.cc
1 #include "caffe2/core/context_gpu.h"
2 #include "caffe2/operators/counter_ops.h"
3 
4 namespace caffe2 {
5 REGISTER_CUDA_OPERATOR(CreateCounter, CreateCounterOp<int64_t, CUDAContext>);
6 REGISTER_CUDA_OPERATOR(ResetCounter, ResetCounterOp<int64_t, CUDAContext>);
7 REGISTER_CUDA_OPERATOR(CountDown, CountDownOp<int64_t, CUDAContext>);
8 REGISTER_CUDA_OPERATOR(
9  CheckCounterDone,
10  CheckCounterDoneOp<int64_t, CUDAContext>);
11 REGISTER_CUDA_OPERATOR(CountUp, CountUpOp<int64_t, CUDAContext>);
12 REGISTER_CUDA_OPERATOR(RetrieveCount, RetrieveCountOp<int64_t, CUDAContext>);
13 } // namespace caffe2
A global dictionary that holds information about what Caffe2 modules have been loaded in the current ...
Definition: blob.h:13