forked from kevinlin311tw/Caffe-DeepBinaryCode
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathcudnn.cpp
More file actions
23 lines (19 loc) · 646 Bytes
/
cudnn.cpp
File metadata and controls
23 lines (19 loc) · 646 Bytes
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
#ifdef USE_CUDNN
#include "caffe/util/cudnn.hpp"
namespace caffe {
namespace cudnn {
float dataType<float>::oneval = 1.0;
float dataType<float>::zeroval = 0.0;
const void* dataType<float>::one =
static_cast<void *>(&dataType<float>::oneval);
const void* dataType<float>::zero =
static_cast<void *>(&dataType<float>::zeroval);
double dataType<double>::oneval = 1.0;
double dataType<double>::zeroval = 0.0;
const void* dataType<double>::one =
static_cast<void *>(&dataType<double>::oneval);
const void* dataType<double>::zero =
static_cast<void *>(&dataType<double>::zeroval);
} // namespace cudnn
} // namespace caffe
#endif