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

add the missed line

Browse Source
This commit is contained in:
Tikhonov Ivan 2023-03-02 17:21:02 +00:00
parent 873d15ce1a
commit 6de1336894
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/common/transformations/src/transformations/common_optimizations/transpose_sinking_interpolate.cpp
View File

@ -119,6 +119,8 @@ ov::pass::TransposeSinkingInterpolateBackward::TransposeSinkingInterpolateBackwa
std::make_shared<Constant>(element::i32, Shape{reversed_transpose_order.size()}, reversed_transpose_order);
const auto& indices = main_node->input_value(3);
auto new_axis = std::make_shared<Gather>(data, indices, axis);
main_node->input(3).replace_source_output(new_axis);
const auto& interpolate = std::dynamic_pointer_cast<Interpolate>(main_node);
if (interpolate) {
op::v4::Interpolate::InterpolateAttrs attrs = interpolate->get_attrs();