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

Auto-batch ConvertLike enabled (#10631)

Browse Source
This commit is contained in:
Evgenya Stepyreva 2022-02-24 14:37:03 +03:00 committed by GitHub
parent 4c01d6c50c
commit 3de428c713
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
src/common/transformations/src/transformations/common_optimizations/dimension_tracking.cpp
View File

@ -143,7 +143,8 @@ P2Btype ov::batch_util::find_batch(const std::shared_ptr<ov::Model>& f) {
for (const auto& output : curr_node->outputs()) { for (const auto& output : curr_node->outputs()) {
// we do not need to walk through shape-of sub-graphs // we do not need to walk through shape-of sub-graphs
for (const auto& t_input : output.get_target_inputs()) { for (const auto& t_input : output.get_target_inputs()) {
if (ov::is_type<ov::opset1::ShapeOf>(t_input.get_node()) || if (ov::is_type<ov::opset1::ConvertLike>(t_input.get_node()) ||
ov::is_type<ov::opset1::ShapeOf>(t_input.get_node()) ||
ov::is_type<ov::opset3::ShapeOf>(t_input.get_node())) ov::is_type<ov::opset3::ShapeOf>(t_input.get_node()))
continue; continue;
nodes.push_back(t_input.get_node()); nodes.push_back(t_input.get_node());
@ -206,7 +207,7 @@ bool ov::batch_util::check_batch_tracks_through_all_the_nodes(const std::shared_
all_outputs_has_batch &= name_stays; // && others_are_static; all_outputs_has_batch &= name_stays; // && others_are_static;
} }
if (any_input_has_batch && !all_outputs_has_batch && !ov::is_type<ov::opset3::ShapeOf>(node) && if (any_input_has_batch && !all_outputs_has_batch && !ov::is_type<ov::opset3::ShapeOf>(node) &&
!ov::is_type<ov::opset1::ShapeOf>(node)) { !ov::is_type<ov::opset1::ShapeOf>(node) && !ov::is_type<ov::opset1::ConvertLike>(node)) {
failed_to_propagate_batch = true; failed_to_propagate_batch = true;
node->validate_and_infer_types(); node->validate_and_infer_types();
} }