CPU: Generalize pooling

src/ngraph/runtime/cpu/cpu_emitter.cpp

@@ -2281,7 +2281,8 @@ namespace ngraph
                     auto result_desc = mkldnn_emitter->build_memory_descriptor(
                         out[0], runtime::cpu::mkldnn_utils::get_output_mkldnn_format(node, 0));
 
-                    size_t max_pool_index = mkldnn_emitter->build_max_pool_forward(
+                    size_t max_pool_index = mkldnn_emitter->build_pooling_forward(
+                        mkldnn::algorithm::pooling_max,
                         input_desc,
                         result_desc,
                         max_pool->get_window_movement_strides(),

src/ngraph/runtime/cpu/mkldnn_emitter.cpp

@@ -222,7 +222,8 @@ size_t MKLDNNEmitter::build_convolution_backward_data(const mkldnn::memory::desc
     return primitive_index;
 }
 
-size_t MKLDNNEmitter::build_max_pool_forward(const mkldnn::memory::desc& input_desc,
+size_t MKLDNNEmitter::build_pooling_forward(mkldnn::algorithm pooling_algorithm,
+                                            const mkldnn::memory::desc& input_desc,
                                             const mkldnn::memory::desc& result_desc,
                                             const ngraph::Strides& window_strides,
                                             const ngraph::Shape& window_shape,
@@ -234,7 +235,7 @@ size_t MKLDNNEmitter::build_max_pool_forward(const mkldnn::memory::desc& input_d
 
     size_t primitive_index = insert_primitive(new mkldnn::pooling_forward(
         {{mkldnn::prop_kind::forward_inference,
-          mkldnn::algorithm::pooling_max,
+          pooling_algorithm,
           input_desc,
           result_desc,
           mkldnn::memory::dims(window_strides.begin(), window_strides.end()),

src/ngraph/runtime/cpu/mkldnn_emitter.hpp

@@ -87,7 +87,8 @@ namespace ngraph
                                                        const ngraph::CoordinateDiff& padding_below,
                                                        const ngraph::CoordinateDiff& padding_above);
 
-                size_t build_max_pool_forward(const mkldnn::memory::desc& input_desc,
+                size_t build_pooling_forward(mkldnn::algorithm pooling_algorithm,
+                                             const mkldnn::memory::desc& input_desc,
                                              const mkldnn::memory::desc& result_desc,
                                              const ngraph::Strides& window_strides,
                                              const ngraph::Shape& window_shape,