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

Transformation component transition to OV namespace (phase 4) (#13558)

Browse Source 
* Use openvino pass graph_rewrite

* Replace use of ngraph::pass to ov::pass

* Remove ngraph::pass aliases

* Update after rebase

* Fix post merge conflicts

* Apply code style

* Apply code style

* Remove bypassing inclusion

* Apply code style
This commit is contained in:
Tomasz Jankowski authored and GitHub committed 2023-01-25 01:02:35 +03:00
1 parent 163baa3c1c
commit 988847f559
555 files changed
+1975 -3041
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

src/plugins/auto_batch/auto_batch.cpp
+4 -4
View File
@@ -851,7 +851,7 @@ InferenceEngine::IExecutableNetworkInternal::Ptr AutoBatchInferencePlugin::LoadN
auto function = clonedNetwork.getFunction();
// find the batch dim
ov::pass::Manager m;
m.register_pass<ngraph::pass::InitNodeInfo>();
m.register_pass<ov::pass::InitNodeInfo>();
m.register_pass<ov::pass::FindBatch>(false, check_dims);
m.run_passes(function);
// do not reshape/re-batch originally batched networks and when there are no inputs with the N* layouts
@@ -869,7 +869,7 @@ InferenceEngine::IExecutableNetworkInternal::Ptr AutoBatchInferencePlugin::LoadN
if (static_shape[0] != 1)
IE_THROW(NotImplemented) << "Auto-batching does not reshape/re-batch originally batched networks!";
batched_inputs.insert(
ngraph::op::util::get_ie_output_name(params[input_id]->output(0))); // batched dim for the input
ov::op::util::get_ie_output_name(params[input_id]->output(0))); // batched dim for the input
} else {
// if the 0-th dim is not for the batch, then we support only the case when NONE dimension is batch
for (size_t s = 1; s < shape.size(); s++)
@@ -889,8 +889,8 @@ InferenceEngine::IExecutableNetworkInternal::Ptr AutoBatchInferencePlugin::LoadN
if (shape[0] != 1)
IE_THROW(NotImplemented) << "Auto-batching does not reshape/re-batch originally batched networks!";
const auto& node = output->input_value(0);
batched_outputs.insert(ngraph::op::util::get_ie_output_name(
ov::Output<const ov::Node>(node.get_node(), node.get_index())));
batched_outputs.insert(
ov::op::util::get_ie_output_name(ov::Output<const ov::Node>(node.get_node(), node.get_index())));
} else {
// if the 0-th dim is not for the batch, then we support only the case when NONE dimension is batch
for (size_t s = 1; s < shape.size(); s++)