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

fix Round tests (#6294)

Browse Source 
issue:test_mode always use one type

Signed-off-by: Hu, Yuan2 <yuan2.hu@intel.com>
This commit is contained in:
Yuan Hu 2021-06-23 17:37:28 +08:00 committed by GitHub
parent 3307b1f17c
commit 240e195025
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
ngraph/test/visitors/op/round.cpp
View File

@ -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());