File tree Expand file tree Collapse file tree 1 file changed +16
-0
lines changed
Expand file tree Collapse file tree 1 file changed +16
-0
lines changed Original file line number Diff line number Diff line change @@ -91,6 +91,22 @@ 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+
94110inline ActivationMode StringToActivationMode (const std::string& str) {
95111 if (str == " identity" ) {
96112 return ActivationMode::kNone ;
You can’t perform that action at this time.
0 commit comments