File tree Expand file tree Collapse file tree 1 file changed +0
-17
lines changed
Expand file tree Collapse file tree 1 file changed +0
-17
lines changed Original file line number Diff line number Diff line change @@ -91,23 +91,6 @@ enum class ActivationMode {
9191 kBandPass ,
9292};
9393
94- inline cudnnPoolingMode_t GetPoolingMode (const PoolingMode& mode) {
95- switch (mode) {
96- case PoolingMode::kMaximumDeterministic :
97- return CUDNN_POOLING_MAX_DETERMINISTIC;
98- case PoolingMode::kAverageExclusive :
99- return CUDNN_POOLING_AVERAGE_COUNT_EXCLUDE_PADDING;
100- case PoolingMode::kAverageInclusive :
101- return CUDNN_POOLING_AVERAGE_COUNT_INCLUDE_PADDING;
102- case PoolingMode::kMaximum :
103- return CUDNN_POOLING_MAX;
104- default :
105- PADDLE_THROW (
106- platform::errors::Unimplemented (" Unexpected CUDNN pooling mode." ));
107- }
108- }
109- #endif // CUDNN_VERSION < 6000
110-
11194inline ActivationMode StringToActivationMode (const std::string& str) {
11295 if (str == " identity" ) {
11396 return ActivationMode::kNone ;
You can’t perform that action at this time.
0 commit comments