Fixes after review

openvinotoolkit · May 12, 2021 · f6c7687 · f6c7687
1 parent df4fd54
commit f6c7687
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 2 deletions.
diff --git a/inference-engine/src/mkldnn_plugin/mkldnn_graph.cpp b/inference-engine/src/mkldnn_plugin/mkldnn_graph.cpp
@@ -754,7 +754,7 @@ void MKLDNNGraph::PullOutputData(const BlobMap &out) {
         const MKLDNNMemory& intr_blob = node->getParentEdgeAt(0)->getMemory();
 
         if (!out.count(name)) {
-            continue;
+            IE_THROW(Unexpected) << "The network outputs do not contain mkldnn graph output node name: \"" << name << "\"";
         }
 
         const Blob::Ptr &ext_blob = out.at(name);

diff --git a/inference-engine/src/mkldnn_plugin/mkldnn_infer_request.cpp b/inference-engine/src/mkldnn_plugin/mkldnn_infer_request.cpp
@@ -251,7 +251,7 @@ InferenceEngine::Blob::Ptr MKLDNNPlugin::MKLDNNInferRequest::GetBlob(const std::
         if (_outputs.find(name) == _outputs.end()) {
             if (!data) {
                 InferenceEngine::TensorDesc desc = _networkOutputs[name]->getTensorDesc();
-        desc.setPrecision(normalizeToSupportedPrecision(desc.getPrecision()));
+                desc.setPrecision(normalizeToSupportedPrecision(desc.getPrecision()));
 
                 // WA: need to avoid exception thrown when we compare blocking desc in SetBlob
                 // in situation if we push output blobs as inputs for next network (in Hetero plugin)

diff --git a/inference-engine/tests/functional/plugin/cpu/subgraph_tests/src/param_result_custom_blob.cpp b/inference-engine/tests/functional/plugin/cpu/subgraph_tests/src/param_result_custom_blob.cpp
@@ -20,6 +20,7 @@ class ParameterResultCustomBlobTest : public ParameterResultSubgraphTest {
         auto inputBlob = inputs.front();
         const size_t elementsCount = inputBlob->size();
         for (size_t i = 0; i < inferIterations; ++i) {
+            CommonTestUtils::fill_data_random<Precision::FP32>(inputBlob, 10, 0, 1, i);
             const auto& inputsInfo = cnnNetwork.getInputsInfo().begin()->second;
             std::string inputName = cnnNetwork.getInputsInfo().begin()->first;