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

ONNX Resize fix (#2103)

Browse Source
This commit is contained in:
Tomasz Dołbniak 2020-09-08 08:51:30 +02:00 committed by GitHub
parent 928eed9a51
commit f42921ce47
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
ngraph/frontend/onnx_import/src/op/resize.cpp
View File

@ -217,7 +217,7 @@ namespace ngraph
const auto converted_sizes =
std::make_shared<default_opset::Convert>(sizes, ngraph::element::f32);
const auto divide =
std::make_shared<default_opset::Divide>(sizes, shape_of_data);
std::make_shared<default_opset::Divide>(converted_sizes, shape_of_data);
const auto eps_node = std::make_shared<default_opset::Constant>(
ngraph::element::f32, Shape{}, epsilon);
const auto scales = std::make_shared<default_opset::Add>(divide, eps_node);
@ -262,7 +262,7 @@ namespace ngraph
return {
std::make_shared<ngraph::op::v0::Interpolate>(data, output_shape, attrs)};
}
}
} // namespace
namespace set_11
{
@ -336,4 +336,4 @@ namespace ngraph
} // namespace onnx_import
} // namespace ngraph
} // namespace ngraph