doxygen-c/html/tt__contraction__op__gpu_8cc_source.html

 #include "caffe2/core/context_gpu.h"
 #include "caffe2/experiments/operators/tt_contraction_op.h"

 namespace caffe2 {

 REGISTER_CUDA_OPERATOR(TTContraction, TTContractionOp<float, CUDAContext>);
 REGISTER_CUDA_OPERATOR(
     TTContractionGradient,
     TTContractionGradientOp<float, CUDAContext>);

 } // namespace caffe2
caffe2
A global dictionary that holds information about what Caffe2 modules have been loaded in the current ...
Definition: blob.h:13