Caffe2 - C++ API
A deep learning, cross platform ML framework
Related Pages
Modules
Data Structures
Files
C++ API
Python API
GitHub
File List
Globals
caffe2
experiments
operators
tt_contraction_op_gpu.cc
1
#include "caffe2/core/context_gpu.h"
2
#include "caffe2/experiments/operators/tt_contraction_op.h"
3
4
namespace
caffe2
{
5
6
REGISTER_CUDA_OPERATOR(TTContraction, TTContractionOp<float, CUDAContext>);
7
REGISTER_CUDA_OPERATOR(
8
TTContractionGradient,
9
TTContractionGradientOp<float, CUDAContext>);
10
11
}
// namespace caffe2
caffe2
A global dictionary that holds information about what Caffe2 modules have been loaded in the current ...
Definition:
convert_encoded_to_raw_leveldb.cc:47
Generated on Thu Apr 19 2018 13:03:50 for Caffe2 - C++ API by
1.8.11