1 #include "caffe2/core/context_gpu.h" 2 #include "caffe2/experiments/operators/fully_connected_op_decomposition.h" 6 REGISTER_CUDA_OPERATOR(FC_Decomp, FullyConnectedOpDecomp<float, CUDAContext>);
7 REGISTER_CUDA_OPERATOR(FCGradient_Decomp,
8 FullyConnectedDecompGradientOp<float, CUDAContext>);
A global dictionary that holds information about what Caffe2 modules have been loaded in the current ...