Publishing 2019 R1 content

[platform/upstream/dldt.git] / inference-engine / src / inference_engine / ie_data.cpp

diff --git a/inference-engine/src/inference_engine/ie_data.cpp b/inference-engine/src/inference_engine/ie_data.cpp
index 7626620..8f91730 100644 (file)
--- a/inference-engine/src/inference_engine/ie_data.cpp
+++ b/inference-engine/src/inference_engine/ie_data.cpp
@@ -1,4 +1,4 @@
-// Copyright (C) 2018 Intel Corporation
+// Copyright (C) 2018-2019 Intel Corporation
 // SPDX-License-Identifier: Apache-2.0
 //
 
@@ -57,7 +57,7 @@ const TensorDesc& Data::getTensorDesc() const {
 }
 
 bool Data::isInitialized() const {
-    return !dims.empty() || !tensorDesc.getDims().empty();
+    return !dims.empty() || !tensorDesc.getDims().empty() || layout == SCALAR;
 }
 
 void Data::setDims(const SizeVector &a_dims) {
@@ -84,6 +84,14 @@ void Data::setLayout(Layout layout) {
     this->layout = layout;
 }
 
+void Data::reshape(const SizeVector &a_dims, Layout a_layout) {
+    dims = a_dims;
+    layout = a_layout;
+    std::reverse(dims.begin(), dims.end());
+
+    tensorDesc.reshape(a_dims, layout);
+}
+
 CNNLayerWeakPtr &Data::getCreatorLayer() {
     return creatorLayer;
 }