From ddaceb047d3f77f118764ec52a3fe17e0e4d109e Mon Sep 17 00:00:00 2001 From: Katarzyna Mitrus Date: Wed, 9 Sep 2020 09:50:27 +0200 Subject: [PATCH] Use element type instead of F32 for constants (#2121) --- ngraph/frontend/onnx_import/src/op/image_scaler.cpp | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/ngraph/frontend/onnx_import/src/op/image_scaler.cpp b/ngraph/frontend/onnx_import/src/op/image_scaler.cpp index 4f62246eb02..c047fb9fda4 100644 --- a/ngraph/frontend/onnx_import/src/op/image_scaler.cpp +++ b/ngraph/frontend/onnx_import/src/op/image_scaler.cpp @@ -47,10 +47,10 @@ namespace ngraph data_shape[1].get_length()); const auto scale_const = - default_opset::Constant::create(element::f32, Shape{}, {scale}); + default_opset::Constant::create(data.get_element_type(), Shape{}, {scale}); - const auto bias_const = - default_opset::Constant::create(element::f32, {1, bias.size(), 1, 1}, bias); + const auto bias_const = default_opset::Constant::create( + data.get_element_type(), {1, bias.size(), 1, 1}, bias); const auto scaler = std::make_shared( std::make_shared(data, scale_const), bias_const);