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

fix warnings

Browse Source
This commit is contained in:
Ivan 2023-03-17 16:06:15 +04:00
parent 66d16ae45e
commit 1f8fdfb353
2 changed files with 6 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -38,9 +38,10 @@ bool shape_to_squeeze_axes(const std::shared_ptr<Node>& reshape,
const auto input_shape = input_pshape.to_shape();
if (new_shape.size() < input_shape.size()) {
for (size_t i = 0, j = 0; i < new_shape.size(); j++) {
if (new_shape[i] == input_shape[j]) {
const auto input_dim = static_cast<int64_t>(input_shape[j]);
if (new_shape[i] == input_dim) {
i++;
} else if (new_shape[i] != input_shape[j] && input_shape[j] != 1) {
} else if (new_shape[i] != input_dim && input_dim != 1) {
return false;
} else {
result_axes.push_back(j);

14
src/common/transformations/src/transformations/transpose_sinking/ts_unsqueeze.cpp
View File

@ -38,9 +38,10 @@ bool shape_to_unsqueeze_axes(const std::shared_ptr<Node>& reshape,
const auto input_shape = input_pshape.to_shape();
if (new_shape.size() > input_shape.size()) {
for (size_t i = 0, j = 0; i < input_shape.size();j++) {
if (input_shape[i] == new_shape[j]) {
const auto input_dim = static_cast<int64_t>(input_shape[i]);
if (input_dim == new_shape[j]) {
i++;
} else if (input_shape[i] != new_shape[j] && new_shape[j] != 1) {
} else if (input_dim != new_shape[j] && new_shape[j] != 1) {
return false;
} else {
result_axes.push_back(j);
@ -102,20 +103,11 @@ TSUnsqueezeForward::TSUnsqueezeForward() {
}
auto ts_order_values = transpose_order->cast_vector<size_t>();
/* std::vector<size_t> new_values;
new_values.reserve(non_negative_axes.size());
for (const auto& axis : non_negative_axes) {
new_values.push_back(ts_order_values[axis]);
}*/
ts_order_values = GetOrderBeforeReduction(non_negative_axes, ts_order_values);
auto new_transpose_order = Constant::create(transpose_order->get_element_type(),
{ts_order_values.size()},
ts_order_values);
/*if (as_type_ptr<Reshape>(unsqueeze)) {
new_values = unsqueeze_axes_to_shape(unsqueeze, new_values);
}*/
auto new_unsqueeze = unsqueeze->clone_with_new_inputs({transpose->input_value(0), unsqueeze->input_value(1)});
auto new_transpose = transpose->clone_with_new_inputs({new_unsqueeze, new_transpose_order});