Added MaxPoolBackprop to the mkldnn_util

716fd2e9 · Pruthvi Gowda Thorehosur Appajigowda · 3208c02b · 716fd2e9
Commit 716fd2e9 authored Feb 14, 2018 by Pruthvi Gowda Thorehosur Appajigowda
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

mkldnn_utils.cpp src/ngraph/runtime/cpu/mkldnn_utils.cpp +2 -1

No files found.
--- a/src/ngraph/runtime/cpu/mkldnn_utils.cpp
+++ b/src/ngraph/runtime/cpu/mkldnn_utils.cpp
@@ -41,7 +41,8 @@ namespace ngraph
                    TI(ngraph::op::Convolution),
                    TI(ngraph::op::ConvolutionBackpropData),
                    TI(ngraph::op::ConvolutionBackpropFilters),
-                    TI(ngraph::op::MaxPool)};
+                    TI(ngraph::op::MaxPool),
+                    T1(ngraph::op::MaxPoolBackprop)};

                bool IsMKLDNNOp(ngraph::Node& op)
                {