提交 2b9ef7e2 编写于 作者: D dzhwinter

"fix"

上级 75d8e8ca
...@@ -68,6 +68,19 @@ inline const char* cudnnGetErrorString(cudnnStatus_t status) { ...@@ -68,6 +68,19 @@ inline const char* cudnnGetErrorString(cudnnStatus_t status) {
} \ } \
} while (false) } while (false)
enum class DataLayout { // Not use
kNHWC,
kNCHW,
kNCDHW,
kNCHW_VECT_C,
};
enum class PoolingMode {
kMaximum,
kAverage,
kMaximumDeterministic,
};
#if CUDNN_VERSION < 6000 #if CUDNN_VERSION < 6000
#pragma message "CUDNN version under 6.0 is supported at best effort." #pragma message "CUDNN version under 6.0 is supported at best effort."
#pragma message "We strongly encourage you to move to 6.0 and above." #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,19 +116,6 @@ inline cudnnPoolingMode_t GetPoolingMode(const PoolingMode& mode) {
} }
#endif // CUDNN_VERSION < 6000 #endif // CUDNN_VERSION < 6000
enum class DataLayout { // Not use
kNHWC,
kNCHW,
kNCDHW,
kNCHW_VECT_C,
};
enum class PoolingMode {
kMaximum,
kAverage,
kMaximumDeterministic,
};
template <typename T> template <typename T>
class CudnnDataType; class CudnnDataType;
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册