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

fix /gather_sinking_transpose_reshape

Browse Source
This commit is contained in:
Evgeny Kotov 2023-03-20 19:48:56 +01:00 committed by Mikhail Ryzhov
parent e6650cfd1c
commit 7b1bbf77ee
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/plugins/intel_gna/src/transformations/gather_sinking_transpose_reshape.cpp
View File

@ -133,7 +133,7 @@ NodePair SinkBackward(NodePtr transpose, std::shared_ptr<Constant> transpose_con
ov::copy_runtime_info({transpose}, {gather, gather_indices, gather_axis, reshape_new, reshape_const_new});
reshape_new->set_friendly_name(transpose->get_friendly_name());
return std::make_pair(transpose, reshape_new);
return std::make_pair(gather, reshape_new);
}
bool AreFlattenShapes(const Shape& shape1, const Shape& shape2) {