17 #include "caffe2/core/context_gpu.h" 18 #include "caffe2/experiments/operators/tt_contraction_op.h" 22 REGISTER_CUDA_OPERATOR(TTContraction, TTContractionOp<float, CUDAContext>);
23 REGISTER_CUDA_OPERATOR(
24 TTContractionGradient,
25 TTContractionGradientOp<float, CUDAContext>);
A global dictionary that holds information about what Caffe2 modules have been loaded in the current ...