Caffe2 - C++ API
A deep learning, cross platform ML framework
locally_connected_op_gpu.cc
1 
17 #include "caffe2/core/context_gpu.h"
18 #include "caffe2/operators/locally_connected_op.h"
19 #include "caffe2/operators/locally_connected_op_impl.h"
20 
21 namespace caffe2 {
22 
23 REGISTER_CUDA_OPERATOR(LC, LocallyConnectedOp<float, CUDAContext>);
24 REGISTER_CUDA_OPERATOR(
25  LCGradient,
26  LocallyConnectedGradientOp<float, CUDAContext>);
27 
28 REGISTER_CUDA_OPERATOR(LC1D, LocallyConnectedOp<float, CUDAContext>);
29 REGISTER_CUDA_OPERATOR(
30  LC1DGradient,
31  LocallyConnectedGradientOp<float, CUDAContext>);
32 
33 REGISTER_CUDA_OPERATOR(LC2D, LocallyConnectedOp<float, CUDAContext>);
34 REGISTER_CUDA_OPERATOR(
35  LC2DGradient,
36  LocallyConnectedGradientOp<float, CUDAContext>);
37 
38 REGISTER_CUDA_OPERATOR(LC3D, LocallyConnectedOp<float, CUDAContext>);
39 REGISTER_CUDA_OPERATOR(
40  LC3DGradient,
41  LocallyConnectedGradientOp<float, CUDAContext>);
42 
43 } // namespace caffe2
Copyright (c) 2016-present, Facebook, Inc.