From b2714f18af5398c14a29cb399adbe7f7923f3d42 Mon Sep 17 00:00:00 2001 From: Anton Pankratv Date: Tue, 24 Nov 2020 15:50:15 +0300 Subject: [PATCH] Used dynamic shape (#3234) --- .../src/hetero_plugin/hetero_executable_network.cpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/inference-engine/src/hetero_plugin/hetero_executable_network.cpp b/inference-engine/src/hetero_plugin/hetero_executable_network.cpp index 59ba9d3947d..19dcd54877f 100644 --- a/inference-engine/src/hetero_plugin/hetero_executable_network.cpp +++ b/inference-engine/src/hetero_plugin/hetero_executable_network.cpp @@ -561,7 +561,7 @@ void HeteroExecutableNetwork::InitNgraph(const InferenceEngine::ICNNNetwork& net output.remove_target_input(input); auto result = std::make_shared(output); ngraph::copy_runtime_info(output.get_node_shared_ptr(), result); - auto parameter = std::make_shared(output.get_element_type(), output.get_shape()); + auto parameter = std::make_shared(output.get_element_type(), output.get_partial_shape()); ngraph::copy_runtime_info(input.get_node()->shared_from_this(), parameter); input.replace_source_output(parameter->output(0)); results.push_back(result);