From 0a5dfc62b2e2713f6d0c905abfd2c2d2ccbeead7 Mon Sep 17 00:00:00 2001
From: Lukasz Debski <Lukasz.Debski@intel.com>
Date: Mon, 24 Aug 2020 12:03:21 +0200
Subject: [PATCH] [IE CLDNN] Forcing bfzyx format in MVN layer (#1903)

---
 inference-engine/thirdparty/clDNN/src/layout_optimizer.cpp | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/inference-engine/thirdparty/clDNN/src/layout_optimizer.cpp b/inference-engine/thirdparty/clDNN/src/layout_optimizer.cpp
index feee03512ef..04bb8247fd4 100644
--- a/inference-engine/thirdparty/clDNN/src/layout_optimizer.cpp
+++ b/inference-engine/thirdparty/clDNN/src/layout_optimizer.cpp
@@ -852,6 +852,11 @@ format layout_optimizer::get_preferred_format(program_node& node) {
         auto& deconv_node = node.as<deconvolution>();
         auto weights_layout = deconv_node.weights(0).get_output_layout();
         expected = get_expected_layout(output_layout, deconv_node, weights_layout).format;
+    } else if (node.is_type<mvn>()) {
+        auto input_layout = node.get_dependency(0).get_output_layout();
+        if (input_layout.format.dimension() == 5 &&
+            (input_layout.data_type == data_types::f32 || input_layout.data_type == data_types::f16 ))
+            expected = format::bfzyx;
     }
 
     return expected;