disable squeeze layout optimizations for padded layouts (#2118)

* disable squeeze layout optimizations for padded layouts * knob squeeze padded layouts

disable squeeze layout optimizations for padded layouts (#2118)
* disable squeeze layout optimizations for padded layouts * knob squeeze padded layouts
bdfa4719 · Jayaram Bobba · Scott Cyphers · 5934fc61 · bdfa4719
Commit bdfa4719 authored Nov 27, 2018 by Jayaram Bobba Committed by Scott Cyphers Nov 27, 2018
Hide whitespace changes
Inline Side-by-side

Showing with 11 additions and 0 deletions

cpu_layout.cpp src/ngraph/runtime/cpu/pass/cpu_layout.cpp +11 -0

No files found.
--- a/src/ngraph/runtime/cpu/pass/cpu_layout.cpp
+++ b/src/ngraph/runtime/cpu/pass/cpu_layout.cpp
@@ -1311,7 +1311,18 @@ namespace ngraph
                    }

                    if (mkldnn_utils::is_mkldnn_padded_layout(md, squeezed_axis))
+                    {
                        return false;
+                    }
+
+                    if (std::getenv("NGRAPH_CPU_ENABLE_SQUEEZE_PADDED_LAYOUTS") == nullptr)
+                    {
+                        if (mkldnn_utils::is_mkldnn_padded_layout(
+                                md, ngraph::get_default_order(input_shape)))
+                        {
+                            return false;
+                        }
+                    }

                    return true;
                }