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

GroupedGatherElimination short circuit (#12380)

Browse Source 
* Disable GroupedGatherElimination in case of scalar inputs containing indices

* clang format
This commit is contained in:
Tomasz Dołbniak 2022-08-03 11:47:22 +02:00 committed by GitHub
parent 9d5e799c62
commit 62f79c3222
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 10 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
src/common/transformations/src/transformations/common_optimizations/simplify_shape_of_sub_graph.cpp
View File

@ -75,7 +75,16 @@ ngraph::pass::GroupedGatherElimination::GroupedGatherElimination() {
(curr->input_value(0) != next->input_value(0))) { (curr->input_value(0) != next->input_value(0))) {
++i; ++i;
continue; continue;
} // curr and next are the same type of gather which takes data from the same source }
// Scalar inputs are not supported by Concat and we don't want to throw an exception here.
// The transformation should not be applied instead.
if (curr->input_value(1).get_partial_shape().same_scheme(Shape{}) ||
next->input_value(1).get_partial_shape().same_scheme(Shape{})) {
return false;
}
// curr and next are the same type of gather which takes data from the same source
auto joint_indices = ngraph::op::util::make_try_fold<opset1::Concat>( auto joint_indices = ngraph::op::util::make_try_fold<opset1::Concat>(
OutputVector{curr->input_value(1), next->input_value(1)}, OutputVector{curr->input_value(1), next->input_value(1)},
0); 0);