Caffe2 - C++ API
A deep learning, cross platform ML framework
conv_op_cache_cudnn.cc
1 
17 #include "caffe2/operators/conv_op_cache_cudnn.h"
18 
19 #include <cudnn.h>
20 
21 #include "caffe2/core/logging.h"
22 #include "caffe2/core/tensor.h"
23 
24 namespace caffe2 {
25 
26 template class AlgorithmsCache<cudnnConvolutionFwdAlgo_t>;
27 template class AlgorithmsCache<cudnnConvolutionBwdFilterAlgo_t>;
28 template class AlgorithmsCache<cudnnConvolutionBwdDataAlgo_t>;
29 template class AlgorithmsCache<int>; // For testing.
30 } // namespace caffe2
Copyright (c) 2016-present, Facebook, Inc.