IntenseWebs/openvino
3
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Used dynamic shape (#3234)

Browse Source
This commit is contained in:
Anton Pankratv 2020-11-24 15:50:15 +03:00 committed by GitHub
parent 13806bf719
commit b2714f18af
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
inference-engine/src/hetero_plugin/hetero_executable_network.cpp
View File

@ -561,7 +561,7 @@ void HeteroExecutableNetwork::InitNgraph(const InferenceEngine::ICNNNetwork& net
output.remove_target_input(input); output.remove_target_input(input);
auto result = std::make_shared<ngraph::op::Result>(output); auto result = std::make_shared<ngraph::op::Result>(output);
ngraph::copy_runtime_info(output.get_node_shared_ptr(), result); ngraph::copy_runtime_info(output.get_node_shared_ptr(), result);
auto parameter = std::make_shared<ngraph::op::Parameter>(output.get_element_type(), output.get_shape()); auto parameter = std::make_shared<ngraph::op::Parameter>(output.get_element_type(), output.get_partial_shape());
ngraph::copy_runtime_info(input.get_node()->shared_from_this(), parameter); ngraph::copy_runtime_info(input.get_node()->shared_from_this(), parameter);
input.replace_source_output(parameter->output(0)); input.replace_source_output(parameter->output(0));
results.push_back(result); results.push_back(result);