diff --git a/inference-engine/tests/functional/inference_engine/transformations/convert_precision.cpp b/inference-engine/tests/functional/inference_engine/transformations/convert_precision.cpp index 1bbe676543c..a52b677037d 100644 --- a/inference-engine/tests/functional/inference_engine/transformations/convert_precision.cpp +++ b/inference-engine/tests/functional/inference_engine/transformations/convert_precision.cpp @@ -119,6 +119,29 @@ TEST(TransformationTests, ConvertPrecision_ShapeOf) { ASSERT_FALSE(has_type(f)); } +TEST(TransformationTests, ConvertPrecision_ConstantRelu) { + std::shared_ptr f(nullptr); + { + auto input = opset4::Constant::create(element::f16, Shape{1, 1000, 4}, {0}); + auto relu1 = std::make_shared(input); + auto relu2 = std::make_shared(relu1); + + f = std::make_shared(NodeVector{relu2}, ParameterVector{}); + + pass::Manager manager; + + static const precisions_array precisions = { + { ngraph::element::f16, ngraph::element::f32 } + }; + + manager.register_pass(precisions); + manager.run_passes(f); + } + + ASSERT_FALSE(has_type(f)); + ASSERT_FALSE(has_type(f)); +} + TEST(TransformationTests, ConvertPrecision_Convert) { std::shared_ptr f(nullptr); { diff --git a/ngraph/core/src/pass/convert_precision.cpp b/ngraph/core/src/pass/convert_precision.cpp index 8fdef4c9356..c5bfe5e9589 100644 --- a/ngraph/core/src/pass/convert_precision.cpp +++ b/ngraph/core/src/pass/convert_precision.cpp @@ -917,6 +917,7 @@ bool fuse_type_to_constant(const std::shared_ptr& node, new_const->validate_and_infer_types(); new_const->set_friendly_name(constant->get_friendly_name()); + return true; } return false; }