diff --git a/ngraph/test/visitors/op/round.cpp b/ngraph/test/visitors/op/round.cpp index 159d07edd33..f99e3925297 100644 --- a/ngraph/test/visitors/op/round.cpp +++ b/ngraph/test/visitors/op/round.cpp @@ -22,7 +22,7 @@ void static test_mode(opset5::Round::RoundMode mode) { NodeBuilder::get_ops().register_factory(); auto data = make_shared(element::f32, Shape{200}); - auto round = make_shared(data, opset5::Round::RoundMode::HALF_TO_EVEN); + auto round = make_shared(data, mode); NodeBuilder builder(round); auto g_round = as_type_ptr(builder.create());