diff --git a/ngraph/test/visitors/op/swish.cpp b/ngraph/test/visitors/op/swish.cpp index a87d90f4a3f5c6..0cb39f353cddaf 100644 --- a/ngraph/test/visitors/op/swish.cpp +++ b/ngraph/test/visitors/op/swish.cpp @@ -1,26 +1,11 @@ -// Copyright (C) 2018-2021 Intel Corporation +// Copyright (C) 2021 Intel Corporation // SPDX-License-Identifier: Apache-2.0 // +#include "unary_ops.hpp" -#include "gtest/gtest.h" +using Type = ::testing::Types>; -#include "ngraph/ngraph.hpp" -#include "ngraph/op/util/attr_types.hpp" -#include "ngraph/opsets/opset4.hpp" -#include "util/visitor.hpp" - -using namespace std; -using namespace ngraph; -using ngraph::test::NodeBuilder; - -TEST(attributes, swish_op) -{ - NodeBuilder::get_ops().register_factory(); - const auto A = make_shared(element::f32, Shape{5, 2}); - - const auto swish = make_shared(A); - NodeBuilder builder(swish); - - const auto expected_attr_count = 0; - EXPECT_EQ(builder.get_value_map_size(), expected_attr_count); -} +INSTANTIATE_TYPED_TEST_CASE_P(visitor_without_atrribute, + UnaryOperatorVisitor, + Type, + UnaryOperatorTypeName);