File tree Expand file tree Collapse file tree 1 file changed +13
-13
lines changed Expand file tree Collapse file tree 1 file changed +13
-13
lines changed Original file line number Diff line number Diff line change @@ -68,6 +68,19 @@ inline const char* cudnnGetErrorString(cudnnStatus_t status) {
68
68
} \
69
69
} while (false )
70
70
71
+ enum class DataLayout { // Not use
72
+ kNHWC ,
73
+ kNCHW ,
74
+ kNCDHW ,
75
+ kNCHW_VECT_C ,
76
+ };
77
+
78
+ enum class PoolingMode {
79
+ kMaximum ,
80
+ kAverage ,
81
+ kMaximumDeterministic ,
82
+ };
83
+
71
84
#if CUDNN_VERSION < 6000
72
85
#pragma message "CUDNN version under 6.0 is supported at best effort."
73
86
#pragma message "We strongly encourage you to move to 6.0 and above."
@@ -103,19 +116,6 @@ inline cudnnPoolingMode_t GetPoolingMode(const PoolingMode& mode) {
103
116
}
104
117
#endif // CUDNN_VERSION < 6000
105
118
106
- enum class DataLayout { // Not use
107
- kNHWC ,
108
- kNCHW ,
109
- kNCDHW ,
110
- kNCHW_VECT_C ,
111
- };
112
-
113
- enum class PoolingMode {
114
- kMaximum ,
115
- kAverage ,
116
- kMaximumDeterministic ,
117
- };
118
-
119
119
template <typename T>
120
120
class CudnnDataType ;
121
121
You can’t perform that action at this time.
0 commit comments