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

ShuffleChannels - add to revised ops (#6624)

Browse Source 
* Update opset in attributes test

* Update attributes test input shape

* Add to revised ops
This commit is contained in:
Katarzyna Mitrus
2021-07-15 07:55:29 +02:00
committed by GitHub
parent 23a1d1f574
commit ebb3c8518c
2 changed files with 5 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
ngraph/test/visitors/op/shuffle_channels.cpp
View File

@@ -5,7 +5,7 @@
#include "gtest/gtest.h"
#include "ngraph/ngraph.hpp"
#include "ngraph/opsets/opset1.hpp"
#include "ngraph/opsets/opset3.hpp"
#include "util/visitor.hpp"
@@ -15,11 +15,11 @@ using ngraph::test::NodeBuilder;
TEST(attributes, shuffle_channels_op)
{
using ShuffleChannels = opset1::ShuffleChannels;
using ShuffleChannels = opset3::ShuffleChannels;
NodeBuilder::get_ops().register_factory<ShuffleChannels>();
auto data = make_shared<op::Parameter>(element::i32, Shape{200});
auto axis = 0;
auto data = make_shared<op::Parameter>(element::i32, Shape{2, 64, 16, 16});
auto axis = 1;
auto groups = 2;
auto shuffle_channels = make_shared<ShuffleChannels>(data, axis, groups);
NodeBuilder builder(shuffle_channels);