From a7ede592c3632e5b8eed3f6b399e18ab9bc9def5 Mon Sep 17 00:00:00 2001
From: Anton Chetverikov <Anton.Chetverikov@intel.com>
Date: Thu, 3 Dec 2020 19:33:07 +0300
Subject: [PATCH] Fix mode attribute value in DepthToSpace ONNX operation
 extractor (#3466)

---
 model-optimizer/extensions/front/onnx/depth_to_space_ext.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/model-optimizer/extensions/front/onnx/depth_to_space_ext.py b/model-optimizer/extensions/front/onnx/depth_to_space_ext.py
index f043a1c89c9..d8f95ea92a3 100644
--- a/model-optimizer/extensions/front/onnx/depth_to_space_ext.py
+++ b/model-optimizer/extensions/front/onnx/depth_to_space_ext.py
@@ -34,9 +34,9 @@ class DepthToSpaceFrontExtractor(FrontExtractorOp):
         onnx_mode = onnx_attr(node, 'mode', 's', default=b'DCR').decode()
         assert onnx_mode in ['DCR', 'CRD'], 'Unrecognized mode provided for DepthToSpace node {}'.format(node_name)
         if onnx_mode == 'DCR':
-            mode = 'depth_first'
-        else:
             mode = 'blocks_first'
+        else:
+            mode = 'depth_first'
 
         DepthToSpaceOp.update_node_stat(node, {'block_size': block_size, 'mode': mode})
         return cls.enabled