From 240e1950258d4c8fdd2743d8aafb7687f8123385 Mon Sep 17 00:00:00 2001
From: Yuan Hu <yuan2.hu@intel.com>
Date: Wed, 23 Jun 2021 17:37:28 +0800
Subject: [PATCH] fix Round tests (#6294)

issue:test_mode always use one type

Signed-off-by: Hu, Yuan2 <yuan2.hu@intel.com>
---
 ngraph/test/visitors/op/round.cpp | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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<opset5::Round>();
     auto data = make_shared<op::Parameter>(element::f32, Shape{200});
-    auto round = make_shared<opset5::Round>(data, opset5::Round::RoundMode::HALF_TO_EVEN);
+    auto round = make_shared<opset5::Round>(data, mode);
     NodeBuilder builder(round);
     auto g_round = as_type_ptr<opset5::Round>(builder.create());