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

fix TSSqueeze transformation

Browse Source
This commit is contained in:
Tikhonov Ivan 2023-03-21 21:41:37 +00:00
parent 9720a28fa6
commit 31bd608ac4
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/common/transformations/src/transformations/transpose_sinking/ts_squeeze.cpp
View File

@ -109,7 +109,7 @@ TSSqueezeForward::TSSqueezeForward() {
// if 2nd input to squeeze is empty then all '1' dims will be deleted.
if (non_negative_axes.empty()) {
auto input_pshape = transpose->input_value(0).get_partial_shape();
auto input_pshape = transpose->output(0).get_partial_shape();
if (input_pshape.is_dynamic()) {
return false;
}
@ -136,7 +136,7 @@ TSSqueezeForward::TSSqueezeForward() {
new_values = squeeze_axes_to_shape(transpose->input_value(0), new_values);
}
auto new_const = Constant::create(squeeze_axes->get_element_type(), squeeze_axes->get_shape(), new_values);
auto new_const = Constant::create(squeeze_axes->get_element_type(), {new_values.size()}, new_values);
auto new_squeeze = squeeze->clone_with_new_inputs({transpose->input_value(0), new_const});
auto new_transpose = transpose->clone_with_new_inputs({new_squeeze, new_transpose_order});