Move Visitor tests to new api (#19379)

* Moved visitor tests to new API

* Fixed build for Windows
This commit is contained in:
Ilya Churaev 2023-08-25 10:50:59 +04:00 committed by GitHub
parent 350c4d2363
commit 679369c707
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
183 changed files with 2433 additions and 2529 deletions

View File

@ -2,52 +2,45 @@
// SPDX-License-Identifier: Apache-2.0 // SPDX-License-Identifier: Apache-2.0
// //
#include "common_test_utils/visitor.hpp" #include <gtest/gtest.h>
#include "gtest/gtest.h"
#include "ngraph/ngraph.hpp" #include "openvino/core/attribute_visitor.hpp"
#include "ngraph/op/util/attr_types.hpp" #include "visitors/visitors.hpp"
#include "ngraph/opsets/opset1.hpp"
#include "ngraph/opsets/opset3.hpp"
#include "ngraph/opsets/opset4.hpp"
#include "ngraph/opsets/opset5.hpp"
using namespace std; using namespace std;
using namespace ngraph;
using ngraph::test::NodeBuilder;
using ngraph::test::ValueMap;
TEST(attributes, dimension) { TEST(attributes, dimension) {
NodeBuilder builder; ov::test::NodeBuilder builder;
AttributeVisitor& loader = builder.get_node_loader(); ov::AttributeVisitor& loader = builder.get_node_loader();
AttributeVisitor& saver = builder.get_node_saver(); ov::AttributeVisitor& saver = builder.get_node_saver();
Dimension dyn = Dimension(-1); ov::Dimension dyn = ov::Dimension(-1);
saver.on_attribute("dyn", dyn); saver.on_attribute("dyn", dyn);
Dimension g_dyn; ov::Dimension g_dyn;
loader.on_attribute("dyn", g_dyn); loader.on_attribute("dyn", g_dyn);
EXPECT_EQ(dyn, g_dyn); EXPECT_EQ(dyn, g_dyn);
Dimension scalar = Dimension(10); ov::Dimension scalar = ov::Dimension(10);
saver.on_attribute("scalar", scalar); saver.on_attribute("scalar", scalar);
Dimension g_scalar; ov::Dimension g_scalar;
loader.on_attribute("scalar", g_scalar); loader.on_attribute("scalar", g_scalar);
EXPECT_EQ(scalar, g_scalar); EXPECT_EQ(scalar, g_scalar);
Dimension boundaries1 = Dimension(2, 100); ov::Dimension boundaries1 = ov::Dimension(2, 100);
saver.on_attribute("boundaries1", boundaries1); saver.on_attribute("boundaries1", boundaries1);
Dimension g_boundaries1; ov::Dimension g_boundaries1;
loader.on_attribute("boundaries1", g_boundaries1); loader.on_attribute("boundaries1", g_boundaries1);
EXPECT_EQ(boundaries1, g_boundaries1); EXPECT_EQ(boundaries1, g_boundaries1);
Dimension boundaries2 = Dimension(-1, 100); ov::Dimension boundaries2 = ov::Dimension(-1, 100);
saver.on_attribute("boundaries2", boundaries2); saver.on_attribute("boundaries2", boundaries2);
Dimension g_boundaries2; ov::Dimension g_boundaries2;
loader.on_attribute("boundaries2", g_boundaries2); loader.on_attribute("boundaries2", g_boundaries2);
EXPECT_EQ(boundaries2, g_boundaries2); EXPECT_EQ(boundaries2, g_boundaries2);
Dimension boundaries3 = Dimension(5, -1); ov::Dimension boundaries3 = ov::Dimension(5, -1);
saver.on_attribute("boundaries3", boundaries3); saver.on_attribute("boundaries3", boundaries3);
Dimension g_boundaries3; ov::Dimension g_boundaries3;
loader.on_attribute("boundaries3", g_boundaries3); loader.on_attribute("boundaries3", g_boundaries3);
EXPECT_EQ(boundaries3, g_boundaries3); EXPECT_EQ(boundaries3, g_boundaries3);
} }

View File

@ -2,8 +2,10 @@
// SPDX-License-Identifier: Apache-2.0 // SPDX-License-Identifier: Apache-2.0
// //
#include "openvino/op/abs.hpp"
#include "unary_ops.hpp" #include "unary_ops.hpp"
using Type = ::testing::Types<UnaryOperatorType<ov::op::v0::Abs, ngraph::element::f32>>; using Type = ::testing::Types<UnaryOperatorType<ov::op::v0::Abs, ov::element::f32>>;
INSTANTIATE_TYPED_TEST_SUITE_P(visitor_without_attribute, UnaryOperatorVisitor, Type, UnaryOperatorTypeName); INSTANTIATE_TYPED_TEST_SUITE_P(visitor_without_attribute, UnaryOperatorVisitor, Type, UnaryOperatorTypeName);

View File

@ -2,8 +2,10 @@
// SPDX-License-Identifier: Apache-2.0 // SPDX-License-Identifier: Apache-2.0
// //
#include "openvino/op/acos.hpp"
#include "unary_ops.hpp" #include "unary_ops.hpp"
using Type = ::testing::Types<UnaryOperatorType<ov::op::v0::Acos, ngraph::element::f32>>; using Type = ::testing::Types<UnaryOperatorType<ov::op::v0::Acos, ov::element::f32>>;
INSTANTIATE_TYPED_TEST_SUITE_P(visitor_without_attribute, UnaryOperatorVisitor, Type, UnaryOperatorTypeName); INSTANTIATE_TYPED_TEST_SUITE_P(visitor_without_attribute, UnaryOperatorVisitor, Type, UnaryOperatorTypeName);

View File

@ -2,8 +2,10 @@
// SPDX-License-Identifier: Apache-2.0 // SPDX-License-Identifier: Apache-2.0
// //
#include "openvino/op/acosh.hpp"
#include "unary_ops.hpp" #include "unary_ops.hpp"
using Type = ::testing::Types<UnaryOperatorType<ov::op::v3::Acosh, ngraph::element::f32>>; using Type = ::testing::Types<UnaryOperatorType<ov::op::v3::Acosh, ov::element::f32>>;
INSTANTIATE_TYPED_TEST_SUITE_P(visitor_without_attribute, UnaryOperatorVisitor, Type, UnaryOperatorTypeName); INSTANTIATE_TYPED_TEST_SUITE_P(visitor_without_attribute, UnaryOperatorVisitor, Type, UnaryOperatorTypeName);

View File

@ -2,23 +2,21 @@
// SPDX-License-Identifier: Apache-2.0 // SPDX-License-Identifier: Apache-2.0
// //
#include "common_test_utils/visitor.hpp" #include "openvino/op/adaptive_avg_pool.hpp"
#include "gtest/gtest.h"
#include "ngraph/ngraph.hpp"
#include "ngraph/op/util/attr_types.hpp"
#include "ngraph/opsets/opset8.hpp"
using namespace std; #include <gtest/gtest.h>
using namespace ngraph;
using ngraph::test::NodeBuilder; #include "openvino/op/constant.hpp"
#include "openvino/op/parameter.hpp"
#include "visitors/visitors.hpp"
TEST(attributes, adaptive_avg_pool_op) { TEST(attributes, adaptive_avg_pool_op) {
NodeBuilder::get_ops().register_factory<opset8::AdaptiveAvgPool>(); ov::test::NodeBuilder::get_ops().register_factory<ov::op::v8::AdaptiveAvgPool>();
const auto A = make_shared<op::Parameter>(element::f32, Shape{1, 3, 5, 4}); const auto A = std::make_shared<ov::op::v0::Parameter>(ov::element::f32, ov::Shape{1, 3, 5, 4});
const auto out_shape = op::Constant::create<int64_t>(element::i64, Shape{2}, {4, 3}); const auto out_shape = ov::op::v0::Constant::create<int64_t>(ov::element::i64, ov::Shape{2}, {4, 3});
const auto adaptive_pool = make_shared<opset8::AdaptiveAvgPool>(A, out_shape); const auto adaptive_pool = std::make_shared<ov::op::v8::AdaptiveAvgPool>(A, out_shape);
NodeBuilder builder(adaptive_pool, {A, out_shape}); ov::test::NodeBuilder builder(adaptive_pool, {A, out_shape});
const auto expected_attr_count = 0; const auto expected_attr_count = 0;
EXPECT_EQ(builder.get_value_map_size(), expected_attr_count); EXPECT_EQ(builder.get_value_map_size(), expected_attr_count);

View File

@ -2,24 +2,22 @@
// SPDX-License-Identifier: Apache-2.0 // SPDX-License-Identifier: Apache-2.0
// //
#include "common_test_utils/visitor.hpp" #include "openvino/op/adaptive_max_pool.hpp"
#include "gtest/gtest.h"
#include "ngraph/ngraph.hpp"
#include "ngraph/op/util/attr_types.hpp"
#include "ngraph/opsets/opset8.hpp"
using namespace std; #include <gtest/gtest.h>
using namespace ngraph;
using ngraph::test::NodeBuilder; #include "openvino/op/constant.hpp"
#include "openvino/op/parameter.hpp"
#include "visitors/visitors.hpp"
TEST(attributes, adaptive_max_pool_op) { TEST(attributes, adaptive_max_pool_op) {
NodeBuilder::get_ops().register_factory<opset8::AdaptiveMaxPool>(); ov::test::NodeBuilder::get_ops().register_factory<ov::op::v8::AdaptiveMaxPool>();
const auto A = make_shared<op::Parameter>(element::f32, Shape{1, 3, 5, 4}); const auto A = std::make_shared<ov::op::v0::Parameter>(ov::element::f32, ov::Shape{1, 3, 5, 4});
const auto out_shape = op::Constant::create<int64_t>(element::i64, Shape{2}, {4, 3}); const auto out_shape = ov::op::v0::Constant::create<int64_t>(ov::element::i64, ov::Shape{2}, {4, 3});
const auto adaptive_pool = make_shared<opset8::AdaptiveMaxPool>(A, out_shape); const auto adaptive_pool = std::make_shared<ov::op::v8::AdaptiveMaxPool>(A, out_shape);
NodeBuilder builder(adaptive_pool, {A, out_shape}); ov::test::NodeBuilder builder(adaptive_pool, {A, out_shape});
auto g_adaptive_pool = ov::as_type_ptr<opset8::AdaptiveMaxPool>(builder.create()); auto g_adaptive_pool = ov::as_type_ptr<ov::op::v8::AdaptiveMaxPool>(builder.create());
const auto expected_attr_count = 1; const auto expected_attr_count = 1;
EXPECT_EQ(builder.get_value_map_size(), expected_attr_count); EXPECT_EQ(builder.get_value_map_size(), expected_attr_count);

View File

@ -2,9 +2,10 @@
// SPDX-License-Identifier: Apache-2.0 // SPDX-License-Identifier: Apache-2.0
// //
#include "binary_ops.hpp" #include "openvino/op/add.hpp"
#include "ngraph/opsets/opset1.hpp"
using Type = ::testing::Types<BinaryOperatorType<ov::op::v1::Add, ngraph::element::f32>>; #include "binary_ops.hpp"
using Type = ::testing::Types<BinaryOperatorType<ov::op::v1::Add, ov::element::f32>>;
INSTANTIATE_TYPED_TEST_SUITE_P(visitor_with_auto_broadcast, BinaryOperatorVisitor, Type, BinaryOperatorTypeName); INSTANTIATE_TYPED_TEST_SUITE_P(visitor_with_auto_broadcast, BinaryOperatorVisitor, Type, BinaryOperatorTypeName);

View File

@ -2,7 +2,10 @@
// SPDX-License-Identifier: Apache-2.0 // SPDX-License-Identifier: Apache-2.0
// //
#include "openvino/op/asin.hpp"
#include "unary_ops.hpp" #include "unary_ops.hpp"
using Type = ::testing::Types<UnaryOperatorType<ov::op::v0::Asin, ngraph::element::f32>>;
using Type = ::testing::Types<UnaryOperatorType<ov::op::v0::Asin, ov::element::f32>>;
INSTANTIATE_TYPED_TEST_SUITE_P(visitor_without_attribute, UnaryOperatorVisitor, Type, UnaryOperatorTypeName); INSTANTIATE_TYPED_TEST_SUITE_P(visitor_without_attribute, UnaryOperatorVisitor, Type, UnaryOperatorTypeName);

View File

@ -2,7 +2,10 @@
// SPDX-License-Identifier: Apache-2.0 // SPDX-License-Identifier: Apache-2.0
// //
#include "openvino/op/asinh.hpp"
#include "unary_ops.hpp" #include "unary_ops.hpp"
using Type = ::testing::Types<UnaryOperatorType<ov::op::v3::Asinh, ngraph::element::f32>>;
using Type = ::testing::Types<UnaryOperatorType<ov::op::v3::Asinh, ov::element::f32>>;
INSTANTIATE_TYPED_TEST_SUITE_P(visitor_without_attribute, UnaryOperatorVisitor, Type, UnaryOperatorTypeName); INSTANTIATE_TYPED_TEST_SUITE_P(visitor_without_attribute, UnaryOperatorVisitor, Type, UnaryOperatorTypeName);

View File

@ -2,25 +2,19 @@
// SPDX-License-Identifier: Apache-2.0 // SPDX-License-Identifier: Apache-2.0
// //
#include "common_test_utils/visitor.hpp" #include <gtest/gtest.h>
#include "gtest/gtest.h"
#include "ngraph/ngraph.hpp"
#include "ngraph/op/util/attr_types.hpp"
#include "ngraph/opsets/opset3.hpp"
#include "ngraph/opsets/opset6.hpp"
using namespace std; #include "openvino/op/read_value.hpp"
using namespace ngraph; #include "openvino/op/util/variable.hpp"
using ngraph::test::NodeBuilder; #include "visitors/visitors.hpp"
using ngraph::test::ValueMap;
TEST(attributes, assign_v3_op) { TEST(attributes, assign_v3_op) {
NodeBuilder::get_ops().register_factory<opset3::Assign>(); ov::test::NodeBuilder::get_ops().register_factory<ov::op::v3::Assign>();
const auto in = make_shared<op::Parameter>(element::f32, Shape{1}); const auto in = std::make_shared<ov::op::v0::Parameter>(ov::element::f32, ov::Shape{1});
const string variable_id = "v0"; const std::string variable_id = "v0";
const auto read_value = make_shared<opset3::ReadValue>(in, variable_id); const auto read_value = std::make_shared<ov::op::v3::ReadValue>(in, variable_id);
const auto assign = make_shared<opset3::Assign>(read_value, variable_id); const auto assign = std::make_shared<ov::op::v3::Assign>(read_value, variable_id);
NodeBuilder builder(assign, {read_value}); ov::test::NodeBuilder builder(assign, {read_value});
// attribute count // attribute count
const auto expected_attr_count = 1; const auto expected_attr_count = 1;
@ -28,12 +22,13 @@ TEST(attributes, assign_v3_op) {
} }
TEST(attributes, assign_v6_op) { TEST(attributes, assign_v6_op) {
NodeBuilder::get_ops().register_factory<opset6::Assign>(); ov::test::NodeBuilder::get_ops().register_factory<ov::op::v6::Assign>();
const auto in = make_shared<op::Parameter>(element::f32, Shape{1}); const auto in = std::make_shared<ov::op::v0::Parameter>(ov::element::f32, ov::Shape{1});
const auto variable = std::make_shared<Variable>(VariableInfo{PartialShape::dynamic(), element::dynamic, "v0"}); const auto variable = std::make_shared<ov::op::util::Variable>(
const auto read_value = make_shared<opset6::ReadValue>(in, variable); ov::op::util::VariableInfo{ov::PartialShape::dynamic(), ov::element::dynamic, "v0"});
const auto assign = make_shared<opset6::Assign>(read_value, variable); const auto read_value = std::make_shared<ov::op::v6::ReadValue>(in, variable);
NodeBuilder builder(assign, {read_value}); const auto assign = std::make_shared<ov::op::v6::Assign>(read_value, variable);
ov::test::NodeBuilder builder(assign, {read_value});
// attribute count // attribute count
const auto expected_attr_count = 1; const auto expected_attr_count = 1;

View File

@ -2,8 +2,10 @@
// SPDX-License-Identifier: Apache-2.0 // SPDX-License-Identifier: Apache-2.0
// //
#include "openvino/op/atan.hpp"
#include "unary_ops.hpp" #include "unary_ops.hpp"
using Types = ::testing::Types<UnaryOperatorType<ov::op::v0::Atan, ngraph::element::f32>>; using Types = ::testing::Types<UnaryOperatorType<ov::op::v0::Atan, ov::element::f32>>;
INSTANTIATE_TYPED_TEST_SUITE_P(visitor_without_attribute, UnaryOperatorVisitor, Types, UnaryOperatorTypeName); INSTANTIATE_TYPED_TEST_SUITE_P(visitor_without_attribute, UnaryOperatorVisitor, Types, UnaryOperatorTypeName);

View File

@ -2,8 +2,10 @@
// SPDX-License-Identifier: Apache-2.0 // SPDX-License-Identifier: Apache-2.0
// //
#include "openvino/op/atanh.hpp"
#include "unary_ops.hpp" #include "unary_ops.hpp"
using Types = ::testing::Types<UnaryOperatorType<ov::op::v3::Atanh, ngraph::element::f32>>; using Types = ::testing::Types<UnaryOperatorType<ov::op::v3::Atanh, ov::element::f32>>;
INSTANTIATE_TYPED_TEST_SUITE_P(visitor_without_attribute, UnaryOperatorVisitor, Types, UnaryOperatorTypeName); INSTANTIATE_TYPED_TEST_SUITE_P(visitor_without_attribute, UnaryOperatorVisitor, Types, UnaryOperatorTypeName);

View File

@ -2,21 +2,19 @@
// SPDX-License-Identifier: Apache-2.0 // SPDX-License-Identifier: Apache-2.0
// //
#include "common_test_utils/visitor.hpp" #include "openvino/op/avg_pool.hpp"
#include "gtest/gtest.h"
#include "ngraph/ngraph.hpp" #include <gtest/gtest.h>
#include "ngraph/op/util/attr_types.hpp"
#include "ngraph/opsets/opset1.hpp" #include "openvino/op/parameter.hpp"
#include "ngraph/opsets/opset8.hpp" #include "visitors/visitors.hpp"
using namespace std; using namespace std;
using namespace ngraph; using namespace ov;
using ngraph::test::NodeBuilder;
using ngraph::test::ValueMap;
TEST(attributes, avg_pool_op) { TEST(attributes, avg_pool_op) {
NodeBuilder::get_ops().register_factory<opset1::AvgPool>(); ov::test::NodeBuilder::get_ops().register_factory<op::v1::AvgPool>();
auto data = make_shared<op::Parameter>(element::f32, Shape{64, 3, 5}); auto data = make_shared<op::v0::Parameter>(element::f32, Shape{64, 3, 5});
auto strides = Strides{2}; auto strides = Strides{2};
auto pads_begin = Shape{1}; auto pads_begin = Shape{1};
@ -27,13 +25,13 @@ TEST(attributes, avg_pool_op) {
auto auto_pad = op::PadType::EXPLICIT; auto auto_pad = op::PadType::EXPLICIT;
auto avg_pool = auto avg_pool =
make_shared<opset1::AvgPool>(data, strides, pads_begin, pads_end, kernel, exclude_pad, rounding_mode, auto_pad); make_shared<op::v1::AvgPool>(data, strides, pads_begin, pads_end, kernel, exclude_pad, rounding_mode, auto_pad);
avg_pool->set_pads_begin(pads_begin); avg_pool->set_pads_begin(pads_begin);
avg_pool->set_pads_end(pads_end); avg_pool->set_pads_end(pads_end);
NodeBuilder builder(avg_pool, {data}); ov::test::NodeBuilder builder(avg_pool, {data});
auto g_avg_pool = ov::as_type_ptr<opset1::AvgPool>(builder.create()); auto g_avg_pool = ov::as_type_ptr<op::v1::AvgPool>(builder.create());
EXPECT_EQ(g_avg_pool->get_strides(), avg_pool->get_strides()); EXPECT_EQ(g_avg_pool->get_strides(), avg_pool->get_strides());
EXPECT_EQ(g_avg_pool->get_pads_begin(), avg_pool->get_pads_begin()); EXPECT_EQ(g_avg_pool->get_pads_begin(), avg_pool->get_pads_begin());
@ -44,8 +42,8 @@ TEST(attributes, avg_pool_op) {
} }
TEST(attributes, avg_pool_op_valid) { TEST(attributes, avg_pool_op_valid) {
NodeBuilder::get_ops().register_factory<opset1::AvgPool>(); ov::test::NodeBuilder::get_ops().register_factory<op::v1::AvgPool>();
auto data = make_shared<op::Parameter>(element::f32, Shape{64, 3, 5}); auto data = make_shared<op::v0::Parameter>(element::f32, Shape{64, 3, 5});
auto strides = Strides{2}; auto strides = Strides{2};
auto pads_begin = Shape{1}; auto pads_begin = Shape{1};
@ -56,10 +54,10 @@ TEST(attributes, avg_pool_op_valid) {
auto auto_pad = op::PadType::VALID; auto auto_pad = op::PadType::VALID;
auto avg_pool = auto avg_pool =
make_shared<opset1::AvgPool>(data, strides, pads_begin, pads_end, kernel, exclude_pad, rounding_mode, auto_pad); make_shared<op::v1::AvgPool>(data, strides, pads_begin, pads_end, kernel, exclude_pad, rounding_mode, auto_pad);
NodeBuilder builder(avg_pool, {data}); ov::test::NodeBuilder builder(avg_pool, {data});
auto g_avg_pool = ov::as_type_ptr<opset1::AvgPool>(builder.create()); auto g_avg_pool = ov::as_type_ptr<op::v1::AvgPool>(builder.create());
EXPECT_EQ(g_avg_pool->get_strides(), avg_pool->get_strides()); EXPECT_EQ(g_avg_pool->get_strides(), avg_pool->get_strides());
EXPECT_EQ(g_avg_pool->get_pads_begin(), avg_pool->get_pads_begin()); EXPECT_EQ(g_avg_pool->get_pads_begin(), avg_pool->get_pads_begin());
@ -70,8 +68,8 @@ TEST(attributes, avg_pool_op_valid) {
} }
TEST(attributes, avg_pool_v8_op) { TEST(attributes, avg_pool_v8_op) {
NodeBuilder::get_ops().register_factory<opset8::AvgPool>(); ov::test::NodeBuilder::get_ops().register_factory<op::v1::AvgPool>();
const auto data = make_shared<op::Parameter>(element::i32, Shape{1, 3, 37, 37}); const auto data = make_shared<op::v0::Parameter>(element::i32, Shape{1, 3, 37, 37});
const auto strides = Strides{1, 1}; const auto strides = Strides{1, 1};
const auto pads_begin = Shape{1, 1}; const auto pads_begin = Shape{1, 1};
@ -82,9 +80,9 @@ TEST(attributes, avg_pool_v8_op) {
const auto auto_pad = op::PadType::EXPLICIT; const auto auto_pad = op::PadType::EXPLICIT;
const auto avg_pool = const auto avg_pool =
make_shared<opset8::AvgPool>(data, strides, pads_begin, pads_end, kernel, exclude_pad, rounding_mode, auto_pad); make_shared<op::v1::AvgPool>(data, strides, pads_begin, pads_end, kernel, exclude_pad, rounding_mode, auto_pad);
NodeBuilder builder(avg_pool, {data}); ov::test::NodeBuilder builder(avg_pool, {data});
auto g_avg_pool = ov::as_type_ptr<opset8::AvgPool>(builder.create()); auto g_avg_pool = ov::as_type_ptr<op::v1::AvgPool>(builder.create());
EXPECT_EQ(g_avg_pool->get_strides(), avg_pool->get_strides()); EXPECT_EQ(g_avg_pool->get_strides(), avg_pool->get_strides());
EXPECT_EQ(g_avg_pool->get_pads_begin(), avg_pool->get_pads_begin()); EXPECT_EQ(g_avg_pool->get_pads_begin(), avg_pool->get_pads_begin());

View File

@ -2,19 +2,17 @@
// SPDX-License-Identifier: Apache-2.0 // SPDX-License-Identifier: Apache-2.0
// //
#include "common_test_utils/visitor.hpp" #include "openvino/op/batch_norm.hpp"
#include "gtest/gtest.h"
#include "ngraph/ngraph.hpp" #include <gtest/gtest.h>
#include "ngraph/op/util/attr_types.hpp"
#include "ngraph/opsets/opset1.hpp" #include "openvino/core/partial_shape.hpp"
#include "ngraph/opsets/opset3.hpp" #include "openvino/core/type/element_type.hpp"
#include "ngraph/opsets/opset4.hpp" #include "openvino/op/parameter.hpp"
#include "ngraph/opsets/opset5.hpp" #include "visitors/visitors.hpp"
using namespace std; using namespace std;
using namespace ngraph; using namespace ov;
using ngraph::test::NodeBuilder;
using ngraph::test::ValueMap;
TEST(attributes, batch_norm_inference_op_v5) { TEST(attributes, batch_norm_inference_op_v5) {
PartialShape in_shape{1, 10}; PartialShape in_shape{1, 10};
@ -22,16 +20,16 @@ TEST(attributes, batch_norm_inference_op_v5) {
element::Type et = element::f32; element::Type et = element::f32;
double epsilon = 0.001; double epsilon = 0.001;
NodeBuilder::get_ops().register_factory<op::v5::BatchNormInference>(); test::NodeBuilder::get_ops().register_factory<op::v5::BatchNormInference>();
auto data_batch = make_shared<op::Parameter>(et, in_shape); auto data_batch = make_shared<ov::op::v0::Parameter>(et, in_shape);
auto gamma = make_shared<op::Parameter>(et, ch_shape); auto gamma = make_shared<ov::op::v0::Parameter>(et, ch_shape);
auto beta = make_shared<op::Parameter>(et, ch_shape); auto beta = make_shared<ov::op::v0::Parameter>(et, ch_shape);
auto mean = make_shared<op::Parameter>(et, ch_shape); auto mean = make_shared<ov::op::v0::Parameter>(et, ch_shape);
auto var = make_shared<op::Parameter>(et, ch_shape); auto var = make_shared<ov::op::v0::Parameter>(et, ch_shape);
auto batch_norm = make_shared<op::v5::BatchNormInference>(data_batch, gamma, beta, mean, var, epsilon); auto batch_norm = make_shared<op::v5::BatchNormInference>(data_batch, gamma, beta, mean, var, epsilon);
const auto expected_attr_count = 1; const auto expected_attr_count = 1;
NodeBuilder builder(batch_norm, {data_batch, gamma, beta, mean, var}); test::NodeBuilder builder(batch_norm, {data_batch, gamma, beta, mean, var});
EXPECT_EQ(builder.get_value_map_size(), expected_attr_count); EXPECT_EQ(builder.get_value_map_size(), expected_attr_count);
auto g_batch_norm = ov::as_type_ptr<op::v5::BatchNormInference>(builder.create()); auto g_batch_norm = ov::as_type_ptr<op::v5::BatchNormInference>(builder.create());
EXPECT_EQ(g_batch_norm->get_eps_value(), batch_norm->get_eps_value()); EXPECT_EQ(g_batch_norm->get_eps_value(), batch_norm->get_eps_value());
@ -43,16 +41,16 @@ TEST(attributes, batch_norm_inference_op_v0) {
element::Type et = element::f32; element::Type et = element::f32;
double epsilon = 0.001; double epsilon = 0.001;
NodeBuilder::get_ops().register_factory<op::v0::BatchNormInference>(); test::NodeBuilder::get_ops().register_factory<op::v0::BatchNormInference>();
auto data_batch = make_shared<op::Parameter>(et, in_shape); auto data_batch = make_shared<ov::op::v0::Parameter>(et, in_shape);
auto gamma = make_shared<op::Parameter>(et, ch_shape); auto gamma = make_shared<ov::op::v0::Parameter>(et, ch_shape);
auto beta = make_shared<op::Parameter>(et, ch_shape); auto beta = make_shared<ov::op::v0::Parameter>(et, ch_shape);
auto mean = make_shared<op::Parameter>(et, ch_shape); auto mean = make_shared<ov::op::v0::Parameter>(et, ch_shape);
auto var = make_shared<op::Parameter>(et, ch_shape); auto var = make_shared<ov::op::v0::Parameter>(et, ch_shape);
auto batch_norm = make_shared<op::v0::BatchNormInference>(data_batch, gamma, beta, mean, var, epsilon); auto batch_norm = make_shared<op::v0::BatchNormInference>(data_batch, gamma, beta, mean, var, epsilon);
const auto expected_attr_count = 1; const auto expected_attr_count = 1;
NodeBuilder builder(batch_norm, {gamma, beta, data_batch, mean, var}); test::NodeBuilder builder(batch_norm, {gamma, beta, data_batch, mean, var});
EXPECT_EQ(builder.get_value_map_size(), expected_attr_count); EXPECT_EQ(builder.get_value_map_size(), expected_attr_count);
auto g_batch_norm = ov::as_type_ptr<op::v0::BatchNormInference>(builder.create()); auto g_batch_norm = ov::as_type_ptr<op::v0::BatchNormInference>(builder.create());
EXPECT_EQ(g_batch_norm->get_eps_value(), batch_norm->get_eps_value()); EXPECT_EQ(g_batch_norm->get_eps_value(), batch_norm->get_eps_value());

View File

@ -2,24 +2,26 @@
// SPDX-License-Identifier: Apache-2.0 // SPDX-License-Identifier: Apache-2.0
// //
#include "common_test_utils/visitor.hpp" #include "openvino/op/batch_to_space.hpp"
#include "gtest/gtest.h"
#include "ngraph/ngraph.hpp" #include <gtest/gtest.h>
#include "ngraph/op/util/attr_types.hpp"
#include "openvino/op/constant.hpp"
#include "openvino/op/parameter.hpp"
#include "visitors/visitors.hpp"
using namespace std; using namespace std;
using namespace ngraph; using namespace ov;
using ngraph::test::NodeBuilder;
TEST(attributes, batch_to_space_op) { TEST(attributes, batch_to_space_op) {
NodeBuilder::get_ops().register_factory<op::v1::BatchToSpace>(); test::NodeBuilder::get_ops().register_factory<op::v1::BatchToSpace>();
auto data = make_shared<op::Parameter>(element::f32, Shape{2, 128}); auto data = make_shared<ov::op::v0::Parameter>(element::f32, Shape{2, 128});
auto block_shape = make_shared<op::Constant>(element::i64, Shape{2}, vector<int64_t>{1, 2}); auto block_shape = make_shared<ov::op::v0::Constant>(element::i64, Shape{2}, vector<int64_t>{1, 2});
auto crops_begin = make_shared<op::Constant>(element::i64, Shape{2}, vector<int64_t>{0, 2}); auto crops_begin = make_shared<ov::op::v0::Constant>(element::i64, Shape{2}, vector<int64_t>{0, 2});
auto crops_end = make_shared<op::Constant>(element::i64, Shape{2}, vector<int64_t>{0, 0}); auto crops_end = make_shared<ov::op::v0::Constant>(element::i64, Shape{2}, vector<int64_t>{0, 0});
auto batch2space = make_shared<op::v1::BatchToSpace>(data, block_shape, crops_begin, crops_end); auto batch2space = make_shared<op::v1::BatchToSpace>(data, block_shape, crops_begin, crops_end);
NodeBuilder builder(batch2space, {data}); test::NodeBuilder builder(batch2space, {data});
const auto expected_attr_count = 0; const auto expected_attr_count = 0;
EXPECT_EQ(builder.get_value_map_size(), expected_attr_count); EXPECT_EQ(builder.get_value_map_size(), expected_attr_count);

View File

@ -2,19 +2,16 @@
// SPDX-License-Identifier: Apache-2.0 // SPDX-License-Identifier: Apache-2.0
// //
#include "common_test_utils/visitor.hpp" #include "openvino/op/binary_convolution.hpp"
#include "gtest/gtest.h"
#include "ngraph/ngraph.hpp" #include <gtest/gtest.h>
#include "ngraph/op/util/attr_types.hpp"
#include "ngraph/opsets/opset1.hpp" #include "openvino/op/convolution.hpp"
#include "ngraph/opsets/opset3.hpp" #include "visitors/visitors.hpp"
#include "ngraph/opsets/opset4.hpp"
#include "ngraph/opsets/opset5.hpp"
using namespace std; using namespace std;
using namespace ngraph; using namespace ov;
using ngraph::test::NodeBuilder; using ov::test::NodeBuilder;
using ngraph::test::ValueMap;
TEST(attributes, bin_convolution) { TEST(attributes, bin_convolution) {
NodeBuilder::get_ops().register_factory<op::v1::Convolution>(); NodeBuilder::get_ops().register_factory<op::v1::Convolution>();
@ -28,8 +25,8 @@ TEST(attributes, bin_convolution) {
const float pad_value = 1.0f; const float pad_value = 1.0f;
const auto auto_pad = op::PadType::SAME_LOWER; const auto auto_pad = op::PadType::SAME_LOWER;
auto data_batch = make_shared<op::Parameter>(element::f32, data_batch_shape); auto data_batch = make_shared<ov::op::v0::Parameter>(element::f32, data_batch_shape);
auto filters = make_shared<op::Parameter>(element::u1, filters_shape); auto filters = make_shared<ov::op::v0::Parameter>(element::u1, filters_shape);
auto conv = make_shared<op::v1::BinaryConvolution>(data_batch, auto conv = make_shared<op::v1::BinaryConvolution>(data_batch,
filters, filters,

View File

@ -4,14 +4,15 @@
#pragma once #pragma once
#include "common_test_utils/visitor.hpp" #include <gtest/gtest.h>
#include "gtest/gtest.h"
template <typename T, ngraph::element::Type_t ELEMENT_TYPE> #include "visitors/visitors.hpp"
template <typename T, ov::element::Type_t ELEMENT_TYPE>
class BinaryOperatorType { class BinaryOperatorType {
public: public:
using op_type = T; using op_type = T;
static constexpr ngraph::element::Type_t element_type = ELEMENT_TYPE; static constexpr ov::element::Type_t element_type = ELEMENT_TYPE;
}; };
template <typename T> template <typename T>
@ -22,8 +23,8 @@ public:
template <typename T> template <typename T>
static std::string GetName(int) { static std::string GetName(int) {
using OP_Type = typename T::op_type; using OP_Type = typename T::op_type;
constexpr ngraph::element::Type precision(T::element_type); constexpr ov::element::Type precision(T::element_type);
const ngraph::Node::type_info_t typeinfo = OP_Type::get_type_info_static(); const ov::Node::type_info_t typeinfo = OP_Type::get_type_info_static();
return std::string{typeinfo.name} + "_" + precision.get_type_name(); return std::string{typeinfo.name} + "_" + precision.get_type_name();
} }
}; };
@ -32,17 +33,17 @@ TYPED_TEST_SUITE_P(BinaryOperatorVisitor);
TYPED_TEST_P(BinaryOperatorVisitor, Auto_Broadcast) { TYPED_TEST_P(BinaryOperatorVisitor, Auto_Broadcast) {
using OP_Type = typename TypeParam::op_type; using OP_Type = typename TypeParam::op_type;
const ngraph::element::Type_t element_type = TypeParam::element_type; const ov::element::Type_t element_type = TypeParam::element_type;
ngraph::test::NodeBuilder::get_ops().register_factory<OP_Type>(); ov::test::NodeBuilder::get_ops().register_factory<OP_Type>();
const auto A = std::make_shared<ngraph::op::Parameter>(element_type, ngraph::PartialShape{1, 2, 3}); const auto A = std::make_shared<ov::op::v0::Parameter>(element_type, ov::PartialShape{1, 2, 3});
const auto B = std::make_shared<ngraph::op::Parameter>(element_type, ngraph::PartialShape{3, 2, 1}); const auto B = std::make_shared<ov::op::v0::Parameter>(element_type, ov::PartialShape{3, 2, 1});
auto auto_broadcast = ngraph::op::AutoBroadcastType::NUMPY; auto auto_broadcast = ov::op::AutoBroadcastType::NUMPY;
const auto op_func = std::make_shared<OP_Type>(A, B, auto_broadcast); const auto op_func = std::make_shared<OP_Type>(A, B, auto_broadcast);
ngraph::test::NodeBuilder builder(op_func, {A, B}); ov::test::NodeBuilder builder(op_func, {A, B});
const auto g_op_func = ngraph::as_type_ptr<OP_Type>(builder.create()); const auto g_op_func = ov::as_type_ptr<OP_Type>(builder.create());
const auto expected_attr_count = 1; const auto expected_attr_count = 1;
EXPECT_EQ(builder.get_value_map_size(), expected_attr_count); EXPECT_EQ(builder.get_value_map_size(), expected_attr_count);
@ -51,15 +52,15 @@ TYPED_TEST_P(BinaryOperatorVisitor, Auto_Broadcast) {
TYPED_TEST_P(BinaryOperatorVisitor, No_Broadcast) { TYPED_TEST_P(BinaryOperatorVisitor, No_Broadcast) {
using OP_Type = typename TypeParam::op_type; using OP_Type = typename TypeParam::op_type;
const ngraph::element::Type_t element_type = TypeParam::element_type; const ov::element::Type_t element_type = TypeParam::element_type;
ngraph::test::NodeBuilder::get_ops().register_factory<OP_Type>(); ov::test::NodeBuilder::get_ops().register_factory<OP_Type>();
const auto A = std::make_shared<ngraph::op::Parameter>(element_type, ngraph::PartialShape{1, 2, 3}); const auto A = std::make_shared<ov::op::v0::Parameter>(element_type, ov::PartialShape{1, 2, 3});
const auto B = std::make_shared<ngraph::op::Parameter>(element_type, ngraph::PartialShape{1, 2, 3}); const auto B = std::make_shared<ov::op::v0::Parameter>(element_type, ov::PartialShape{1, 2, 3});
const auto op_func = std::make_shared<OP_Type>(A, B); const auto op_func = std::make_shared<OP_Type>(A, B);
ngraph::test::NodeBuilder builder(op_func, {A, B}); ov::test::NodeBuilder builder(op_func, {A, B});
const auto g_op_func = ngraph::as_type_ptr<OP_Type>(builder.create()); const auto g_op_func = ov::as_type_ptr<OP_Type>(builder.create());
const auto expected_attr_count = 1; const auto expected_attr_count = 1;
EXPECT_EQ(builder.get_value_map_size(), expected_attr_count); EXPECT_EQ(builder.get_value_map_size(), expected_attr_count);

View File

@ -2,55 +2,53 @@
// SPDX-License-Identifier: Apache-2.0 // SPDX-License-Identifier: Apache-2.0
// //
#include "common_test_utils/visitor.hpp" #include "openvino/op/broadcast.hpp"
#include "gtest/gtest.h"
#include "ngraph/ngraph.hpp" #include <gtest/gtest.h>
#include "ngraph/op/util/attr_types.hpp"
#include "ngraph/opsets/opset1.hpp" #include "visitors/visitors.hpp"
#include "ngraph/opsets/opset3.hpp"
using namespace std; using namespace std;
using namespace ngraph; using namespace ov;
using ngraph::test::NodeBuilder; using ov::test::NodeBuilder;
using ngraph::test::ValueMap;
TEST(attributes, broadcast_v1) { TEST(attributes, broadcast_v1) {
NodeBuilder::get_ops().register_factory<opset1::Broadcast>(); NodeBuilder::get_ops().register_factory<ov::op::v1::Broadcast>();
const auto arg = make_shared<op::Parameter>(element::i64, Shape{1, 3, 1}); const auto arg = make_shared<ov::op::v0::Parameter>(element::i64, Shape{1, 3, 1});
const auto shape = make_shared<op::Parameter>(element::i64, Shape{3}); const auto shape = make_shared<ov::op::v0::Parameter>(element::i64, Shape{3});
const auto axes_mapping = make_shared<op::Parameter>(element::i64, Shape{3}); const auto axes_mapping = make_shared<ov::op::v0::Parameter>(element::i64, Shape{3});
const auto broadcast_spec = ov::op::AutoBroadcastType::NONE; const auto broadcast_spec = ov::op::AutoBroadcastType::NONE;
const auto broadcast_v3 = make_shared<op::v1::Broadcast>(arg, shape, axes_mapping, broadcast_spec); const auto broadcast_v3 = make_shared<op::v1::Broadcast>(arg, shape, axes_mapping, broadcast_spec);
NodeBuilder builder(broadcast_v3, {arg, shape, axes_mapping}); NodeBuilder builder(broadcast_v3, {arg, shape, axes_mapping});
auto g_broadcast_v3 = ov::as_type_ptr<opset1::Broadcast>(builder.create()); auto g_broadcast_v3 = ov::as_type_ptr<ov::op::v1::Broadcast>(builder.create());
EXPECT_EQ(g_broadcast_v3->get_broadcast_spec().m_type, broadcast_spec); EXPECT_EQ(g_broadcast_v3->get_broadcast_spec().m_type, broadcast_spec);
} }
TEST(attributes, broadcast_v3) { TEST(attributes, broadcast_v3) {
NodeBuilder::get_ops().register_factory<opset3::Broadcast>(); NodeBuilder::get_ops().register_factory<ov::op::v3::Broadcast>();
const auto arg = make_shared<op::Parameter>(element::i64, Shape{1, 3, 1}); const auto arg = make_shared<ov::op::v0::Parameter>(element::i64, Shape{1, 3, 1});
const auto shape = make_shared<op::Parameter>(element::i64, Shape{3}); const auto shape = make_shared<ov::op::v0::Parameter>(element::i64, Shape{3});
const auto broadcast_spec = op::BroadcastType::BIDIRECTIONAL; const auto broadcast_spec = op::BroadcastType::BIDIRECTIONAL;
const auto broadcast_v3 = make_shared<op::v3::Broadcast>(arg, shape, broadcast_spec); const auto broadcast_v3 = make_shared<op::v3::Broadcast>(arg, shape, broadcast_spec);
NodeBuilder builder(broadcast_v3, {arg, shape}); NodeBuilder builder(broadcast_v3, {arg, shape});
auto g_broadcast_v3 = ov::as_type_ptr<opset3::Broadcast>(builder.create()); auto g_broadcast_v3 = ov::as_type_ptr<ov::op::v3::Broadcast>(builder.create());
EXPECT_EQ(g_broadcast_v3->get_broadcast_spec(), broadcast_spec); EXPECT_EQ(g_broadcast_v3->get_broadcast_spec(), broadcast_spec);
} }
TEST(attributes, broadcast_v3_explicit) { TEST(attributes, broadcast_v3_explicit) {
NodeBuilder::get_ops().register_factory<opset3::Broadcast>(); NodeBuilder::get_ops().register_factory<ov::op::v3::Broadcast>();
const auto arg = make_shared<op::Parameter>(element::i64, Shape{1, 3, 1}); const auto arg = make_shared<ov::op::v0::Parameter>(element::i64, Shape{1, 3, 1});
const auto shape = make_shared<op::Parameter>(element::i64, Shape{3}); const auto shape = make_shared<ov::op::v0::Parameter>(element::i64, Shape{3});
const auto axes_mapping = make_shared<op::Parameter>(element::i64, Shape{3}); const auto axes_mapping = make_shared<ov::op::v0::Parameter>(element::i64, Shape{3});
const auto broadcast_spec = op::BroadcastType::EXPLICIT; const auto broadcast_spec = op::BroadcastType::EXPLICIT;
const auto broadcast_v3 = make_shared<op::v3::Broadcast>(arg, shape, axes_mapping, broadcast_spec); const auto broadcast_v3 = make_shared<op::v3::Broadcast>(arg, shape, axes_mapping, broadcast_spec);
NodeBuilder builder(broadcast_v3, {arg, shape, axes_mapping}); NodeBuilder builder(broadcast_v3, {arg, shape, axes_mapping});
auto g_broadcast_v3 = ov::as_type_ptr<opset3::Broadcast>(builder.create()); auto g_broadcast_v3 = ov::as_type_ptr<ov::op::v3::Broadcast>(builder.create());
EXPECT_EQ(g_broadcast_v3->get_broadcast_spec(), broadcast_spec); EXPECT_EQ(g_broadcast_v3->get_broadcast_spec(), broadcast_spec);
} }

View File

@ -2,44 +2,41 @@
// SPDX-License-Identifier: Apache-2.0 // SPDX-License-Identifier: Apache-2.0
// //
#include "common_test_utils/visitor.hpp" #include "openvino/op/bucketize.hpp"
#include "gtest/gtest.h"
#include "ngraph/ngraph.hpp" #include <gtest/gtest.h>
#include "ngraph/op/util/attr_types.hpp"
#include "ngraph/opsets/opset1.hpp" #include "openvino/op/parameter.hpp"
#include "ngraph/opsets/opset3.hpp" #include "visitors/visitors.hpp"
#include "ngraph/opsets/opset4.hpp"
#include "ngraph/opsets/opset5.hpp"
using namespace std; using namespace std;
using namespace ngraph; using namespace ov;
using ngraph::test::NodeBuilder; using ov::test::NodeBuilder;
using ngraph::test::ValueMap;
TEST(attributes, bucketize_v3_op_default_attributes) { TEST(attributes, bucketize_v3_op_default_attributes) {
NodeBuilder::get_ops().register_factory<opset3::Bucketize>(); NodeBuilder::get_ops().register_factory<ov::op::v3::Bucketize>();
auto data = make_shared<op::Parameter>(element::f32, Shape{2, 3, 4}); auto data = make_shared<ov::op::v0::Parameter>(element::f32, Shape{2, 3, 4});
auto buckets = make_shared<op::Parameter>(element::f32, Shape{5}); auto buckets = make_shared<ov::op::v0::Parameter>(element::f32, Shape{5});
auto bucketize = make_shared<opset3::Bucketize>(data, buckets); auto bucketize = make_shared<ov::op::v3::Bucketize>(data, buckets);
NodeBuilder builder(bucketize, {data, buckets}); NodeBuilder builder(bucketize, {data, buckets});
auto g_bucketize = ov::as_type_ptr<opset3::Bucketize>(builder.create()); auto g_bucketize = ov::as_type_ptr<ov::op::v3::Bucketize>(builder.create());
EXPECT_EQ(g_bucketize->get_output_type(), bucketize->get_output_type()); EXPECT_EQ(g_bucketize->get_output_type(), bucketize->get_output_type());
EXPECT_EQ(g_bucketize->get_with_right_bound(), bucketize->get_with_right_bound()); EXPECT_EQ(g_bucketize->get_with_right_bound(), bucketize->get_with_right_bound());
} }
TEST(attributes, bucketize_v3_op_custom_attributes) { TEST(attributes, bucketize_v3_op_custom_attributes) {
NodeBuilder::get_ops().register_factory<opset3::Bucketize>(); NodeBuilder::get_ops().register_factory<ov::op::v3::Bucketize>();
auto data = make_shared<op::Parameter>(element::f32, Shape{2, 3, 4}); auto data = make_shared<ov::op::v0::Parameter>(element::f32, Shape{2, 3, 4});
auto buckets = make_shared<op::Parameter>(element::f32, Shape{5}); auto buckets = make_shared<ov::op::v0::Parameter>(element::f32, Shape{5});
element::Type output_type = element::i32; element::Type output_type = element::i32;
bool with_right_bound = false; bool with_right_bound = false;
auto bucketize = make_shared<opset3::Bucketize>(data, buckets, output_type, with_right_bound); auto bucketize = make_shared<ov::op::v3::Bucketize>(data, buckets, output_type, with_right_bound);
NodeBuilder builder(bucketize, {data, buckets}); NodeBuilder builder(bucketize, {data, buckets});
auto g_bucketize = ov::as_type_ptr<opset3::Bucketize>(builder.create()); auto g_bucketize = ov::as_type_ptr<ov::op::v3::Bucketize>(builder.create());
EXPECT_EQ(g_bucketize->get_output_type(), bucketize->get_output_type()); EXPECT_EQ(g_bucketize->get_output_type(), bucketize->get_output_type());
EXPECT_EQ(g_bucketize->get_with_right_bound(), bucketize->get_with_right_bound()); EXPECT_EQ(g_bucketize->get_with_right_bound(), bucketize->get_with_right_bound());

View File

@ -2,8 +2,10 @@
// SPDX-License-Identifier: Apache-2.0 // SPDX-License-Identifier: Apache-2.0
// //
#include "openvino/op/ceiling.hpp"
#include "unary_ops.hpp" #include "unary_ops.hpp"
using Type = ::testing::Types<UnaryOperatorType<ov::op::v0::Ceiling, ngraph::element::f32>>; using Type = ::testing::Types<UnaryOperatorType<ov::op::v0::Ceiling, ov::element::f32>>;
INSTANTIATE_TYPED_TEST_SUITE_P(visitor_without_attribute, UnaryOperatorVisitor, Type, UnaryOperatorTypeName); INSTANTIATE_TYPED_TEST_SUITE_P(visitor_without_attribute, UnaryOperatorVisitor, Type, UnaryOperatorTypeName);

View File

@ -2,27 +2,27 @@
// SPDX-License-Identifier: Apache-2.0 // SPDX-License-Identifier: Apache-2.0
// //
#include "common_test_utils/visitor.hpp" #include "openvino/op/clamp.hpp"
#include "gtest/gtest.h"
#include "ngraph/ngraph.hpp" #include <gtest/gtest.h>
#include "ngraph/op/util/attr_types.hpp"
#include "ngraph/opsets/opset1.hpp" #include "openvino/op/parameter.hpp"
#include "visitors/visitors.hpp"
using namespace std; using namespace std;
using namespace ngraph; using namespace ov;
using ngraph::test::NodeBuilder; using ov::test::NodeBuilder;
using ngraph::test::ValueMap;
TEST(attributes, clamp_op) { TEST(attributes, clamp_op) {
NodeBuilder::get_ops().register_factory<opset1::Clamp>(); NodeBuilder::get_ops().register_factory<ov::op::v0::Clamp>();
auto data = make_shared<op::Parameter>(element::f32, Shape{2, 4}); auto data = make_shared<op::v0::Parameter>(element::f32, Shape{2, 4});
double min = 0.4; double min = 0.4;
double max = 5.6; double max = 5.6;
const auto clamp = make_shared<opset1::Clamp>(data, min, max); const auto clamp = make_shared<op::v0::Clamp>(data, min, max);
NodeBuilder builder(clamp, {data}); NodeBuilder builder(clamp, {data});
auto g_clamp = ov::as_type_ptr<opset1::Clamp>(builder.create()); auto g_clamp = ov::as_type_ptr<op::v0::Clamp>(builder.create());
const auto expected_attr_count = 2; const auto expected_attr_count = 2;
EXPECT_EQ(builder.get_value_map_size(), expected_attr_count); EXPECT_EQ(builder.get_value_map_size(), expected_attr_count);

View File

@ -2,27 +2,26 @@
// SPDX-License-Identifier: Apache-2.0 // SPDX-License-Identifier: Apache-2.0
// //
#include "common_test_utils/visitor.hpp" #include "openvino/op/concat.hpp"
#include "gtest/gtest.h"
#include "ngraph/ngraph.hpp" #include <gtest/gtest.h>
#include "ngraph/op/util/attr_types.hpp"
#include "ngraph/opsets/opset1.hpp" #include "visitors/visitors.hpp"
using namespace std; using namespace std;
using namespace ngraph; using namespace ov;
using ngraph::test::NodeBuilder; using ov::test::NodeBuilder;
using ngraph::test::ValueMap;
TEST(attributes, concat_op) { TEST(attributes, concat_op) {
NodeBuilder::get_ops().register_factory<opset1::Concat>(); NodeBuilder::get_ops().register_factory<ov::op::v0::Concat>();
auto input1 = make_shared<op::Parameter>(element::i64, Shape{1, 2, 3}); auto input1 = make_shared<ov::op::v0::Parameter>(element::i64, Shape{1, 2, 3});
auto input2 = make_shared<op::Parameter>(element::i64, Shape{1, 2, 3}); auto input2 = make_shared<ov::op::v0::Parameter>(element::i64, Shape{1, 2, 3});
auto input3 = make_shared<op::Parameter>(element::i64, Shape{1, 2, 3}); auto input3 = make_shared<ov::op::v0::Parameter>(element::i64, Shape{1, 2, 3});
int64_t axis = 2; int64_t axis = 2;
auto concat = make_shared<opset1::Concat>(ov::NodeVector{input1, input2, input3}, axis); auto concat = make_shared<ov::op::v0::Concat>(ov::NodeVector{input1, input2, input3}, axis);
NodeBuilder builder(concat, {input1, input2, input3}); NodeBuilder builder(concat, {input1, input2, input3});
auto g_concat = ov::as_type_ptr<opset1::Concat>(builder.create()); auto g_concat = ov::as_type_ptr<ov::op::v0::Concat>(builder.create());
EXPECT_EQ(g_concat->get_axis(), concat->get_axis()); EXPECT_EQ(g_concat->get_axis(), concat->get_axis());
} }

View File

@ -2,20 +2,15 @@
// SPDX-License-Identifier: Apache-2.0 // SPDX-License-Identifier: Apache-2.0
// //
#include "common_test_utils/visitor.hpp" #include "openvino/op/constant.hpp"
#include "gtest/gtest.h"
#include "ngraph/ngraph.hpp" #include <gtest/gtest.h>
#include "ngraph/op/util/attr_types.hpp"
#include "ngraph/opsets/opset1.hpp" #include "visitors/visitors.hpp"
#include "ngraph/opsets/opset3.hpp"
#include "ngraph/opsets/opset4.hpp"
#include "ngraph/opsets/opset5.hpp"
#include "ngraph/runtime/host_tensor.hpp"
using namespace std; using namespace std;
using namespace ngraph; using namespace ov;
using ngraph::test::NodeBuilder; using ov::test::NodeBuilder;
using ngraph::test::ValueMap;
TEST(attributes, constant_op) { TEST(attributes, constant_op) {
vector<float> data{5.0f, 4.0f, 3.0f, 2.0f, 1.0f, 0.0f}; vector<float> data{5.0f, 4.0f, 3.0f, 2.0f, 1.0f, 0.0f};
@ -58,10 +53,9 @@ TEST(attributes, constant_op_identical_elements) {
ASSERT_TRUE(g_k->get_all_data_elements_bitwise_identical()); ASSERT_TRUE(g_k->get_all_data_elements_bitwise_identical());
} }
OPENVINO_SUPPRESS_DEPRECATED_START
TEST(attributes, constant_op_from_host_tensor_different_elements) { TEST(attributes, constant_op_from_host_tensor_different_elements) {
vector<int64_t> data{5, 4, 3, 2, 1, 0}; vector<int64_t> data{5, 4, 3, 2, 1, 0};
auto tensor = std::make_shared<runtime::HostTensor>(element::i64, Shape{2, 3}, &data[0]); auto tensor = ov::Tensor(element::i64, Shape{2, 3}, &data[0]);
auto k = make_shared<op::v0::Constant>(tensor); auto k = make_shared<op::v0::Constant>(tensor);
ASSERT_FALSE(k->get_all_data_elements_bitwise_identical()); ASSERT_FALSE(k->get_all_data_elements_bitwise_identical());
NodeBuilder builder(k); NodeBuilder builder(k);
@ -77,7 +71,7 @@ TEST(attributes, constant_op_from_host_tensor_different_elements) {
TEST(attributes, constant_op_from_host_tensor_identical_elements) { TEST(attributes, constant_op_from_host_tensor_identical_elements) {
vector<int64_t> data{5, 5, 5, 5, 5, 5}; vector<int64_t> data{5, 5, 5, 5, 5, 5};
auto tensor = std::make_shared<runtime::HostTensor>(element::i64, Shape{2, 3}, &data[0]); auto tensor = ov::Tensor(element::i64, Shape{2, 3}, &data[0]);
auto k = make_shared<op::v0::Constant>(tensor); auto k = make_shared<op::v0::Constant>(tensor);
ASSERT_TRUE(k->get_all_data_elements_bitwise_identical()); ASSERT_TRUE(k->get_all_data_elements_bitwise_identical());
NodeBuilder builder(k); NodeBuilder builder(k);

View File

@ -2,19 +2,20 @@
// SPDX-License-Identifier: Apache-2.0 // SPDX-License-Identifier: Apache-2.0
// //
#include "common_test_utils/visitor.hpp" #include "openvino/op/convert.hpp"
#include "gtest/gtest.h"
#include "ngraph/ngraph.hpp"
using namespace ngraph; #include <gtest/gtest.h>
using ngraph::test::NodeBuilder;
using ngraph::test::ValueMap; #include "visitors/visitors.hpp"
using namespace ov;
using ov::test::NodeBuilder;
TEST(attributes, convert_op_v0) { TEST(attributes, convert_op_v0) {
using Convert = op::v0::Convert; using Convert = op::v0::Convert;
NodeBuilder::get_ops().register_factory<Convert>(); NodeBuilder::get_ops().register_factory<Convert>();
auto data = std::make_shared<op::Parameter>(element::f32, Shape{2, 4}); auto data = std::make_shared<op::v0::Parameter>(element::f32, Shape{2, 4});
const element::Type destination_type = element::Type_t::i32; const element::Type destination_type = element::Type_t::i32;
const auto convert = std::make_shared<Convert>(data, destination_type); const auto convert = std::make_shared<Convert>(data, destination_type);

View File

@ -2,16 +2,15 @@
// SPDX-License-Identifier: Apache-2.0 // SPDX-License-Identifier: Apache-2.0
// //
#include "common_test_utils/visitor.hpp" #include <gtest/gtest.h>
#include "gtest/gtest.h"
#include "ngraph/op/util/attr_types.hpp"
#include "openvino/op/i420_to_bgr.hpp" #include "openvino/op/i420_to_bgr.hpp"
#include "openvino/op/i420_to_rgb.hpp" #include "openvino/op/i420_to_rgb.hpp"
#include "visitors/visitors.hpp"
using namespace std; using namespace std;
using namespace ov; using namespace ov;
using ngraph::test::NodeBuilder; using ov::test::NodeBuilder;
using ngraph::test::ValueMap;
TEST(attributes, convert_color_i420_rgb) { TEST(attributes, convert_color_i420_rgb) {
NodeBuilder::get_ops().register_factory<op::v8::I420toRGB>(); NodeBuilder::get_ops().register_factory<op::v8::I420toRGB>();

View File

@ -2,16 +2,16 @@
// SPDX-License-Identifier: Apache-2.0 // SPDX-License-Identifier: Apache-2.0
// //
#include "common_test_utils/visitor.hpp" #include <gtest/gtest.h>
#include "gtest/gtest.h"
#include "ngraph/op/util/attr_types.hpp" #include "ngraph/op/util/attr_types.hpp"
#include "openvino/op/nv12_to_bgr.hpp" #include "openvino/op/nv12_to_bgr.hpp"
#include "openvino/op/nv12_to_rgb.hpp" #include "openvino/op/nv12_to_rgb.hpp"
#include "visitors/visitors.hpp"
using namespace std; using namespace std;
using namespace ov; using namespace ov;
using ngraph::test::NodeBuilder; using ov::test::NodeBuilder;
using ngraph::test::ValueMap;
TEST(attributes, convert_color_nv12_rgb) { TEST(attributes, convert_color_nv12_rgb) {
NodeBuilder::get_ops().register_factory<op::v8::NV12toRGB>(); NodeBuilder::get_ops().register_factory<op::v8::NV12toRGB>();

View File

@ -2,25 +2,25 @@
// SPDX-License-Identifier: Apache-2.0 // SPDX-License-Identifier: Apache-2.0
// //
#include "common_test_utils/visitor.hpp" #include "openvino/op/convert_like.hpp"
#include "gtest/gtest.h"
#include "ngraph/ngraph.hpp" #include <gtest/gtest.h>
#include "ngraph/op/util/attr_types.hpp"
#include "ngraph/opsets/opset1.hpp" #include "openvino/op/concat.hpp"
#include "visitors/visitors.hpp"
using namespace std; using namespace std;
using namespace ngraph; using namespace ov;
using ngraph::test::NodeBuilder; using ov::test::NodeBuilder;
using ngraph::test::ValueMap;
TEST(attributes, convert_like_op) { TEST(attributes, convert_like_op) {
NodeBuilder::get_ops().register_factory<opset1::ConvertLike>(); NodeBuilder::get_ops().register_factory<ov::op::v1::ConvertLike>();
auto data = make_shared<op::Parameter>(element::i64, Shape{1, 2, 3}); auto data = make_shared<ov::op::v0::Parameter>(element::i64, Shape{1, 2, 3});
auto like = make_shared<op::Parameter>(element::i64, Shape{1, 2, 3}); auto like = make_shared<ov::op::v0::Parameter>(element::i64, Shape{1, 2, 3});
auto convertLike = make_shared<opset1::ConvertLike>(data, like); auto convertLike = make_shared<ov::op::v1::ConvertLike>(data, like);
NodeBuilder builder(convertLike, {data, like}); NodeBuilder builder(convertLike, {data, like});
auto g_convertLike = ov::as_type_ptr<opset1::Concat>(builder.create()); auto g_convertLike = ov::as_type_ptr<ov::op::v0::Concat>(builder.create());
const auto expected_attr_count = 0; const auto expected_attr_count = 0;
EXPECT_EQ(builder.get_value_map_size(), expected_attr_count); EXPECT_EQ(builder.get_value_map_size(), expected_attr_count);

View File

@ -2,24 +2,21 @@
// SPDX-License-Identifier: Apache-2.0 // SPDX-License-Identifier: Apache-2.0
// //
#include "common_test_utils/visitor.hpp" #include "openvino/op/convolution.hpp"
#include "gtest/gtest.h"
#include "ngraph/ngraph.hpp" #include <gtest/gtest.h>
#include "ngraph/op/util/attr_types.hpp"
#include "ngraph/opsets/opset1.hpp" #include "openvino/op/parameter.hpp"
#include "ngraph/opsets/opset3.hpp" #include "visitors/visitors.hpp"
#include "ngraph/opsets/opset4.hpp"
#include "ngraph/opsets/opset5.hpp"
using namespace std; using namespace std;
using namespace ngraph; using namespace ov;
using ngraph::test::NodeBuilder; using ov::test::NodeBuilder;
using ngraph::test::ValueMap;
TEST(attributes, convolution) { TEST(attributes, convolution) {
NodeBuilder::get_ops().register_factory<op::v1::Convolution>(); NodeBuilder::get_ops().register_factory<op::v1::Convolution>();
auto data = make_shared<op::Parameter>(element::f32, Shape{1, 16, 124, 124}); auto data = make_shared<ov::op::v0::Parameter>(element::f32, Shape{1, 16, 124, 124});
auto filters = make_shared<op::Parameter>(element::f32, Shape{2, 16, 3, 3}); auto filters = make_shared<ov::op::v0::Parameter>(element::f32, Shape{2, 16, 3, 3});
auto strides = Strides{1, 1}; auto strides = Strides{1, 1};
auto pads_begin = CoordinateDiff{1, 2}; auto pads_begin = CoordinateDiff{1, 2};
auto pads_end = CoordinateDiff{1, 2}; auto pads_end = CoordinateDiff{1, 2};
@ -43,8 +40,8 @@ TEST(attributes, convolution) {
TEST(attributes, convolution2) { TEST(attributes, convolution2) {
NodeBuilder::get_ops().register_factory<op::v1::Convolution>(); NodeBuilder::get_ops().register_factory<op::v1::Convolution>();
auto data = make_shared<op::Parameter>(element::f32, Shape{1, 3, 227, 227}); auto data = make_shared<ov::op::v0::Parameter>(element::f32, Shape{1, 3, 227, 227});
auto filters = make_shared<op::Parameter>(element::f32, Shape{96, 3, 227, 227}); auto filters = make_shared<ov::op::v0::Parameter>(element::f32, Shape{96, 3, 227, 227});
auto strides = Strides{4, 4}; auto strides = Strides{4, 4};
auto pads_begin = CoordinateDiff{0, 0}; auto pads_begin = CoordinateDiff{0, 0};
auto pads_end = CoordinateDiff{0, 0}; auto pads_end = CoordinateDiff{0, 0};

View File

@ -2,29 +2,24 @@
// SPDX-License-Identifier: Apache-2.0 // SPDX-License-Identifier: Apache-2.0
// //
#include "common_test_utils/visitor.hpp" #include <gtest/gtest.h>
#include "gtest/gtest.h"
#include "ngraph/ngraph.hpp" #include "openvino/op/convolution.hpp"
#include "ngraph/op/util/attr_types.hpp" #include "visitors/visitors.hpp"
#include "ngraph/opsets/opset1.hpp"
#include "ngraph/opsets/opset3.hpp"
#include "ngraph/opsets/opset4.hpp"
#include "ngraph/opsets/opset5.hpp"
using namespace std; using namespace std;
using namespace ngraph; using namespace ov;
using ngraph::test::NodeBuilder; using ov::test::NodeBuilder;
using ngraph::test::ValueMap;
TEST(attributes, convolution_backprop_op) { TEST(attributes, convolution_backprop_op) {
NodeBuilder::get_ops().register_factory<opset1::ConvolutionBackpropData>(); NodeBuilder::get_ops().register_factory<ov::op::v1::ConvolutionBackpropData>();
auto data = make_shared<op::Parameter>(element::f32, Shape{1, 16, 124, 124}); auto data = make_shared<ov::op::v0::Parameter>(element::f32, Shape{1, 16, 124, 124});
auto filters = make_shared<op::Parameter>(element::f32, Shape{16, 2, 3, 3}); auto filters = make_shared<ov::op::v0::Parameter>(element::f32, Shape{16, 2, 3, 3});
auto strides = Strides{1, 1}; auto strides = Strides{1, 1};
auto pads_begin = CoordinateDiff{1, 2}; auto pads_begin = CoordinateDiff{1, 2};
auto pads_end = CoordinateDiff{1, 2}; auto pads_end = CoordinateDiff{1, 2};
auto dilations = Strides{1, 1}; auto dilations = Strides{1, 1};
auto convolution = make_shared<opset1::ConvolutionBackpropData>(data, auto convolution = make_shared<ov::op::v1::ConvolutionBackpropData>(data,
filters, filters,
strides, strides,
pads_begin, pads_begin,
@ -32,7 +27,7 @@ TEST(attributes, convolution_backprop_op) {
dilations, dilations,
op::PadType::VALID); op::PadType::VALID);
NodeBuilder builder(convolution, {data, filters}); NodeBuilder builder(convolution, {data, filters});
auto g_convolution = ov::as_type_ptr<opset1::ConvolutionBackpropData>(builder.create()); auto g_convolution = ov::as_type_ptr<ov::op::v1::ConvolutionBackpropData>(builder.create());
// attribute count // attribute count
const auto expected_attr_count = 6; const auto expected_attr_count = 6;
@ -46,10 +41,10 @@ TEST(attributes, convolution_backprop_op) {
} }
TEST(attributes, convolution_backprop_output_shape_output_padding) { TEST(attributes, convolution_backprop_output_shape_output_padding) {
NodeBuilder::get_ops().register_factory<opset1::ConvolutionBackpropData>(); NodeBuilder::get_ops().register_factory<ov::op::v1::ConvolutionBackpropData>();
const auto data = make_shared<op::Parameter>(element::f32, Shape{1, 16, 124, 124}); const auto data = make_shared<ov::op::v0::Parameter>(element::f32, Shape{1, 16, 124, 124});
const auto filter = make_shared<op::Parameter>(element::f32, Shape{16, 2, 3, 3}); const auto filter = make_shared<ov::op::v0::Parameter>(element::f32, Shape{16, 2, 3, 3});
const auto output_shape = make_shared<op::Parameter>(element::i32, Shape{2}); const auto output_shape = make_shared<ov::op::v0::Parameter>(element::i32, Shape{2});
const auto strides = Strides{2, 1}; const auto strides = Strides{2, 1};
const auto pads_begin = CoordinateDiff{3, 4}; const auto pads_begin = CoordinateDiff{3, 4};
@ -65,7 +60,7 @@ TEST(attributes, convolution_backprop_output_shape_output_padding) {
op::PadType::NOTSET}; op::PadType::NOTSET};
for (auto padType : allPadTypes) { for (auto padType : allPadTypes) {
const auto convolution = make_shared<opset1::ConvolutionBackpropData>(data, const auto convolution = make_shared<ov::op::v1::ConvolutionBackpropData>(data,
filter, filter,
output_shape, output_shape,
strides, strides,
@ -75,7 +70,7 @@ TEST(attributes, convolution_backprop_output_shape_output_padding) {
padType, padType,
output_padding); output_padding);
NodeBuilder builder(convolution, {data, filter}); NodeBuilder builder(convolution, {data, filter});
const auto g_convolution = ov::as_type_ptr<opset1::ConvolutionBackpropData>(builder.create()); const auto g_convolution = ov::as_type_ptr<ov::op::v1::ConvolutionBackpropData>(builder.create());
// attribute count // attribute count
const auto expected_attr_count = 6; const auto expected_attr_count = 6;

View File

@ -2,8 +2,10 @@
// SPDX-License-Identifier: Apache-2.0 // SPDX-License-Identifier: Apache-2.0
// //
#include "openvino/op/cos.hpp"
#include "unary_ops.hpp" #include "unary_ops.hpp"
using Type = ::testing::Types<UnaryOperatorType<ov::op::v0::Cos, ngraph::element::f32>>; using Type = ::testing::Types<UnaryOperatorType<ov::op::v0::Cos, ov::element::f32>>;
INSTANTIATE_TYPED_TEST_SUITE_P(visitor_without_attribute, UnaryOperatorVisitor, Type, UnaryOperatorTypeName); INSTANTIATE_TYPED_TEST_SUITE_P(visitor_without_attribute, UnaryOperatorVisitor, Type, UnaryOperatorTypeName);

View File

@ -2,8 +2,10 @@
// SPDX-License-Identifier: Apache-2.0 // SPDX-License-Identifier: Apache-2.0
// //
#include "openvino/op/cosh.hpp"
#include "unary_ops.hpp" #include "unary_ops.hpp"
using Type = ::testing::Types<UnaryOperatorType<ov::op::v0::Cosh, ngraph::element::f32>>; using Type = ::testing::Types<UnaryOperatorType<ov::op::v0::Cosh, ov::element::f32>>;
INSTANTIATE_TYPED_TEST_SUITE_P(visitor_without_attribute, UnaryOperatorVisitor, Type, UnaryOperatorTypeName); INSTANTIATE_TYPED_TEST_SUITE_P(visitor_without_attribute, UnaryOperatorVisitor, Type, UnaryOperatorTypeName);

View File

@ -2,25 +2,25 @@
// SPDX-License-Identifier: Apache-2.0 // SPDX-License-Identifier: Apache-2.0
// //
#include "common_test_utils/visitor.hpp" #include "openvino/op/ctc_greedy_decoder.hpp"
#include "gtest/gtest.h"
#include "ngraph/ngraph.hpp" #include <gtest/gtest.h>
#include "ngraph/opsets/opset1.hpp"
#include "visitors/visitors.hpp"
using namespace std; using namespace std;
using namespace ngraph; using namespace ov;
using ngraph::test::NodeBuilder; using ov::test::NodeBuilder;
using ngraph::test::ValueMap;
TEST(attributes, ctc_greedy_decoder_op) { TEST(attributes, ctc_greedy_decoder_op) {
NodeBuilder::get_ops().register_factory<opset1::CTCGreedyDecoder>(); NodeBuilder::get_ops().register_factory<ov::op::v0::CTCGreedyDecoder>();
bool m_ctc_merge_repeated = false; bool m_ctc_merge_repeated = false;
auto data = make_shared<op::Parameter>(element::f32, Shape{3, 1, 3}); auto data = make_shared<ov::op::v0::Parameter>(element::f32, Shape{3, 1, 3});
auto masks = make_shared<op::Parameter>(element::i32, Shape{3, 1}); auto masks = make_shared<ov::op::v0::Parameter>(element::i32, Shape{3, 1});
auto decoder = make_shared<op::v0::CTCGreedyDecoder>(data, masks, m_ctc_merge_repeated); auto decoder = make_shared<op::v0::CTCGreedyDecoder>(data, masks, m_ctc_merge_repeated);
NodeBuilder builder(decoder, {data, masks}); NodeBuilder builder(decoder, {data, masks});
auto g_decoder = ov::as_type_ptr<opset1::CTCGreedyDecoder>(builder.create()); auto g_decoder = ov::as_type_ptr<ov::op::v0::CTCGreedyDecoder>(builder.create());
EXPECT_EQ(g_decoder->get_ctc_merge_repeated(), decoder->get_ctc_merge_repeated()); EXPECT_EQ(g_decoder->get_ctc_merge_repeated(), decoder->get_ctc_merge_repeated());
} }

View File

@ -2,27 +2,27 @@
// SPDX-License-Identifier: Apache-2.0 // SPDX-License-Identifier: Apache-2.0
// //
#include "common_test_utils/visitor.hpp" #include "openvino/op/ctc_greedy_decoder_seq_len.hpp"
#include "gtest/gtest.h"
#include "ngraph/ngraph.hpp" #include <gtest/gtest.h>
#include "ngraph/opsets/opset1.hpp"
#include "ngraph/opsets/opset6.hpp" #include "openvino/op/constant.hpp"
#include "visitors/visitors.hpp"
using namespace std; using namespace std;
using namespace ngraph; using namespace ov;
using ngraph::test::NodeBuilder; using ov::test::NodeBuilder;
using ngraph::test::ValueMap;
TEST(attributes, ctc_greedy_decoder_seq_len_op) { TEST(attributes, ctc_greedy_decoder_seq_len_op) {
NodeBuilder::get_ops().register_factory<opset6::CTCGreedyDecoderSeqLen>(); NodeBuilder::get_ops().register_factory<ov::op::v6::CTCGreedyDecoderSeqLen>();
bool merge_repeated = false; bool merge_repeated = false;
auto data = make_shared<op::Parameter>(element::f32, Shape{1, 3, 3}); auto data = make_shared<ov::op::v0::Parameter>(element::f32, Shape{1, 3, 3});
auto seq_len = make_shared<op::Parameter>(element::i32, Shape{1}); auto seq_len = make_shared<ov::op::v0::Parameter>(element::i32, Shape{1});
auto blank_index = op::Constant::create<int32_t>(element::i32, Shape{}, {2}); auto blank_index = ov::op::v0::Constant::create<int32_t>(element::i32, Shape{}, {2});
auto decoder = make_shared<op::v6::CTCGreedyDecoderSeqLen>(data, seq_len, blank_index, merge_repeated); auto decoder = make_shared<op::v6::CTCGreedyDecoderSeqLen>(data, seq_len, blank_index, merge_repeated);
NodeBuilder builder(decoder, {data, seq_len, blank_index}); NodeBuilder builder(decoder, {data, seq_len, blank_index});
auto g_decoder = ov::as_type_ptr<opset6::CTCGreedyDecoderSeqLen>(builder.create()); auto g_decoder = ov::as_type_ptr<ov::op::v6::CTCGreedyDecoderSeqLen>(builder.create());
EXPECT_EQ(g_decoder->get_merge_repeated(), decoder->get_merge_repeated()); EXPECT_EQ(g_decoder->get_merge_repeated(), decoder->get_merge_repeated());
} }

View File

@ -2,32 +2,29 @@
// SPDX-License-Identifier: Apache-2.0 // SPDX-License-Identifier: Apache-2.0
// //
#include "common_test_utils/visitor.hpp" #include "openvino/op/ctc_loss.hpp"
#include "gtest/gtest.h"
#include "ngraph/ngraph.hpp" #include <gtest/gtest.h>
#include "ngraph/op/util/attr_types.hpp"
#include "ngraph/opsets/opset1.hpp" #include "openvino/op/parameter.hpp"
#include "ngraph/opsets/opset3.hpp" #include "visitors/visitors.hpp"
#include "ngraph/opsets/opset4.hpp"
#include "ngraph/opsets/opset5.hpp"
using namespace std; using namespace std;
using namespace ngraph; using namespace ov;
using ngraph::test::NodeBuilder; using ov::test::NodeBuilder;
using ngraph::test::ValueMap;
TEST(attributes, ctc_loss) { TEST(attributes, ctc_loss) {
NodeBuilder::get_ops().register_factory<opset4::CTCLoss>(); NodeBuilder::get_ops().register_factory<ov::op::v4::CTCLoss>();
auto logits = make_shared<op::Parameter>(element::f32, Shape{10, 120, 28}); auto logits = make_shared<ov::op::v0::Parameter>(element::f32, Shape{10, 120, 28});
auto logit_length = make_shared<op::Parameter>(element::i32, Shape{10}); auto logit_length = make_shared<ov::op::v0::Parameter>(element::i32, Shape{10});
auto labels = make_shared<op::Parameter>(element::i32, Shape{10, 120}); auto labels = make_shared<ov::op::v0::Parameter>(element::i32, Shape{10, 120});
auto label_length = make_shared<op::Parameter>(element::i32, Shape{10}); auto label_length = make_shared<ov::op::v0::Parameter>(element::i32, Shape{10});
auto blank_index = make_shared<op::Parameter>(element::i32, Shape{}); auto blank_index = make_shared<ov::op::v0::Parameter>(element::i32, Shape{});
auto ctc_loss = make_shared<opset4::CTCLoss>(logits, logit_length, labels, label_length, blank_index); auto ctc_loss = make_shared<ov::op::v4::CTCLoss>(logits, logit_length, labels, label_length, blank_index);
NodeBuilder builder(ctc_loss, {logits, logit_length, labels, label_length, blank_index}); NodeBuilder builder(ctc_loss, {logits, logit_length, labels, label_length, blank_index});
auto g_ctc_loss = as_type_ptr<opset4::CTCLoss>(builder.create()); auto g_ctc_loss = as_type_ptr<ov::op::v4::CTCLoss>(builder.create());
// attribute count // attribute count
const auto expected_attr_count = 3; const auto expected_attr_count = 3;

View File

@ -2,25 +2,25 @@
// SPDX-License-Identifier: Apache-2.0 // SPDX-License-Identifier: Apache-2.0
// //
#include "common_test_utils/visitor.hpp" #include "openvino/op/cum_sum.hpp"
#include "gtest/gtest.h"
#include "ngraph/ngraph.hpp" #include <gtest/gtest.h>
#include "ngraph/op/util/attr_types.hpp"
#include "ngraph/opsets/opset3.hpp" #include "visitors/visitors.hpp"
using namespace std; using namespace std;
using namespace ngraph; using namespace ov;
using ngraph::test::NodeBuilder; using ov::test::NodeBuilder;
TEST(attributes, cum_sum_op_default_attributes_no_axis_input) { TEST(attributes, cum_sum_op_default_attributes_no_axis_input) {
NodeBuilder::get_ops().register_factory<opset3::CumSum>(); NodeBuilder::get_ops().register_factory<ov::op::v0::CumSum>();
Shape shape{1, 4}; Shape shape{1, 4};
auto A = make_shared<op::Parameter>(element::f32, shape); auto A = make_shared<ov::op::v0::Parameter>(element::f32, shape);
auto cs = make_shared<op::CumSum>(A); auto cs = make_shared<op::v0::CumSum>(A);
NodeBuilder builder(cs, {A}); NodeBuilder builder(cs, {A});
auto g_cs = ov::as_type_ptr<opset3::CumSum>(builder.create()); auto g_cs = ov::as_type_ptr<op::v0::CumSum>(builder.create());
const auto expected_attr_count = 2; const auto expected_attr_count = 2;
EXPECT_EQ(builder.get_value_map_size(), expected_attr_count); EXPECT_EQ(builder.get_value_map_size(), expected_attr_count);
@ -30,15 +30,15 @@ TEST(attributes, cum_sum_op_default_attributes_no_axis_input) {
} }
TEST(attributes, cum_sum_op_default_attributes) { TEST(attributes, cum_sum_op_default_attributes) {
NodeBuilder::get_ops().register_factory<opset3::CumSum>(); NodeBuilder::get_ops().register_factory<op::v0::CumSum>();
Shape shape{1, 4}; Shape shape{1, 4};
auto A = make_shared<op::Parameter>(element::f32, shape); auto A = make_shared<ov::op::v0::Parameter>(element::f32, shape);
auto axis = make_shared<op::Parameter>(element::i32, Shape{1}); auto axis = make_shared<ov::op::v0::Parameter>(element::i32, Shape{1});
auto cs = make_shared<op::CumSum>(A, axis); auto cs = make_shared<op::v0::CumSum>(A, axis);
NodeBuilder builder(cs, {A, axis}); NodeBuilder builder(cs, {A, axis});
auto g_cs = ov::as_type_ptr<opset3::CumSum>(builder.create()); auto g_cs = ov::as_type_ptr<op::v0::CumSum>(builder.create());
const auto expected_attr_count = 2; const auto expected_attr_count = 2;
EXPECT_EQ(builder.get_value_map_size(), expected_attr_count); EXPECT_EQ(builder.get_value_map_size(), expected_attr_count);
@ -48,17 +48,17 @@ TEST(attributes, cum_sum_op_default_attributes) {
} }
TEST(attributes, cum_sum_op_custom_attributes) { TEST(attributes, cum_sum_op_custom_attributes) {
NodeBuilder::get_ops().register_factory<opset3::CumSum>(); NodeBuilder::get_ops().register_factory<op::v0::CumSum>();
Shape shape{1, 4}; Shape shape{1, 4};
auto A = make_shared<op::Parameter>(element::f32, shape); auto A = make_shared<ov::op::v0::Parameter>(element::f32, shape);
auto axis = make_shared<op::Parameter>(element::i32, Shape{1}); auto axis = make_shared<ov::op::v0::Parameter>(element::i32, Shape{1});
bool exclusive = true; bool exclusive = true;
bool reverse = true; bool reverse = true;
auto cs = make_shared<op::CumSum>(A, axis, exclusive, reverse); auto cs = make_shared<op::v0::CumSum>(A, axis, exclusive, reverse);
NodeBuilder builder(cs, {A, axis}); NodeBuilder builder(cs, {A, axis});
auto g_cs = ov::as_type_ptr<opset3::CumSum>(builder.create()); auto g_cs = ov::as_type_ptr<op::v0::CumSum>(builder.create());
const auto expected_attr_count = 2; const auto expected_attr_count = 2;
EXPECT_EQ(builder.get_value_map_size(), expected_attr_count); EXPECT_EQ(builder.get_value_map_size(), expected_attr_count);

View File

@ -2,32 +2,35 @@
// SPDX-License-Identifier: Apache-2.0 // SPDX-License-Identifier: Apache-2.0
// //
#include "common_test_utils/visitor.hpp" #include "openvino/op/deformable_convolution.hpp"
#include "gtest/gtest.h"
#include "ngraph/ngraph.hpp" #include <gtest/gtest.h>
#include "ngraph/op/util/attr_types.hpp"
#include "ngraph/opsets/opset1.hpp" #include "visitors/visitors.hpp"
#include "ngraph/opsets/opset8.hpp"
using namespace std; using namespace std;
using namespace ngraph; using namespace ov;
using ngraph::test::NodeBuilder; using ov::test::NodeBuilder;
using ngraph::test::ValueMap;
TEST(attributes, deformable_convolution_default_attributes) { TEST(attributes, deformable_convolution_default_attributes) {
NodeBuilder::get_ops().register_factory<opset1::DeformableConvolution>(); NodeBuilder::get_ops().register_factory<ov::op::v1::DeformableConvolution>();
const Shape inputs_shape{1, 1, 5, 5}; const Shape inputs_shape{1, 1, 5, 5};
auto data = make_shared<op::Parameter>(element::f32, Shape{1, 1, 5, 5}); auto data = make_shared<ov::op::v0::Parameter>(element::f32, Shape{1, 1, 5, 5});
auto filters = make_shared<op::Parameter>(element::f32, Shape{1, 1, 3, 3}); auto filters = make_shared<ov::op::v0::Parameter>(element::f32, Shape{1, 1, 3, 3});
auto offsets = make_shared<op::Parameter>(element::f32, Shape{1, 18, 3, 3}); auto offsets = make_shared<ov::op::v0::Parameter>(element::f32, Shape{1, 18, 3, 3});
auto strides = Strides{1, 1}; auto strides = Strides{1, 1};
auto pads_begin = CoordinateDiff{0, 0}; auto pads_begin = CoordinateDiff{0, 0};
auto pads_end = CoordinateDiff{0, 0}; auto pads_end = CoordinateDiff{0, 0};
auto dilations = Strides{1, 1}; auto dilations = Strides{1, 1};
auto convolution = auto convolution = make_shared<ov::op::v1::DeformableConvolution>(data,
make_shared<opset1::DeformableConvolution>(data, offsets, filters, strides, pads_begin, pads_end, dilations); offsets,
filters,
strides,
pads_begin,
pads_end,
dilations);
NodeBuilder builder(convolution, {data, offsets, filters}); NodeBuilder builder(convolution, {data, offsets, filters});
auto g_convolution = ov::as_type_ptr<opset1::DeformableConvolution>(builder.create()); auto g_convolution = ov::as_type_ptr<ov::op::v1::DeformableConvolution>(builder.create());
// attribute count // attribute count
const auto expected_attr_count = 7; const auto expected_attr_count = 7;
@ -43,16 +46,16 @@ TEST(attributes, deformable_convolution_default_attributes) {
} }
TEST(attributes, deformable_convolution_attributes) { TEST(attributes, deformable_convolution_attributes) {
NodeBuilder::get_ops().register_factory<opset1::DeformableConvolution>(); NodeBuilder::get_ops().register_factory<ov::op::v1::DeformableConvolution>();
const Shape inputs_shape{1, 1, 5, 5}; const Shape inputs_shape{1, 1, 5, 5};
auto data = make_shared<op::Parameter>(element::f32, Shape{1, 2, 5, 5}); auto data = make_shared<ov::op::v0::Parameter>(element::f32, Shape{1, 2, 5, 5});
auto filters = make_shared<op::Parameter>(element::f32, Shape{2, 1, 3, 3}); auto filters = make_shared<ov::op::v0::Parameter>(element::f32, Shape{2, 1, 3, 3});
auto offsets = make_shared<op::Parameter>(element::f32, Shape{1, 36, 5, 5}); auto offsets = make_shared<ov::op::v0::Parameter>(element::f32, Shape{1, 36, 5, 5});
auto strides = Strides{1, 1}; auto strides = Strides{1, 1};
auto pads_begin = CoordinateDiff{0, 0}; auto pads_begin = CoordinateDiff{0, 0};
auto pads_end = CoordinateDiff{0, 0}; auto pads_end = CoordinateDiff{0, 0};
auto dilations = Strides{1, 1}; auto dilations = Strides{1, 1};
auto convolution = make_shared<opset1::DeformableConvolution>(data, auto convolution = make_shared<ov::op::v1::DeformableConvolution>(data,
offsets, offsets,
filters, filters,
strides, strides,
@ -63,7 +66,7 @@ TEST(attributes, deformable_convolution_attributes) {
2, 2,
2); 2);
NodeBuilder builder(convolution, {data, offsets, filters}); NodeBuilder builder(convolution, {data, offsets, filters});
auto g_convolution = ov::as_type_ptr<opset1::DeformableConvolution>(builder.create()); auto g_convolution = ov::as_type_ptr<ov::op::v1::DeformableConvolution>(builder.create());
// attribute count // attribute count
const auto expected_attr_count = 7; const auto expected_attr_count = 7;
@ -79,19 +82,24 @@ TEST(attributes, deformable_convolution_attributes) {
} }
TEST(attributes, deformable_convolution_v8_default_attributes) { TEST(attributes, deformable_convolution_v8_default_attributes) {
NodeBuilder::get_ops().register_factory<opset8::DeformableConvolution>(); NodeBuilder::get_ops().register_factory<ov::op::v8::DeformableConvolution>();
const Shape inputs_shape{1, 1, 5, 5}; const Shape inputs_shape{1, 1, 5, 5};
auto data = make_shared<op::Parameter>(element::f32, Shape{1, 1, 5, 5}); auto data = make_shared<ov::op::v0::Parameter>(element::f32, Shape{1, 1, 5, 5});
auto filters = make_shared<op::Parameter>(element::f32, Shape{1, 1, 3, 3}); auto filters = make_shared<ov::op::v0::Parameter>(element::f32, Shape{1, 1, 3, 3});
auto offsets = make_shared<op::Parameter>(element::f32, Shape{1, 18, 3, 3}); auto offsets = make_shared<ov::op::v0::Parameter>(element::f32, Shape{1, 18, 3, 3});
auto strides = Strides{1, 1}; auto strides = Strides{1, 1};
auto pads_begin = CoordinateDiff{0, 0}; auto pads_begin = CoordinateDiff{0, 0};
auto pads_end = CoordinateDiff{0, 0}; auto pads_end = CoordinateDiff{0, 0};
auto dilations = Strides{1, 1}; auto dilations = Strides{1, 1};
auto convolution = auto convolution = make_shared<ov::op::v8::DeformableConvolution>(data,
make_shared<opset8::DeformableConvolution>(data, offsets, filters, strides, pads_begin, pads_end, dilations); offsets,
filters,
strides,
pads_begin,
pads_end,
dilations);
NodeBuilder builder(convolution, {data, offsets, filters}); NodeBuilder builder(convolution, {data, offsets, filters});
auto g_convolution = ov::as_type_ptr<opset8::DeformableConvolution>(builder.create()); auto g_convolution = ov::as_type_ptr<ov::op::v8::DeformableConvolution>(builder.create());
// attribute count // attribute count
const auto expected_attr_count = 8; const auto expected_attr_count = 8;
@ -108,17 +116,17 @@ TEST(attributes, deformable_convolution_v8_default_attributes) {
} }
TEST(attributes, deformable_convolution_v8_attributes) { TEST(attributes, deformable_convolution_v8_attributes) {
NodeBuilder::get_ops().register_factory<opset8::DeformableConvolution>(); NodeBuilder::get_ops().register_factory<ov::op::v8::DeformableConvolution>();
const Shape inputs_shape{1, 1, 5, 5}; const Shape inputs_shape{1, 1, 5, 5};
auto data = make_shared<op::Parameter>(element::f32, Shape{1, 2, 5, 5}); auto data = make_shared<ov::op::v0::Parameter>(element::f32, Shape{1, 2, 5, 5});
auto filters = make_shared<op::Parameter>(element::f32, Shape{2, 1, 3, 3}); auto filters = make_shared<ov::op::v0::Parameter>(element::f32, Shape{2, 1, 3, 3});
auto offsets = make_shared<op::Parameter>(element::f32, Shape{1, 36, 5, 5}); auto offsets = make_shared<ov::op::v0::Parameter>(element::f32, Shape{1, 36, 5, 5});
auto mask = make_shared<op::Parameter>(element::f32, Shape{1, 18, 5, 5}); auto mask = make_shared<ov::op::v0::Parameter>(element::f32, Shape{1, 18, 5, 5});
auto strides = Strides{1, 1}; auto strides = Strides{1, 1};
auto pads_begin = CoordinateDiff{0, 0}; auto pads_begin = CoordinateDiff{0, 0};
auto pads_end = CoordinateDiff{0, 0}; auto pads_end = CoordinateDiff{0, 0};
auto dilations = Strides{1, 1}; auto dilations = Strides{1, 1};
auto convolution = make_shared<opset8::DeformableConvolution>(data, auto convolution = make_shared<ov::op::v8::DeformableConvolution>(data,
offsets, offsets,
filters, filters,
mask, mask,
@ -131,7 +139,7 @@ TEST(attributes, deformable_convolution_v8_attributes) {
2, 2,
true); true);
NodeBuilder builder(convolution, {data, offsets, filters, mask}); NodeBuilder builder(convolution, {data, offsets, filters, mask});
auto g_convolution = ov::as_type_ptr<opset8::DeformableConvolution>(builder.create()); auto g_convolution = ov::as_type_ptr<ov::op::v8::DeformableConvolution>(builder.create());
// attribute count // attribute count
const auto expected_attr_count = 8; const auto expected_attr_count = 8;

View File

@ -2,21 +2,20 @@
// SPDX-License-Identifier: Apache-2.0 // SPDX-License-Identifier: Apache-2.0
// //
#include "common_test_utils/visitor.hpp" #include "openvino/op/deformable_psroi_pooling.hpp"
#include "gtest/gtest.h"
#include "ngraph/ngraph.hpp" #include <gtest/gtest.h>
#include "ngraph/op/util/attr_types.hpp"
#include "ngraph/opsets/opset1.hpp" #include "visitors/visitors.hpp"
using namespace std; using namespace std;
using namespace ngraph; using namespace ov;
using ngraph::test::NodeBuilder; using ov::test::NodeBuilder;
using ngraph::test::ValueMap;
TEST(attributes, deformable_psroi_pooling_op) { TEST(attributes, deformable_psroi_pooling_op) {
NodeBuilder::get_ops().register_factory<opset1::DeformablePSROIPooling>(); NodeBuilder::get_ops().register_factory<ov::op::v1::DeformablePSROIPooling>();
auto input = make_shared<op::Parameter>(element::f32, Shape{2, 16, 67, 32}); auto input = make_shared<ov::op::v0::Parameter>(element::f32, Shape{2, 16, 67, 32});
auto coords = make_shared<op::Parameter>(element::f32, Shape{300, 5}); auto coords = make_shared<ov::op::v0::Parameter>(element::f32, Shape{300, 5});
const int output_dim = 4; const int output_dim = 4;
const float spatial_scale = 0.0625; const float spatial_scale = 0.0625;
@ -27,7 +26,7 @@ TEST(attributes, deformable_psroi_pooling_op) {
const float trans_std = 0.1f; const float trans_std = 0.1f;
const int part_size = 3; const int part_size = 3;
auto op = make_shared<opset1::DeformablePSROIPooling>(input, auto op = make_shared<ov::op::v1::DeformablePSROIPooling>(input,
coords, coords,
output_dim, output_dim,
spatial_scale, spatial_scale,
@ -38,7 +37,7 @@ TEST(attributes, deformable_psroi_pooling_op) {
trans_std, trans_std,
part_size); part_size);
NodeBuilder builder(op, {input, coords}); NodeBuilder builder(op, {input, coords});
auto g_op = ov::as_type_ptr<opset1::DeformablePSROIPooling>(builder.create()); auto g_op = ov::as_type_ptr<ov::op::v1::DeformablePSROIPooling>(builder.create());
EXPECT_EQ(g_op->get_output_dim(), op->get_output_dim()); EXPECT_EQ(g_op->get_output_dim(), op->get_output_dim());
EXPECT_EQ(g_op->get_spatial_scale(), op->get_spatial_scale()); EXPECT_EQ(g_op->get_spatial_scale(), op->get_spatial_scale());
@ -51,10 +50,10 @@ TEST(attributes, deformable_psroi_pooling_op) {
} }
TEST(attributes, deformable_psroi_pooling_op2) { TEST(attributes, deformable_psroi_pooling_op2) {
NodeBuilder::get_ops().register_factory<opset1::DeformablePSROIPooling>(); NodeBuilder::get_ops().register_factory<ov::op::v1::DeformablePSROIPooling>();
auto input = make_shared<op::Parameter>(element::f32, Shape{2, 16, 67, 32}); auto input = make_shared<ov::op::v0::Parameter>(element::f32, Shape{2, 16, 67, 32});
auto coords = make_shared<op::Parameter>(element::f32, Shape{300, 5}); auto coords = make_shared<ov::op::v0::Parameter>(element::f32, Shape{300, 5});
auto offset = make_shared<op::Parameter>(element::i64, Shape{300, 2, 2, 2}); auto offset = make_shared<ov::op::v0::Parameter>(element::i64, Shape{300, 2, 2, 2});
const int output_dim = 4; const int output_dim = 4;
const float spatial_scale = 0.0625; const float spatial_scale = 0.0625;
@ -65,7 +64,7 @@ TEST(attributes, deformable_psroi_pooling_op2) {
const float trans_std = 0.1f; const float trans_std = 0.1f;
const int part_size = 3; const int part_size = 3;
auto op = make_shared<opset1::DeformablePSROIPooling>(input, auto op = make_shared<ov::op::v1::DeformablePSROIPooling>(input,
coords, coords,
offset, offset,
output_dim, output_dim,
@ -77,7 +76,7 @@ TEST(attributes, deformable_psroi_pooling_op2) {
trans_std, trans_std,
part_size); part_size);
NodeBuilder builder(op, {input, coords, offset}); NodeBuilder builder(op, {input, coords, offset});
auto g_op = ov::as_type_ptr<opset1::DeformablePSROIPooling>(builder.create()); auto g_op = ov::as_type_ptr<ov::op::v1::DeformablePSROIPooling>(builder.create());
EXPECT_EQ(g_op->get_output_dim(), op->get_output_dim()); EXPECT_EQ(g_op->get_output_dim(), op->get_output_dim());
EXPECT_EQ(g_op->get_spatial_scale(), op->get_spatial_scale()); EXPECT_EQ(g_op->get_spatial_scale(), op->get_spatial_scale());

View File

@ -2,29 +2,25 @@
// SPDX-License-Identifier: Apache-2.0 // SPDX-License-Identifier: Apache-2.0
// //
#include "common_test_utils/visitor.hpp" #include "openvino/op/depth_to_space.hpp"
#include "gtest/gtest.h"
#include "ngraph/ngraph.hpp"
#include "ngraph/op/util/attr_types.hpp"
#include "ngraph/opsets/opset1.hpp"
#include "ngraph/opsets/opset3.hpp"
#include "ngraph/opsets/opset4.hpp"
#include "ngraph/opsets/opset5.hpp"
using namespace ngraph; #include <gtest/gtest.h>
using ngraph::test::NodeBuilder;
using ngraph::test::ValueMap; #include "visitors/visitors.hpp"
using namespace ov;
using ov::test::NodeBuilder;
TEST(attributes, depth_to_space) { TEST(attributes, depth_to_space) {
NodeBuilder::get_ops().register_factory<opset1::DepthToSpace>(); NodeBuilder::get_ops().register_factory<ov::op::v0::DepthToSpace>();
auto data = std::make_shared<op::Parameter>(element::f32, Shape{1, 8, 2, 2}); auto data = std::make_shared<ov::op::v0::Parameter>(element::f32, Shape{1, 8, 2, 2});
const auto block_size = 2; const auto block_size = 2;
const auto mode = "blocks_first"; const auto mode = "blocks_first";
const auto dts = std::make_shared<opset1::DepthToSpace>(data, mode, block_size); const auto dts = std::make_shared<ov::op::v0::DepthToSpace>(data, mode, block_size);
NodeBuilder builder(dts, {data}); NodeBuilder builder(dts, {data});
auto g_dts = ov::as_type_ptr<opset1::DepthToSpace>(builder.create()); auto g_dts = ov::as_type_ptr<ov::op::v0::DepthToSpace>(builder.create());
// attribute count // attribute count
const auto expected_attr_count = 2; const auto expected_attr_count = 2;

View File

@ -4,20 +4,17 @@
#include "openvino/op/detection_output.hpp" #include "openvino/op/detection_output.hpp"
#include "common_test_utils/visitor.hpp" #include <gtest/gtest.h>
#include "gtest/gtest.h"
#include "openvino/op/util/attr_types.hpp" #include "openvino/op/util/attr_types.hpp"
#include "openvino/opsets/opset8.hpp" #include "visitors/visitors.hpp"
using namespace std; using namespace std;
using namespace ngraph; using namespace ov;
using namespace ov::op; using ov::test::NodeBuilder;
using namespace ov::op::util;
using ngraph::test::NodeBuilder;
using ngraph::test::ValueMap;
namespace { namespace {
void initialize_attributes(DetectionOutputBase::AttributesBase& attrs) { void initialize_attributes(ov::op::util::DetectionOutputBase::AttributesBase& attrs) {
attrs.background_label_id = 0; attrs.background_label_id = 0;
attrs.top_k = 1; attrs.top_k = 1;
attrs.variance_encoded_in_target = false; attrs.variance_encoded_in_target = false;
@ -34,8 +31,8 @@ void initialize_attributes(DetectionOutputBase::AttributesBase& attrs) {
attrs.input_width = 32; attrs.input_width = 32;
attrs.objectness_score = 0.73f; attrs.objectness_score = 0.73f;
} }
void is_equal_attrs(const DetectionOutputBase::AttributesBase& attrs1, void is_equal_attrs(const ov::op::util::DetectionOutputBase::AttributesBase& attrs1,
const DetectionOutputBase::AttributesBase& attrs2) { const ov::op::util::DetectionOutputBase::AttributesBase& attrs2) {
EXPECT_EQ(attrs1.background_label_id, attrs2.background_label_id); EXPECT_EQ(attrs1.background_label_id, attrs2.background_label_id);
EXPECT_EQ(attrs1.top_k, attrs2.top_k); EXPECT_EQ(attrs1.top_k, attrs2.top_k);
EXPECT_EQ(attrs1.variance_encoded_in_target, attrs2.variance_encoded_in_target); EXPECT_EQ(attrs1.variance_encoded_in_target, attrs2.variance_encoded_in_target);
@ -95,9 +92,9 @@ TEST(attributes, detection_output_v8) {
initialize_attributes(attrs); initialize_attributes(attrs);
auto detection_output = auto detection_output =
make_shared<v8::DetectionOutput>(box_logits, class_preds, proposals, aux_class_preds, aux_box_pred, attrs); make_shared<op::v8::DetectionOutput>(box_logits, class_preds, proposals, aux_class_preds, aux_box_pred, attrs);
NodeBuilder builder(detection_output, {box_logits, class_preds, proposals, aux_class_preds, aux_box_pred}); NodeBuilder builder(detection_output, {box_logits, class_preds, proposals, aux_class_preds, aux_box_pred});
auto g_detection_output = ov::as_type_ptr<v8::DetectionOutput>(builder.create()); auto g_detection_output = ov::as_type_ptr<op::v8::DetectionOutput>(builder.create());
const auto do_attrs = detection_output->get_attrs(); const auto do_attrs = detection_output->get_attrs();
const auto g_do_attrs = g_detection_output->get_attrs(); const auto g_do_attrs = g_detection_output->get_attrs();

View File

@ -2,14 +2,16 @@
// SPDX-License-Identifier: Apache-2.0 // SPDX-License-Identifier: Apache-2.0
// //
#include "common_test_utils/visitor.hpp" #include "openvino/op/dft.hpp"
#include "gtest/gtest.h"
#include "ngraph/ngraph.hpp" #include <gtest/gtest.h>
#include "ngraph/op/util/attr_types.hpp"
#include "openvino/op/constant.hpp"
#include "visitors/visitors.hpp"
using namespace std; using namespace std;
using namespace ngraph; using namespace ov;
using ngraph::test::NodeBuilder; using ov::test::NodeBuilder;
TEST(attributes, dft_op) { TEST(attributes, dft_op) {
NodeBuilder::get_ops().register_factory<op::v7::DFT>(); NodeBuilder::get_ops().register_factory<op::v7::DFT>();
@ -27,7 +29,7 @@ TEST(attributes, dft_op) {
TEST(attributes, dft_op_signal) { TEST(attributes, dft_op_signal) {
NodeBuilder::get_ops().register_factory<op::v7::DFT>(); NodeBuilder::get_ops().register_factory<op::v7::DFT>();
auto data = make_shared<op::v0::Parameter>(element::f32, Shape{2, 10, 10, 2}); auto data = make_shared<op::v0::Parameter>(element::f32, Shape{2, 10, 10, 2});
auto signal = op::Constant::create<int64_t>(element::Type_t::i64, Shape{1}, {20}); auto signal = ov::op::v0::Constant::create<int64_t>(element::Type_t::i64, Shape{1}, {20});
auto axes = op::v0::Constant::create<int64_t>(element::i64, Shape{1}, {2}); auto axes = op::v0::Constant::create<int64_t>(element::i64, Shape{1}, {2});
auto dft = make_shared<op::v7::DFT>(data, axes, signal); auto dft = make_shared<op::v7::DFT>(data, axes, signal);

View File

@ -2,37 +2,27 @@
// SPDX-License-Identifier: Apache-2.0 // SPDX-License-Identifier: Apache-2.0
// //
/* #include "openvino/op/divide.hpp"
#include "binary_ops.hpp"
#include "ngraph/opsets/opset1.hpp"
using Type = ::testing::Types<BinaryOperatorType<ngraph::opset1::Divide, ngraph::element::f32>>; #include <gtest/gtest.h>
INSTANTIATE_TYPED_TEST_SUITE_P(visitor_with_auto_broadcast, BinaryOperatorVisitor, Type, BinaryOperatorTypeName); #include "visitors/visitors.hpp"
*/
#include "common_test_utils/visitor.hpp"
#include "gtest/gtest.h"
#include "ngraph/ngraph.hpp"
#include "ngraph/op/util/attr_types.hpp"
#include "ngraph/opsets/opset1.hpp"
using namespace std; using namespace std;
using namespace ngraph; using namespace ov;
using ngraph::test::NodeBuilder; using ov::test::NodeBuilder;
using ngraph::test::ValueMap;
TEST(attributes, divide) { TEST(attributes, divide) {
NodeBuilder::get_ops().register_factory<opset1::Divide>(); NodeBuilder::get_ops().register_factory<ov::op::v1::Divide>();
const auto in1 = make_shared<op::Parameter>(element::f32, Shape{2, 4}); const auto in1 = make_shared<ov::op::v0::Parameter>(element::f32, Shape{2, 4});
const auto in2 = make_shared<op::Parameter>(element::f32, Shape{2, 4}); const auto in2 = make_shared<ov::op::v0::Parameter>(element::f32, Shape{2, 4});
const bool pythondiv = true; const bool pythondiv = true;
const op::AutoBroadcastSpec& auto_broadcast = op::AutoBroadcastSpec(op::AutoBroadcastType::NUMPY); const op::AutoBroadcastSpec& auto_broadcast = op::AutoBroadcastSpec(op::AutoBroadcastType::NUMPY);
const auto divide = make_shared<opset1::Divide>(in1, in2, pythondiv, auto_broadcast); const auto divide = make_shared<ov::op::v1::Divide>(in1, in2, pythondiv, auto_broadcast);
NodeBuilder builder(divide, {in1, in2}); NodeBuilder builder(divide, {in1, in2});
auto g_divide = ov::as_type_ptr<opset1::Divide>(builder.create()); auto g_divide = ov::as_type_ptr<ov::op::v1::Divide>(builder.create());
const auto expected_attr_count = 2; const auto expected_attr_count = 2;
EXPECT_EQ(builder.get_value_map_size(), expected_attr_count); EXPECT_EQ(builder.get_value_map_size(), expected_attr_count);

View File

@ -2,24 +2,23 @@
// SPDX-License-Identifier: Apache-2.0 // SPDX-License-Identifier: Apache-2.0
// //
#include "common_test_utils/visitor.hpp" #include "openvino/op/einsum.hpp"
#include "gtest/gtest.h"
#include "ngraph/ngraph.hpp" #include <gtest/gtest.h>
#include "ngraph/opsets/opset1.hpp"
#include "ngraph/opsets/opset7.hpp" #include "visitors/visitors.hpp"
using namespace std; using namespace std;
using namespace ngraph; using namespace ov;
using ngraph::test::NodeBuilder; using ov::test::NodeBuilder;
using ngraph::test::ValueMap;
TEST(attributes, einsum_v7_op) { TEST(attributes, einsum_v7_op) {
NodeBuilder::get_ops().register_factory<opset7::Einsum>(); NodeBuilder::get_ops().register_factory<ov::op::v7::Einsum>();
auto input1 = make_shared<opset1::Parameter>(element::i32, Shape{2, 3}); auto input1 = make_shared<ov::op::v0::Parameter>(element::i32, Shape{2, 3});
auto input2 = make_shared<opset1::Parameter>(element::i32, Shape{3, 4}); auto input2 = make_shared<ov::op::v0::Parameter>(element::i32, Shape{3, 4});
std::string equation = "ab,bc->ac"; std::string equation = "ab,bc->ac";
auto einsum = make_shared<opset7::Einsum>(OutputVector{input1, input2}, equation); auto einsum = make_shared<ov::op::v7::Einsum>(OutputVector{input1, input2}, equation);
NodeBuilder builder(einsum, {input1, input2}); NodeBuilder builder(einsum, {input1, input2});
auto g_einsum = ov::as_type_ptr<opset7::Einsum>(builder.create()); auto g_einsum = ov::as_type_ptr<ov::op::v7::Einsum>(builder.create());
EXPECT_EQ(g_einsum->get_equation(), einsum->get_equation()); EXPECT_EQ(g_einsum->get_equation(), einsum->get_equation());
} }

View File

@ -2,29 +2,25 @@
// SPDX-License-Identifier: Apache-2.0 // SPDX-License-Identifier: Apache-2.0
// //
#include "common_test_utils/visitor.hpp" #include "openvino/op/elu.hpp"
#include "gtest/gtest.h"
#include "ngraph/ngraph.hpp" #include <gtest/gtest.h>
#include "ngraph/op/util/attr_types.hpp"
#include "ngraph/opsets/opset1.hpp" #include "visitors/visitors.hpp"
#include "ngraph/opsets/opset3.hpp"
#include "ngraph/opsets/opset4.hpp"
#include "ngraph/opsets/opset5.hpp"
using namespace std; using namespace std;
using namespace ngraph; using namespace ov;
using ngraph::test::NodeBuilder; using ov::test::NodeBuilder;
using ngraph::test::ValueMap;
TEST(attributes, elu_op) { TEST(attributes, elu_op) {
NodeBuilder::get_ops().register_factory<opset1::Elu>(); NodeBuilder::get_ops().register_factory<ov::op::v0::Elu>();
auto data = make_shared<op::Parameter>(element::f32, Shape{2, 4}); auto data = make_shared<ov::op::v0::Parameter>(element::f32, Shape{2, 4});
double alpha = 0.1; double alpha = 0.1;
const auto elu = make_shared<opset1::Elu>(data, alpha); const auto elu = make_shared<ov::op::v0::Elu>(data, alpha);
NodeBuilder builder(elu, {data}); NodeBuilder builder(elu, {data});
auto g_elu = ov::as_type_ptr<opset1::Elu>(builder.create()); auto g_elu = ov::as_type_ptr<ov::op::v0::Elu>(builder.create());
EXPECT_EQ(g_elu->get_alpha(), elu->get_alpha()); EXPECT_EQ(g_elu->get_alpha(), elu->get_alpha());
} }

View File

@ -2,68 +2,68 @@
// SPDX-License-Identifier: Apache-2.0 // SPDX-License-Identifier: Apache-2.0
// //
#include "common_test_utils/visitor.hpp" #include "openvino/op/embedding_segments_sum.hpp"
#include "gtest/gtest.h"
#include "ngraph/ngraph.hpp" #include <gtest/gtest.h>
#include "ngraph/op/util/attr_types.hpp"
#include "ngraph/opsets/opset3.hpp" #include "visitors/visitors.hpp"
using namespace std; using namespace std;
using namespace ngraph; using namespace ov;
using ngraph::test::NodeBuilder; using ov::test::NodeBuilder;
using ngraph::test::ValueMap;
TEST(visitor_without_attribute, embedding_segments_sum_op) { TEST(visitor_without_attribute, embedding_segments_sum_op) {
NodeBuilder::get_ops().register_factory<opset3::EmbeddingSegmentsSum>(); NodeBuilder::get_ops().register_factory<ov::op::v3::EmbeddingSegmentsSum>();
auto emb_table = make_shared<op::Parameter>(element::f32, Shape{5, 2, 3}); auto emb_table = make_shared<ov::op::v0::Parameter>(element::f32, Shape{5, 2, 3});
auto indices = make_shared<op::Parameter>(element::i64, Shape{4}); auto indices = make_shared<ov::op::v0::Parameter>(element::i64, Shape{4});
auto segment_ids = make_shared<op::Parameter>(element::i64, Shape{4}); auto segment_ids = make_shared<ov::op::v0::Parameter>(element::i64, Shape{4});
auto num_segments = make_shared<op::Parameter>(element::i64, Shape{}); auto num_segments = make_shared<ov::op::v0::Parameter>(element::i64, Shape{});
auto per_sample_weights = make_shared<op::Parameter>(element::f32, Shape{4}); auto per_sample_weights = make_shared<ov::op::v0::Parameter>(element::f32, Shape{4});
auto default_index = make_shared<op::Parameter>(element::i64, Shape{}); auto default_index = make_shared<ov::op::v0::Parameter>(element::i64, Shape{});
auto ess = make_shared<opset3::EmbeddingSegmentsSum>(emb_table, auto ess = make_shared<ov::op::v3::EmbeddingSegmentsSum>(emb_table,
indices, indices,
segment_ids, segment_ids,
num_segments, num_segments,
default_index, default_index,
per_sample_weights); per_sample_weights);
NodeBuilder builder(ess, {emb_table, indices, segment_ids, num_segments, default_index, per_sample_weights}); NodeBuilder builder(ess, {emb_table, indices, segment_ids, num_segments, default_index, per_sample_weights});
EXPECT_NO_THROW(auto g_ess = ov::as_type_ptr<opset3::EmbeddingSegmentsSum>(builder.create())); EXPECT_NO_THROW(auto g_ess = ov::as_type_ptr<ov::op::v3::EmbeddingSegmentsSum>(builder.create()));
const auto expected_attr_count = 0; const auto expected_attr_count = 0;
EXPECT_EQ(builder.get_value_map_size(), expected_attr_count); EXPECT_EQ(builder.get_value_map_size(), expected_attr_count);
} }
TEST(visitor_without_attribute, embedding_segments_sum_op2) { TEST(visitor_without_attribute, embedding_segments_sum_op2) {
NodeBuilder::get_ops().register_factory<opset3::EmbeddingSegmentsSum>(); NodeBuilder::get_ops().register_factory<ov::op::v3::EmbeddingSegmentsSum>();
auto emb_table = make_shared<op::Parameter>(element::f32, Shape{5, 2, 3}); auto emb_table = make_shared<ov::op::v0::Parameter>(element::f32, Shape{5, 2, 3});
auto indices = make_shared<op::Parameter>(element::i64, Shape{4}); auto indices = make_shared<ov::op::v0::Parameter>(element::i64, Shape{4});
auto segment_ids = make_shared<op::Parameter>(element::i64, Shape{4}); auto segment_ids = make_shared<ov::op::v0::Parameter>(element::i64, Shape{4});
auto num_segments = make_shared<op::Parameter>(element::i64, Shape{}); auto num_segments = make_shared<ov::op::v0::Parameter>(element::i64, Shape{});
auto default_index = make_shared<op::Parameter>(element::i64, Shape{}); auto default_index = make_shared<ov::op::v0::Parameter>(element::i64, Shape{});
auto ess = make_shared<opset3::EmbeddingSegmentsSum>(emb_table, indices, segment_ids, num_segments, default_index); auto ess =
make_shared<ov::op::v3::EmbeddingSegmentsSum>(emb_table, indices, segment_ids, num_segments, default_index);
NodeBuilder builder(ess, {emb_table, indices, segment_ids, num_segments, default_index}); NodeBuilder builder(ess, {emb_table, indices, segment_ids, num_segments, default_index});
EXPECT_NO_THROW(auto g_ess = ov::as_type_ptr<opset3::EmbeddingSegmentsSum>(builder.create())); EXPECT_NO_THROW(auto g_ess = ov::as_type_ptr<ov::op::v3::EmbeddingSegmentsSum>(builder.create()));
const auto expected_attr_count = 0; const auto expected_attr_count = 0;
EXPECT_EQ(builder.get_value_map_size(), expected_attr_count); EXPECT_EQ(builder.get_value_map_size(), expected_attr_count);
} }
TEST(visitor_without_attribute, embedding_segments_sum_op3) { TEST(visitor_without_attribute, embedding_segments_sum_op3) {
NodeBuilder::get_ops().register_factory<opset3::EmbeddingSegmentsSum>(); NodeBuilder::get_ops().register_factory<ov::op::v3::EmbeddingSegmentsSum>();
auto emb_table = make_shared<op::Parameter>(element::f32, Shape{5, 2, 3}); auto emb_table = make_shared<ov::op::v0::Parameter>(element::f32, Shape{5, 2, 3});
auto indices = make_shared<op::Parameter>(element::i64, Shape{4}); auto indices = make_shared<ov::op::v0::Parameter>(element::i64, Shape{4});
auto segment_ids = make_shared<op::Parameter>(element::i64, Shape{4}); auto segment_ids = make_shared<ov::op::v0::Parameter>(element::i64, Shape{4});
auto num_segments = make_shared<op::Parameter>(element::i64, Shape{}); auto num_segments = make_shared<ov::op::v0::Parameter>(element::i64, Shape{});
auto ess = make_shared<opset3::EmbeddingSegmentsSum>(emb_table, indices, segment_ids, num_segments); auto ess = make_shared<ov::op::v3::EmbeddingSegmentsSum>(emb_table, indices, segment_ids, num_segments);
NodeBuilder builder(ess, {emb_table, indices, segment_ids, num_segments}); NodeBuilder builder(ess, {emb_table, indices, segment_ids, num_segments});
EXPECT_NO_THROW(auto g_ess = ov::as_type_ptr<opset3::EmbeddingSegmentsSum>(builder.create())); EXPECT_NO_THROW(auto g_ess = ov::as_type_ptr<ov::op::v3::EmbeddingSegmentsSum>(builder.create()));
const auto expected_attr_count = 0; const auto expected_attr_count = 0;
EXPECT_EQ(builder.get_value_map_size(), expected_attr_count); EXPECT_EQ(builder.get_value_map_size(), expected_attr_count);

View File

@ -2,61 +2,59 @@
// SPDX-License-Identifier: Apache-2.0 // SPDX-License-Identifier: Apache-2.0
// //
#include "common_test_utils/visitor.hpp" #include <gtest/gtest.h>
#include "gtest/gtest.h"
#include "ngraph/ngraph.hpp" #include "openvino/op/embeddingbag_offsets_sum.hpp"
#include "ngraph/op/util/attr_types.hpp" #include "visitors/visitors.hpp"
#include "ngraph/opsets/opset3.hpp"
using namespace std; using namespace std;
using namespace ngraph; using namespace ov;
using ngraph::test::NodeBuilder; using ov::test::NodeBuilder;
using ngraph::test::ValueMap;
TEST(visitor_without_attribute, embedding_bag_offsets_sum_op) { TEST(visitor_without_attribute, embedding_bag_offsets_sum_op) {
NodeBuilder::get_ops().register_factory<opset3::EmbeddingBagOffsetsSum>(); NodeBuilder::get_ops().register_factory<ov::op::v3::EmbeddingBagOffsetsSum>();
auto emb_table = make_shared<op::Parameter>(element::f32, Shape{5, 2, 3}); auto emb_table = make_shared<ov::op::v0::Parameter>(element::f32, Shape{5, 2, 3});
auto indices = make_shared<op::Parameter>(element::i64, Shape{4}); auto indices = make_shared<ov::op::v0::Parameter>(element::i64, Shape{4});
auto offsets = make_shared<op::Parameter>(element::i64, Shape{4}); auto offsets = make_shared<ov::op::v0::Parameter>(element::i64, Shape{4});
auto default_index = make_shared<op::Parameter>(element::i64, Shape{}); auto default_index = make_shared<ov::op::v0::Parameter>(element::i64, Shape{});
auto per_sample_weights = make_shared<op::Parameter>(element::f32, Shape{4}); auto per_sample_weights = make_shared<ov::op::v0::Parameter>(element::f32, Shape{4});
auto ebos = auto ebos =
make_shared<opset3::EmbeddingBagOffsetsSum>(emb_table, indices, offsets, default_index, per_sample_weights); make_shared<ov::op::v3::EmbeddingBagOffsetsSum>(emb_table, indices, offsets, default_index, per_sample_weights);
NodeBuilder builder(ebos, {emb_table, indices, offsets, default_index, per_sample_weights}); NodeBuilder builder(ebos, {emb_table, indices, offsets, default_index, per_sample_weights});
EXPECT_NO_THROW(auto g_ebos = ov::as_type_ptr<opset3::EmbeddingBagOffsetsSum>(builder.create())); EXPECT_NO_THROW(auto g_ebos = ov::as_type_ptr<ov::op::v3::EmbeddingBagOffsetsSum>(builder.create()));
const auto expected_attr_count = 0; const auto expected_attr_count = 0;
EXPECT_EQ(builder.get_value_map_size(), expected_attr_count); EXPECT_EQ(builder.get_value_map_size(), expected_attr_count);
} }
TEST(visitor_without_attribute, embedding_bag_offsets_sum_op2) { TEST(visitor_without_attribute, embedding_bag_offsets_sum_op2) {
NodeBuilder::get_ops().register_factory<opset3::EmbeddingBagOffsetsSum>(); NodeBuilder::get_ops().register_factory<ov::op::v3::EmbeddingBagOffsetsSum>();
auto emb_table = make_shared<op::Parameter>(element::f32, Shape{5, 2, 3}); auto emb_table = make_shared<ov::op::v0::Parameter>(element::f32, Shape{5, 2, 3});
auto indices = make_shared<op::Parameter>(element::i64, Shape{4}); auto indices = make_shared<ov::op::v0::Parameter>(element::i64, Shape{4});
auto offsets = make_shared<op::Parameter>(element::i64, Shape{4}); auto offsets = make_shared<ov::op::v0::Parameter>(element::i64, Shape{4});
auto default_index = make_shared<op::Parameter>(element::i64, Shape{}); auto default_index = make_shared<ov::op::v0::Parameter>(element::i64, Shape{});
auto ebos = make_shared<opset3::EmbeddingBagOffsetsSum>(emb_table, indices, offsets, default_index); auto ebos = make_shared<ov::op::v3::EmbeddingBagOffsetsSum>(emb_table, indices, offsets, default_index);
NodeBuilder builder(ebos, {emb_table, indices, offsets, default_index}); NodeBuilder builder(ebos, {emb_table, indices, offsets, default_index});
EXPECT_NO_THROW(auto g_ebos = ov::as_type_ptr<opset3::EmbeddingBagOffsetsSum>(builder.create())); EXPECT_NO_THROW(auto g_ebos = ov::as_type_ptr<ov::op::v3::EmbeddingBagOffsetsSum>(builder.create()));
const auto expected_attr_count = 0; const auto expected_attr_count = 0;
EXPECT_EQ(builder.get_value_map_size(), expected_attr_count); EXPECT_EQ(builder.get_value_map_size(), expected_attr_count);
} }
TEST(visitor_without_attribute, embedding_bag_offsets_sum_op3) { TEST(visitor_without_attribute, embedding_bag_offsets_sum_op3) {
NodeBuilder::get_ops().register_factory<opset3::EmbeddingBagOffsetsSum>(); NodeBuilder::get_ops().register_factory<ov::op::v3::EmbeddingBagOffsetsSum>();
auto emb_table = make_shared<op::Parameter>(element::f32, Shape{5, 2, 3}); auto emb_table = make_shared<ov::op::v0::Parameter>(element::f32, Shape{5, 2, 3});
auto indices = make_shared<op::Parameter>(element::i64, Shape{4}); auto indices = make_shared<ov::op::v0::Parameter>(element::i64, Shape{4});
auto offsets = make_shared<op::Parameter>(element::i64, Shape{4}); auto offsets = make_shared<ov::op::v0::Parameter>(element::i64, Shape{4});
auto ebos = make_shared<opset3::EmbeddingBagOffsetsSum>(emb_table, indices, offsets); auto ebos = make_shared<ov::op::v3::EmbeddingBagOffsetsSum>(emb_table, indices, offsets);
NodeBuilder builder(ebos, {emb_table, indices, offsets}); NodeBuilder builder(ebos, {emb_table, indices, offsets});
EXPECT_NO_THROW(auto g_ebos = ov::as_type_ptr<opset3::EmbeddingBagOffsetsSum>(builder.create())); EXPECT_NO_THROW(auto g_ebos = ov::as_type_ptr<ov::op::v3::EmbeddingBagOffsetsSum>(builder.create()));
const auto expected_attr_count = 0; const auto expected_attr_count = 0;
EXPECT_EQ(builder.get_value_map_size(), expected_attr_count); EXPECT_EQ(builder.get_value_map_size(), expected_attr_count);

View File

@ -2,39 +2,39 @@
// SPDX-License-Identifier: Apache-2.0 // SPDX-License-Identifier: Apache-2.0
// //
#include "common_test_utils/visitor.hpp" #include "openvino/op/embeddingbag_packedsum.hpp"
#include "gtest/gtest.h"
#include "ngraph/ngraph.hpp" #include <gtest/gtest.h>
#include "ngraph/op/util/attr_types.hpp"
#include "ngraph/opsets/opset3.hpp" #include "openvino/op/embeddingbag_offsets_sum.hpp"
#include "visitors/visitors.hpp"
using namespace std; using namespace std;
using namespace ngraph; using namespace ov;
using ngraph::test::NodeBuilder; using ov::test::NodeBuilder;
using ngraph::test::ValueMap;
TEST(visitor_without_attribute, embedding_bag_packed_sum_op) { TEST(visitor_without_attribute, embedding_bag_packed_sum_op) {
NodeBuilder::get_ops().register_factory<opset3::EmbeddingBagOffsetsSum>(); NodeBuilder::get_ops().register_factory<ov::op::v3::EmbeddingBagOffsetsSum>();
auto emb_table = make_shared<op::Parameter>(element::f32, Shape{5, 2}); auto emb_table = make_shared<ov::op::v0::Parameter>(element::f32, Shape{5, 2});
auto indices = make_shared<op::Parameter>(element::i64, Shape{3, 4}); auto indices = make_shared<ov::op::v0::Parameter>(element::i64, Shape{3, 4});
auto per_sample_weights = make_shared<op::Parameter>(element::f32, Shape{3, 4}); auto per_sample_weights = make_shared<ov::op::v0::Parameter>(element::f32, Shape{3, 4});
auto ebps = make_shared<opset3::EmbeddingBagPackedSum>(emb_table, indices, per_sample_weights); auto ebps = make_shared<ov::op::v3::EmbeddingBagPackedSum>(emb_table, indices, per_sample_weights);
NodeBuilder builder(ebps, {emb_table, indices, per_sample_weights}); NodeBuilder builder(ebps, {emb_table, indices, per_sample_weights});
EXPECT_NO_THROW(auto g_ebps = ov::as_type_ptr<opset3::EmbeddingBagPackedSum>(builder.create())); EXPECT_NO_THROW(auto g_ebps = ov::as_type_ptr<ov::op::v3::EmbeddingBagPackedSum>(builder.create()));
const auto expected_attr_count = 0; const auto expected_attr_count = 0;
EXPECT_EQ(builder.get_value_map_size(), expected_attr_count); EXPECT_EQ(builder.get_value_map_size(), expected_attr_count);
} }
TEST(visitor_without_attribute, embedding_bag_packed_sum_op2) { TEST(visitor_without_attribute, embedding_bag_packed_sum_op2) {
NodeBuilder::get_ops().register_factory<opset3::EmbeddingBagOffsetsSum>(); NodeBuilder::get_ops().register_factory<ov::op::v3::EmbeddingBagOffsetsSum>();
auto emb_table = make_shared<op::Parameter>(element::f32, Shape{5, 2}); auto emb_table = make_shared<ov::op::v0::Parameter>(element::f32, Shape{5, 2});
auto indices = make_shared<op::Parameter>(element::i64, Shape{3, 4}); auto indices = make_shared<ov::op::v0::Parameter>(element::i64, Shape{3, 4});
auto ebps = make_shared<opset3::EmbeddingBagPackedSum>(emb_table, indices); auto ebps = make_shared<ov::op::v3::EmbeddingBagPackedSum>(emb_table, indices);
NodeBuilder builder(ebps, {emb_table, indices}); NodeBuilder builder(ebps, {emb_table, indices});
EXPECT_NO_THROW(auto g_ebps = ov::as_type_ptr<opset3::EmbeddingBagPackedSum>(builder.create())); EXPECT_NO_THROW(auto g_ebps = ov::as_type_ptr<ov::op::v3::EmbeddingBagPackedSum>(builder.create()));
const auto expected_attr_count = 0; const auto expected_attr_count = 0;
EXPECT_EQ(builder.get_value_map_size(), expected_attr_count); EXPECT_EQ(builder.get_value_map_size(), expected_attr_count);

View File

@ -2,9 +2,10 @@
// SPDX-License-Identifier: Apache-2.0 // SPDX-License-Identifier: Apache-2.0
// //
#include "binary_ops.hpp" #include "openvino/op/equal.hpp"
#include "ngraph/opsets/opset1.hpp"
using Type = ::testing::Types<BinaryOperatorType<ngraph::opset1::Equal, ngraph::element::f32>>; #include "binary_ops.hpp"
using Type = ::testing::Types<BinaryOperatorType<ov::op::v1::Equal, ov::element::f32>>;
INSTANTIATE_TYPED_TEST_SUITE_P(visitor_with_auto_broadcast, BinaryOperatorVisitor, Type, BinaryOperatorTypeName); INSTANTIATE_TYPED_TEST_SUITE_P(visitor_with_auto_broadcast, BinaryOperatorVisitor, Type, BinaryOperatorTypeName);

View File

@ -2,8 +2,10 @@
// SPDX-License-Identifier: Apache-2.0 // SPDX-License-Identifier: Apache-2.0
// //
#include "openvino/op/erf.hpp"
#include "unary_ops.hpp" #include "unary_ops.hpp"
using Type = ::testing::Types<UnaryOperatorType<ov::op::v0::Erf, ngraph::element::f32>>; using Type = ::testing::Types<UnaryOperatorType<ov::op::v0::Erf, ov::element::f32>>;
INSTANTIATE_TYPED_TEST_SUITE_P(visitor_without_atrribute, UnaryOperatorVisitor, Type, UnaryOperatorTypeName); INSTANTIATE_TYPED_TEST_SUITE_P(visitor_without_atrribute, UnaryOperatorVisitor, Type, UnaryOperatorTypeName);

View File

@ -2,8 +2,10 @@
// SPDX-License-Identifier: Apache-2.0 // SPDX-License-Identifier: Apache-2.0
// //
#include "openvino/op/exp.hpp"
#include "unary_ops.hpp" #include "unary_ops.hpp"
using Type = ::testing::Types<UnaryOperatorType<ov::op::v0::Exp, ngraph::element::f32>>; using Type = ::testing::Types<UnaryOperatorType<ov::op::v0::Exp, ov::element::f32>>;
INSTANTIATE_TYPED_TEST_SUITE_P(visitor_without_atrribute, UnaryOperatorVisitor, Type, UnaryOperatorTypeName); INSTANTIATE_TYPED_TEST_SUITE_P(visitor_without_atrribute, UnaryOperatorVisitor, Type, UnaryOperatorTypeName);

View File

@ -2,21 +2,20 @@
// SPDX-License-Identifier: Apache-2.0 // SPDX-License-Identifier: Apache-2.0
// //
#include "openvino/op/experimental_detectron_detection_output.hpp"
#include <gtest/gtest.h>
#include <vector> #include <vector>
#include "common_test_utils/visitor.hpp" #include "visitors/visitors.hpp"
#include "gtest/gtest.h"
#include "ngraph/ngraph.hpp"
#include "ngraph/op/util/attr_types.hpp"
#include "ngraph/opsets/opset6.hpp"
using namespace std; using namespace std;
using namespace ngraph; using namespace ov;
using ngraph::test::NodeBuilder; using ov::test::NodeBuilder;
using ngraph::test::ValueMap;
using Attrs = opset6::ExperimentalDetectronDetectionOutput::Attributes; using Attrs = ov::op::v6::ExperimentalDetectronDetectionOutput::Attributes;
using ExperimentalDetection = opset6::ExperimentalDetectronDetectionOutput; using ExperimentalDetection = ov::op::v6::ExperimentalDetectronDetectionOutput;
TEST(attributes, detectron_detection_output) { TEST(attributes, detectron_detection_output) {
NodeBuilder::get_ops().register_factory<ExperimentalDetection>(); NodeBuilder::get_ops().register_factory<ExperimentalDetection>();
@ -31,10 +30,10 @@ TEST(attributes, detectron_detection_output) {
attrs.post_nms_count = 2000; attrs.post_nms_count = 2000;
attrs.score_threshold = 0.05000000074505806f; attrs.score_threshold = 0.05000000074505806f;
auto rois = std::make_shared<op::Parameter>(element::f32, Shape{1000, 4}); auto rois = std::make_shared<ov::op::v0::Parameter>(element::f32, Shape{1000, 4});
auto deltas = std::make_shared<op::Parameter>(element::f32, Shape{1000, 324}); auto deltas = std::make_shared<ov::op::v0::Parameter>(element::f32, Shape{1000, 324});
auto scores = std::make_shared<op::Parameter>(element::f32, Shape{1000, 81}); auto scores = std::make_shared<ov::op::v0::Parameter>(element::f32, Shape{1000, 81});
auto im_info = std::make_shared<op::Parameter>(element::f32, Shape{1, 3}); auto im_info = std::make_shared<ov::op::v0::Parameter>(element::f32, Shape{1, 3});
auto detection = std::make_shared<ExperimentalDetection>(rois, deltas, scores, im_info, attrs); auto detection = std::make_shared<ExperimentalDetection>(rois, deltas, scores, im_info, attrs);

View File

@ -2,21 +2,20 @@
// SPDX-License-Identifier: Apache-2.0 // SPDX-License-Identifier: Apache-2.0
// //
#include "openvino/op/experimental_detectron_generate_proposals.hpp"
#include <gtest/gtest.h>
#include <vector> #include <vector>
#include "common_test_utils/visitor.hpp" #include "visitors/visitors.hpp"
#include "gtest/gtest.h"
#include "ngraph/ngraph.hpp"
#include "ngraph/op/util/attr_types.hpp"
#include "ngraph/opsets/opset6.hpp"
using namespace std; using namespace std;
using namespace ngraph; using namespace ov;
using ngraph::test::NodeBuilder; using ov::test::NodeBuilder;
using ngraph::test::ValueMap;
using ExperimentalProposals = opset6::ExperimentalDetectronGenerateProposalsSingleImage; using ExperimentalProposals = ov::op::v6::ExperimentalDetectronGenerateProposalsSingleImage;
using Attrs = opset6::ExperimentalDetectronGenerateProposalsSingleImage::Attributes; using Attrs = ov::op::v6::ExperimentalDetectronGenerateProposalsSingleImage::Attributes;
TEST(attributes, detectron_proposals) { TEST(attributes, detectron_proposals) {
NodeBuilder::get_ops().register_factory<ExperimentalProposals>(); NodeBuilder::get_ops().register_factory<ExperimentalProposals>();
@ -27,10 +26,10 @@ TEST(attributes, detectron_proposals) {
attrs.post_nms_count = 1000; attrs.post_nms_count = 1000;
attrs.pre_nms_count = 1000; attrs.pre_nms_count = 1000;
auto im_info = std::make_shared<op::Parameter>(element::f32, Shape{3}); auto im_info = std::make_shared<ov::op::v0::Parameter>(element::f32, Shape{3});
auto anchors = std::make_shared<op::Parameter>(element::f32, Shape{201600, 4}); auto anchors = std::make_shared<ov::op::v0::Parameter>(element::f32, Shape{201600, 4});
auto deltas = std::make_shared<op::Parameter>(element::f32, Shape{12, 200, 336}); auto deltas = std::make_shared<ov::op::v0::Parameter>(element::f32, Shape{12, 200, 336});
auto scores = std::make_shared<op::Parameter>(element::f32, Shape{3, 200, 336}); auto scores = std::make_shared<ov::op::v0::Parameter>(element::f32, Shape{3, 200, 336});
auto proposals = std::make_shared<ExperimentalProposals>(im_info, anchors, deltas, scores, attrs); auto proposals = std::make_shared<ExperimentalProposals>(im_info, anchors, deltas, scores, attrs);

View File

@ -2,21 +2,20 @@
// SPDX-License-Identifier: Apache-2.0 // SPDX-License-Identifier: Apache-2.0
// //
#include "openvino/op/experimental_detectron_prior_grid_generator.hpp"
#include <gtest/gtest.h>
#include <vector> #include <vector>
#include "common_test_utils/visitor.hpp" #include "visitors/visitors.hpp"
#include "gtest/gtest.h"
#include "ngraph/ngraph.hpp"
#include "ngraph/op/util/attr_types.hpp"
#include "ngraph/opsets/opset6.hpp"
using namespace std; using namespace std;
using namespace ngraph; using namespace ov;
using ngraph::test::NodeBuilder; using ov::test::NodeBuilder;
using ngraph::test::ValueMap;
using ExperimentalGenerator = opset6::ExperimentalDetectronPriorGridGenerator; using ExperimentalGenerator = ov::op::v6::ExperimentalDetectronPriorGridGenerator;
using Attrs = opset6::ExperimentalDetectronPriorGridGenerator::Attributes; using Attrs = ov::op::v6::ExperimentalDetectronPriorGridGenerator::Attributes;
TEST(attributes, detectron_prior_grid_generator) { TEST(attributes, detectron_prior_grid_generator) {
NodeBuilder::get_ops().register_factory<ExperimentalGenerator>(); NodeBuilder::get_ops().register_factory<ExperimentalGenerator>();
@ -28,9 +27,9 @@ TEST(attributes, detectron_prior_grid_generator) {
attrs.stride_x = 64; attrs.stride_x = 64;
attrs.stride_y = 64; attrs.stride_y = 64;
auto priors = std::make_shared<op::Parameter>(element::f32, Shape{3, 4}); auto priors = std::make_shared<ov::op::v0::Parameter>(element::f32, Shape{3, 4});
auto feature_map = std::make_shared<op::Parameter>(element::f32, Shape{1, 3, 100, 100}); auto feature_map = std::make_shared<ov::op::v0::Parameter>(element::f32, Shape{1, 3, 100, 100});
auto im_data = std::make_shared<op::Parameter>(element::f32, Shape{1, 3, 100, 200}); auto im_data = std::make_shared<ov::op::v0::Parameter>(element::f32, Shape{1, 3, 100, 200});
auto proposals = std::make_shared<ExperimentalGenerator>(priors, feature_map, im_data, attrs); auto proposals = std::make_shared<ExperimentalGenerator>(priors, feature_map, im_data, attrs);

View File

@ -2,26 +2,25 @@
// SPDX-License-Identifier: Apache-2.0 // SPDX-License-Identifier: Apache-2.0
// //
#include "common_test_utils/visitor.hpp" #include "openvino/op/experimental_detectron_topkrois.hpp"
#include "gtest/gtest.h"
#include "ngraph/ngraph.hpp" #include <gtest/gtest.h>
#include "ngraph/opsets/opset1.hpp"
#include "ngraph/opsets/opset6.hpp" #include "visitors/visitors.hpp"
using namespace std; using namespace std;
using namespace ngraph; using namespace ov;
using ngraph::test::NodeBuilder; using ov::test::NodeBuilder;
using ngraph::test::ValueMap;
TEST(attributes, experimental_detectron_topkrois_op) { TEST(attributes, experimental_detectron_topkrois_op) {
NodeBuilder::get_ops().register_factory<opset6::ExperimentalDetectronTopKROIs>(); NodeBuilder::get_ops().register_factory<ov::op::v6::ExperimentalDetectronTopKROIs>();
size_t num_rois = 1; size_t num_rois = 1;
auto input_rois = std::make_shared<op::Parameter>(element::f32, Shape{2, 4}); auto input_rois = std::make_shared<ov::op::v0::Parameter>(element::f32, Shape{2, 4});
auto input_probs = std::make_shared<op::Parameter>(element::f32, Shape{2}); auto input_probs = std::make_shared<ov::op::v0::Parameter>(element::f32, Shape{2});
auto topkrois = std::make_shared<op::v6::ExperimentalDetectronTopKROIs>(input_rois, input_probs, num_rois); auto topkrois = std::make_shared<op::v6::ExperimentalDetectronTopKROIs>(input_rois, input_probs, num_rois);
NodeBuilder builder(topkrois, {input_rois, input_probs}); NodeBuilder builder(topkrois, {input_rois, input_probs});
auto g_topkrois = ov::as_type_ptr<opset6::ExperimentalDetectronTopKROIs>(builder.create()); auto g_topkrois = ov::as_type_ptr<ov::op::v6::ExperimentalDetectronTopKROIs>(builder.create());
EXPECT_EQ(g_topkrois->get_max_rois(), topkrois->get_max_rois()); EXPECT_EQ(g_topkrois->get_max_rois(), topkrois->get_max_rois());
} }

View File

@ -2,32 +2,29 @@
// SPDX-License-Identifier: Apache-2.0 // SPDX-License-Identifier: Apache-2.0
// //
#include "common_test_utils/visitor.hpp" #include "openvino/op/extractimagepatches.hpp"
#include "gtest/gtest.h"
#include "ngraph/ngraph.hpp" #include <gtest/gtest.h>
#include "ngraph/op/util/attr_types.hpp"
#include "ngraph/opsets/opset1.hpp" #include "visitors/visitors.hpp"
#include "ngraph/opsets/opset3.hpp"
#include "ngraph/opsets/opset4.hpp"
#include "ngraph/opsets/opset5.hpp"
using namespace std; using namespace std;
using namespace ngraph; using namespace ov;
using ngraph::test::NodeBuilder; using ov::test::NodeBuilder;
using ngraph::test::ValueMap;
TEST(attributes, extractimagepatches_op) { TEST(attributes, extractimagepatches_op) {
NodeBuilder::get_ops().register_factory<opset3::ExtractImagePatches>(); NodeBuilder::get_ops().register_factory<ov::op::v3::ExtractImagePatches>();
auto data = make_shared<op::Parameter>(element::i32, Shape{64, 3, 10, 10}); auto data = make_shared<ov::op::v0::Parameter>(element::i32, Shape{64, 3, 10, 10});
auto sizes = Shape{3, 3}; auto sizes = Shape{3, 3};
auto strides = Strides{5, 5}; auto strides = Strides{5, 5};
auto rates = Shape{1, 1}; auto rates = Shape{1, 1};
auto padtype_padding = ngraph::op::PadType::VALID; auto padtype_padding = ngraph::op::PadType::VALID;
auto extractimagepatches = make_shared<opset3::ExtractImagePatches>(data, sizes, strides, rates, padtype_padding); auto extractimagepatches =
make_shared<ov::op::v3::ExtractImagePatches>(data, sizes, strides, rates, padtype_padding);
NodeBuilder builder(extractimagepatches, {data}); NodeBuilder builder(extractimagepatches, {data});
auto g_extractimagepatches = ov::as_type_ptr<opset3::ExtractImagePatches>(builder.create()); auto g_extractimagepatches = ov::as_type_ptr<ov::op::v3::ExtractImagePatches>(builder.create());
const auto expected_attr_count = 4; const auto expected_attr_count = 4;
EXPECT_EQ(builder.get_value_map_size(), expected_attr_count); EXPECT_EQ(builder.get_value_map_size(), expected_attr_count);

View File

@ -2,16 +2,16 @@
// SPDX-License-Identifier: Apache-2.0 // SPDX-License-Identifier: Apache-2.0
// //
#include "common_test_utils/visitor.hpp" #include "openvino/op/eye.hpp"
#include "gtest/gtest.h"
#include "ngraph/ngraph.hpp" #include <gtest/gtest.h>
#include "ngraph/op/util/attr_types.hpp"
#include "openvino/opsets/opset9.hpp" #include "openvino/op/constant.hpp"
#include "visitors/visitors.hpp"
using namespace std; using namespace std;
using namespace ov; using namespace ov;
using ngraph::test::NodeBuilder; using ov::test::NodeBuilder;
using ngraph::test::ValueMap;
TEST(attributes, eye_op) { TEST(attributes, eye_op) {
NodeBuilder::get_ops().register_factory<op::v9::Eye>(); NodeBuilder::get_ops().register_factory<op::v9::Eye>();

View File

@ -2,35 +2,31 @@
// SPDX-License-Identifier: Apache-2.0 // SPDX-License-Identifier: Apache-2.0
// //
#include "common_test_utils/visitor.hpp" #include "openvino/op/fake_quantize.hpp"
#include "gtest/gtest.h"
#include "ngraph/ngraph.hpp" #include <gtest/gtest.h>
#include "ngraph/op/util/attr_types.hpp"
#include "ngraph/opsets/opset1.hpp" #include "visitors/visitors.hpp"
#include "ngraph/opsets/opset3.hpp"
#include "ngraph/opsets/opset4.hpp"
#include "ngraph/opsets/opset5.hpp"
using namespace std; using namespace std;
using namespace ngraph; using namespace ov;
using ngraph::test::NodeBuilder; using ov::test::NodeBuilder;
using ngraph::test::ValueMap;
TEST(attributes, fake_quantize_op) { TEST(attributes, fake_quantize_op) {
NodeBuilder::get_ops().register_factory<opset1::FakeQuantize>(); NodeBuilder::get_ops().register_factory<ov::op::v0::FakeQuantize>();
const auto data = make_shared<op::Parameter>(element::f32, Shape{1, 2, 3, 4}); const auto data = make_shared<ov::op::v0::Parameter>(element::f32, Shape{1, 2, 3, 4});
const auto input_low = make_shared<op::Parameter>(element::f32, Shape{}); const auto input_low = make_shared<ov::op::v0::Parameter>(element::f32, Shape{});
const auto input_high = make_shared<op::Parameter>(element::f32, Shape{}); const auto input_high = make_shared<ov::op::v0::Parameter>(element::f32, Shape{});
const auto output_low = make_shared<op::Parameter>(element::f32, Shape{}); const auto output_low = make_shared<ov::op::v0::Parameter>(element::f32, Shape{});
const auto output_high = make_shared<op::Parameter>(element::f32, Shape{}); const auto output_high = make_shared<ov::op::v0::Parameter>(element::f32, Shape{});
auto levels = 5; auto levels = 5;
auto auto_broadcast = op::AutoBroadcastType::NUMPY; auto auto_broadcast = op::AutoBroadcastType::NUMPY;
const auto fake_quantize = const auto fake_quantize =
make_shared<op::FakeQuantize>(data, input_low, input_high, output_low, output_high, levels, auto_broadcast); make_shared<op::v0::FakeQuantize>(data, input_low, input_high, output_low, output_high, levels, auto_broadcast);
NodeBuilder builder(fake_quantize, {data, input_low, input_high, output_low, output_high}); NodeBuilder builder(fake_quantize, {data, input_low, input_high, output_low, output_high});
auto g_fake_quantize = ov::as_type_ptr<opset1::FakeQuantize>(builder.create()); auto g_fake_quantize = ov::as_type_ptr<op::v0::FakeQuantize>(builder.create());
// attribute count // attribute count
const auto expected_attr_count = 2; const auto expected_attr_count = 2;

View File

@ -2,6 +2,8 @@
// SPDX-License-Identifier: Apache-2.0 // SPDX-License-Identifier: Apache-2.0
// //
#include "openvino/op/floor.hpp"
#include "unary_ops.hpp" #include "unary_ops.hpp"
using Types = ::testing::Types<UnaryOperatorType<ov::op::v0::Floor, ngraph::element::f32>, using Types = ::testing::Types<UnaryOperatorType<ov::op::v0::Floor, ngraph::element::f32>,

View File

@ -2,9 +2,10 @@
// SPDX-License-Identifier: Apache-2.0 // SPDX-License-Identifier: Apache-2.0
// //
#include "binary_ops.hpp" #include "openvino/op/floor_mod.hpp"
#include "ngraph/opsets/opset1.hpp"
using Type = ::testing::Types<BinaryOperatorType<ngraph::opset1::FloorMod, ngraph::element::f32>>; #include "binary_ops.hpp"
using Type = ::testing::Types<BinaryOperatorType<ov::op::v1::FloorMod, ngraph::element::f32>>;
INSTANTIATE_TYPED_TEST_SUITE_P(visitor_with_auto_broadcast, BinaryOperatorVisitor, Type, BinaryOperatorTypeName); INSTANTIATE_TYPED_TEST_SUITE_P(visitor_with_auto_broadcast, BinaryOperatorVisitor, Type, BinaryOperatorTypeName);

View File

@ -4,31 +4,31 @@
#include "openvino/op/util/framework_node.hpp" #include "openvino/op/util/framework_node.hpp"
#include <gtest/gtest.h>
#include "common_test_utils/graph_comparator.hpp" #include "common_test_utils/graph_comparator.hpp"
#include "common_test_utils/visitor.hpp" #include "openvino/op/constant.hpp"
#include "gtest/gtest.h" #include "openvino/op/multiply.hpp"
#include "openvino/op/util/attr_types.hpp" #include "openvino/op/parameter.hpp"
#include "openvino/opsets/opset10.hpp" #include "visitors/visitors.hpp"
using namespace std; using namespace std;
using namespace ov; using namespace ov;
using namespace ov::opset10; using ov::test::NodeBuilder;
using ngraph::test::NodeBuilder;
using ngraph::test::ValueMap;
TEST(attributes, framework_node_op) { TEST(attributes, framework_node_op) {
NodeBuilder::get_ops().register_factory<op::util::FrameworkNode>(); NodeBuilder::get_ops().register_factory<op::util::FrameworkNode>();
auto X = make_shared<Parameter>(element::f32, Shape{1, 2, 2}); auto X = make_shared<op::v0::Parameter>(element::f32, Shape{1, 2, 2});
auto Y = make_shared<Parameter>(element::f32, Shape{1, 2, 2}); auto Y = make_shared<op::v0::Parameter>(element::f32, Shape{1, 2, 2});
auto cond = make_shared<Constant>(element::boolean, Shape{1}, true); auto cond = make_shared<op::v0::Constant>(element::boolean, Shape{1}, true);
auto cond2 = make_shared<Constant>(element::boolean, Shape{1}, false); auto cond2 = make_shared<op::v0::Constant>(element::boolean, Shape{1}, false);
auto Xt = make_shared<Parameter>(element::f32, PartialShape::dynamic()); auto Xt = make_shared<op::v0::Parameter>(element::f32, PartialShape::dynamic());
auto Yt = make_shared<Parameter>(element::f32, PartialShape::dynamic()); auto Yt = make_shared<op::v0::Parameter>(element::f32, PartialShape::dynamic());
auto Xe = make_shared<Parameter>(element::f32, PartialShape::dynamic()); auto Xe = make_shared<op::v0::Parameter>(element::f32, PartialShape::dynamic());
auto Ye = make_shared<Parameter>(element::f32, PartialShape::dynamic()); auto Ye = make_shared<op::v0::Parameter>(element::f32, PartialShape::dynamic());
auto then_op = make_shared<Multiply>(Xt, Yt); auto then_op = make_shared<op::v1::Multiply>(Xt, Yt);
auto res0 = make_shared<Result>(then_op); auto res0 = make_shared<op::v0::Result>(then_op);
auto res1 = make_shared<Result>(Xe); auto res1 = make_shared<op::v0::Result>(Xe);
auto body1 = make_shared<Model>(OutputVector{res0}, ParameterVector{Xt, Yt}); auto body1 = make_shared<Model>(OutputVector{res0}, ParameterVector{Xt, Yt});
auto body2 = make_shared<Model>(OutputVector{res1}, ParameterVector{Xe}); auto body2 = make_shared<Model>(OutputVector{res1}, ParameterVector{Xe});
auto fn_op = make_shared<op::util::FrameworkNode>(OutputVector{cond}, 0, 2); auto fn_op = make_shared<op::util::FrameworkNode>(OutputVector{cond}, 0, 2);

View File

@ -2,55 +2,54 @@
// SPDX-License-Identifier: Apache-2.0 // SPDX-License-Identifier: Apache-2.0
// //
#include "common_test_utils/visitor.hpp" #include "openvino/op/gather.hpp"
#include "gtest/gtest.h"
#include "ngraph/ngraph.hpp" #include <gtest/gtest.h>
#include "ngraph/opsets/opset1.hpp"
#include "ngraph/opsets/opset7.hpp" #include "openvino/op/constant.hpp"
#include "ngraph/opsets/opset8.hpp" #include "visitors/visitors.hpp"
using namespace std; using namespace std;
using namespace ngraph; using namespace ov;
using ngraph::test::NodeBuilder; using ov::test::NodeBuilder;
using ngraph::test::ValueMap;
TEST(attributes, gather_v1_op) { TEST(attributes, gather_v1_op) {
NodeBuilder::get_ops().register_factory<opset1::Gather>(); NodeBuilder::get_ops().register_factory<ov::op::v1::Gather>();
auto data = make_shared<opset1::Parameter>(element::i32, Shape{2, 3, 4}); auto data = make_shared<ov::op::v0::Parameter>(element::i32, Shape{2, 3, 4});
auto indices = make_shared<opset1::Parameter>(element::i32, Shape{2}); auto indices = make_shared<ov::op::v0::Parameter>(element::i32, Shape{2});
auto axis = make_shared<opset1::Constant>(element::i32, Shape{}, 2); auto axis = make_shared<ov::op::v0::Constant>(element::i32, Shape{}, 2);
auto gather = make_shared<opset1::Gather>(data, indices, axis); auto gather = make_shared<ov::op::v1::Gather>(data, indices, axis);
NodeBuilder builder(gather, {data, indices, axis}); NodeBuilder builder(gather, {data, indices, axis});
auto g_gather = ov::as_type_ptr<opset1::Gather>(builder.create()); auto g_gather = ov::as_type_ptr<ov::op::v1::Gather>(builder.create());
EXPECT_EQ(g_gather->get_batch_dims(), gather->get_batch_dims()); EXPECT_EQ(g_gather->get_batch_dims(), gather->get_batch_dims());
} }
TEST(attributes, gather_v7_op) { TEST(attributes, gather_v7_op) {
NodeBuilder::get_ops().register_factory<opset7::Gather>(); NodeBuilder::get_ops().register_factory<ov::op::v7::Gather>();
auto data = make_shared<opset1::Parameter>(element::i32, Shape{2, 3, 4}); auto data = make_shared<ov::op::v0::Parameter>(element::i32, Shape{2, 3, 4});
auto indices = make_shared<opset1::Parameter>(element::i32, Shape{2}); auto indices = make_shared<ov::op::v0::Parameter>(element::i32, Shape{2});
auto axis = make_shared<opset1::Constant>(element::i32, Shape{}, 2); auto axis = make_shared<ov::op::v0::Constant>(element::i32, Shape{}, 2);
int64_t batch_dims = 1; int64_t batch_dims = 1;
auto gather = make_shared<opset7::Gather>(data, indices, axis, batch_dims); auto gather = make_shared<ov::op::v7::Gather>(data, indices, axis, batch_dims);
NodeBuilder builder(gather, {data, indices, axis}); NodeBuilder builder(gather, {data, indices, axis});
auto g_gather = ov::as_type_ptr<opset7::Gather>(builder.create()); auto g_gather = ov::as_type_ptr<ov::op::v7::Gather>(builder.create());
EXPECT_EQ(g_gather->get_batch_dims(), gather->get_batch_dims()); EXPECT_EQ(g_gather->get_batch_dims(), gather->get_batch_dims());
} }
TEST(attributes, gather_v8_op) { TEST(attributes, gather_v8_op) {
NodeBuilder::get_ops().register_factory<opset8::Gather>(); NodeBuilder::get_ops().register_factory<ov::op::v8::Gather>();
auto data = make_shared<opset1::Parameter>(element::i32, Shape{2, 3, 4}); auto data = make_shared<ov::op::v0::Parameter>(element::i32, Shape{2, 3, 4});
auto indices = make_shared<opset1::Parameter>(element::i32, Shape{2}); auto indices = make_shared<ov::op::v0::Parameter>(element::i32, Shape{2});
auto axis = make_shared<opset1::Constant>(element::i32, Shape{}, 2); auto axis = make_shared<ov::op::v0::Constant>(element::i32, Shape{}, 2);
int64_t batch_dims = 1; int64_t batch_dims = 1;
auto gather = make_shared<opset8::Gather>(data, indices, axis, batch_dims); auto gather = make_shared<ov::op::v8::Gather>(data, indices, axis, batch_dims);
NodeBuilder builder(gather, {data, indices, axis}); NodeBuilder builder(gather, {data, indices, axis});
auto g_gather = ov::as_type_ptr<opset8::Gather>(builder.create()); auto g_gather = ov::as_type_ptr<ov::op::v8::Gather>(builder.create());
EXPECT_EQ(g_gather->get_batch_dims(), gather->get_batch_dims()); EXPECT_EQ(g_gather->get_batch_dims(), gather->get_batch_dims());
} }

View File

@ -2,26 +2,25 @@
// SPDX-License-Identifier: Apache-2.0 // SPDX-License-Identifier: Apache-2.0
// //
#include "common_test_utils/visitor.hpp" #include "openvino/op/gather_elements.hpp"
#include "gtest/gtest.h"
#include "ngraph/ngraph.hpp" #include <gtest/gtest.h>
#include "ngraph/opsets/opset1.hpp"
#include "ngraph/opsets/opset6.hpp" #include "visitors/visitors.hpp"
using namespace std; using namespace std;
using namespace ngraph; using namespace ov;
using ngraph::test::NodeBuilder; using ov::test::NodeBuilder;
using ngraph::test::ValueMap;
TEST(attributes, gather_elements_op) { TEST(attributes, gather_elements_op) {
NodeBuilder::get_ops().register_factory<opset6::GatherElements>(); NodeBuilder::get_ops().register_factory<ov::op::v6::GatherElements>();
auto arg1 = make_shared<opset1::Parameter>(element::i32, PartialShape{3}); auto arg1 = make_shared<ov::op::v0::Parameter>(element::i32, PartialShape{3});
auto arg2 = make_shared<opset1::Parameter>(element::i32, PartialShape{7}); auto arg2 = make_shared<ov::op::v0::Parameter>(element::i32, PartialShape{7});
int64_t axis = 0; int64_t axis = 0;
auto gather_el = make_shared<opset6::GatherElements>(arg1, arg2, axis); auto gather_el = make_shared<ov::op::v6::GatherElements>(arg1, arg2, axis);
NodeBuilder builder(gather_el, {arg1, arg2}); NodeBuilder builder(gather_el, {arg1, arg2});
auto g_gather_el = ov::as_type_ptr<opset6::GatherElements>(builder.create()); auto g_gather_el = ov::as_type_ptr<ov::op::v6::GatherElements>(builder.create());
EXPECT_EQ(g_gather_el->get_axis(), gather_el->get_axis()); EXPECT_EQ(g_gather_el->get_axis(), gather_el->get_axis());
} }

View File

@ -2,40 +2,38 @@
// SPDX-License-Identifier: Apache-2.0 // SPDX-License-Identifier: Apache-2.0
// //
#include "common_test_utils/visitor.hpp" #include "openvino/op/gather_nd.hpp"
#include "gtest/gtest.h"
#include "ngraph/ngraph.hpp" #include <gtest/gtest.h>
#include "ngraph/opsets/opset1.hpp"
#include "ngraph/opsets/opset5.hpp" #include "visitors/visitors.hpp"
#include "ngraph/opsets/opset8.hpp"
using namespace std; using namespace std;
using namespace ngraph; using namespace ov;
using ngraph::test::NodeBuilder; using ov::test::NodeBuilder;
using ngraph::test::ValueMap;
TEST(attributes, gather_nd_v5_op) { TEST(attributes, gather_nd_v5_op) {
NodeBuilder::get_ops().register_factory<opset5::GatherND>(); NodeBuilder::get_ops().register_factory<ov::op::v5::GatherND>();
int batch_dims = 1; int batch_dims = 1;
auto P = make_shared<op::Parameter>(element::f32, Shape{2, 3, 4}); auto P = make_shared<ov::op::v0::Parameter>(element::f32, Shape{2, 3, 4});
auto I = make_shared<op::Parameter>(element::i32, Shape{2, 1}); auto I = make_shared<ov::op::v0::Parameter>(element::i32, Shape{2, 1});
auto G = make_shared<op::v5::GatherND>(P, I, batch_dims); auto G = make_shared<op::v5::GatherND>(P, I, batch_dims);
NodeBuilder builder(G, {P, I}); NodeBuilder builder(G, {P, I});
auto g_G = ov::as_type_ptr<opset5::GatherND>(builder.create()); auto g_G = ov::as_type_ptr<ov::op::v5::GatherND>(builder.create());
EXPECT_EQ(g_G->get_batch_dims(), G->get_batch_dims()); EXPECT_EQ(g_G->get_batch_dims(), G->get_batch_dims());
} }
TEST(attributes, gather_nd_v8_op) { TEST(attributes, gather_nd_v8_op) {
NodeBuilder::get_ops().register_factory<opset8::GatherND>(); NodeBuilder::get_ops().register_factory<ov::op::v8::GatherND>();
int batch_dims = 1; int batch_dims = 1;
auto P = make_shared<op::Parameter>(element::f32, Shape{2, 3, 4}); auto P = make_shared<ov::op::v0::Parameter>(element::f32, Shape{2, 3, 4});
auto I = make_shared<op::Parameter>(element::i32, Shape{2, 1}); auto I = make_shared<ov::op::v0::Parameter>(element::i32, Shape{2, 1});
auto G = make_shared<op::v8::GatherND>(P, I, batch_dims); auto G = make_shared<op::v8::GatherND>(P, I, batch_dims);
NodeBuilder builder(G, {P, I}); NodeBuilder builder(G, {P, I});
auto g_G = ov::as_type_ptr<opset8::GatherND>(builder.create()); auto g_G = ov::as_type_ptr<ov::op::v8::GatherND>(builder.create());
EXPECT_EQ(g_G->get_batch_dims(), G->get_batch_dims()); EXPECT_EQ(g_G->get_batch_dims(), G->get_batch_dims());
} }

View File

@ -2,27 +2,26 @@
// SPDX-License-Identifier: Apache-2.0 // SPDX-License-Identifier: Apache-2.0
// //
#include "common_test_utils/visitor.hpp" #include "openvino/op/gather_tree.hpp"
#include "gtest/gtest.h"
#include "ngraph/ngraph.hpp"
#include "ngraph/op/util/attr_types.hpp"
#include "ngraph/opsets/opset1.hpp"
using namespace ngraph; #include <gtest/gtest.h>
using ngraph::test::NodeBuilder;
using ngraph::test::ValueMap; #include "visitors/visitors.hpp"
using namespace ov;
using ov::test::NodeBuilder;
TEST(attributes, gather_tree_op) { TEST(attributes, gather_tree_op) {
NodeBuilder::get_ops().register_factory<opset1::GatherTree>(); NodeBuilder::get_ops().register_factory<ov::op::v1::GatherTree>();
auto step_ids = std::make_shared<op::Parameter>(element::f32, Shape{1, 2, 3}); auto step_ids = std::make_shared<ov::op::v0::Parameter>(element::f32, Shape{1, 2, 3});
auto parent_idx = std::make_shared<op::Parameter>(element::f32, Shape{1, 2, 3}); auto parent_idx = std::make_shared<ov::op::v0::Parameter>(element::f32, Shape{1, 2, 3});
auto max_seq_len = std::make_shared<op::Parameter>(element::f32, Shape{2}); auto max_seq_len = std::make_shared<ov::op::v0::Parameter>(element::f32, Shape{2});
auto end_token = std::make_shared<op::Parameter>(element::f32, Shape{}); auto end_token = std::make_shared<ov::op::v0::Parameter>(element::f32, Shape{});
auto gather_tree = std::make_shared<opset1::GatherTree>(step_ids, parent_idx, max_seq_len, end_token); auto gather_tree = std::make_shared<ov::op::v1::GatherTree>(step_ids, parent_idx, max_seq_len, end_token);
NodeBuilder builder(gather_tree, {step_ids, parent_idx, max_seq_len, end_token}); NodeBuilder builder(gather_tree, {step_ids, parent_idx, max_seq_len, end_token});
EXPECT_NO_THROW(auto g_gather_tree = ov::as_type_ptr<opset1::GatherTree>(builder.create())); EXPECT_NO_THROW(auto g_gather_tree = ov::as_type_ptr<ov::op::v1::GatherTree>(builder.create()));
const auto expected_attr_count = 0; const auto expected_attr_count = 0;
EXPECT_EQ(builder.get_value_map_size(), expected_attr_count); EXPECT_EQ(builder.get_value_map_size(), expected_attr_count);

View File

@ -2,53 +2,53 @@
// SPDX-License-Identifier: Apache-2.0 // SPDX-License-Identifier: Apache-2.0
// //
#include "common_test_utils/visitor.hpp" #include "openvino/op/gelu.hpp"
#include "gtest/gtest.h"
#include "ngraph/ngraph.hpp" #include <gtest/gtest.h>
#include "ngraph/op/util/attr_types.hpp"
#include "ngraph/opsets/opset2.hpp" #include "openvino/op/dft.hpp"
#include "ngraph/opsets/opset7.hpp" #include "visitors/visitors.hpp"
using namespace std; using namespace std;
using namespace ngraph; using namespace ov;
using ngraph::test::NodeBuilder; using ov::test::NodeBuilder;
TEST(attributes, gelu_op_erf) { TEST(attributes, gelu_op_erf) {
NodeBuilder::get_ops().register_factory<opset7::Gelu>(); NodeBuilder::get_ops().register_factory<ov::op::v7::Gelu>();
const auto data_input = make_shared<op::Parameter>(element::f32, Shape{1, 2, 3}); const auto data_input = make_shared<ov::op::v0::Parameter>(element::f32, Shape{1, 2, 3});
const auto approximation_mode = op::GeluApproximationMode::ERF; const auto approximation_mode = op::GeluApproximationMode::ERF;
const auto gelu = make_shared<opset7::Gelu>(data_input, approximation_mode); const auto gelu = make_shared<ov::op::v7::Gelu>(data_input, approximation_mode);
NodeBuilder builder(gelu, {data_input}); NodeBuilder builder(gelu, {data_input});
auto g_gelu = ov::as_type_ptr<opset7::Gelu>(builder.create()); auto g_gelu = ov::as_type_ptr<ov::op::v7::Gelu>(builder.create());
EXPECT_EQ(g_gelu->get_approximation_mode(), gelu->get_approximation_mode()); EXPECT_EQ(g_gelu->get_approximation_mode(), gelu->get_approximation_mode());
} }
TEST(attributes, gelu_op_tanh) { TEST(attributes, gelu_op_tanh) {
NodeBuilder::get_ops().register_factory<opset7::Gelu>(); NodeBuilder::get_ops().register_factory<ov::op::v7::Gelu>();
const auto data_input = make_shared<op::Parameter>(element::f32, Shape{1, 2, 3}); const auto data_input = make_shared<ov::op::v0::Parameter>(element::f32, Shape{1, 2, 3});
const auto approximation_mode = op::GeluApproximationMode::TANH; const auto approximation_mode = op::GeluApproximationMode::TANH;
const auto gelu = make_shared<opset7::Gelu>(data_input, approximation_mode); const auto gelu = make_shared<ov::op::v7::Gelu>(data_input, approximation_mode);
NodeBuilder builder(gelu, {data_input}); NodeBuilder builder(gelu, {data_input});
auto g_gelu = ov::as_type_ptr<opset7::Gelu>(builder.create()); auto g_gelu = ov::as_type_ptr<ov::op::v7::Gelu>(builder.create());
EXPECT_EQ(g_gelu->get_approximation_mode(), gelu->get_approximation_mode()); EXPECT_EQ(g_gelu->get_approximation_mode(), gelu->get_approximation_mode());
} }
TEST(attributes, gelu_op) { TEST(attributes, gelu_op) {
NodeBuilder::get_ops().register_factory<opset7::Gelu>(); NodeBuilder::get_ops().register_factory<ov::op::v7::Gelu>();
const auto data_input = make_shared<op::Parameter>(element::f32, Shape{1, 2, 3}); const auto data_input = make_shared<ov::op::v0::Parameter>(element::f32, Shape{1, 2, 3});
const auto gelu = make_shared<opset7::Gelu>(data_input); const auto gelu = make_shared<ov::op::v7::Gelu>(data_input);
NodeBuilder builder(gelu, {data_input}); NodeBuilder builder(gelu, {data_input});
auto g_gelu = ov::as_type_ptr<opset7::Gelu>(builder.create()); auto g_gelu = ov::as_type_ptr<ov::op::v7::Gelu>(builder.create());
EXPECT_EQ(g_gelu->get_approximation_mode(), gelu->get_approximation_mode()); EXPECT_EQ(g_gelu->get_approximation_mode(), gelu->get_approximation_mode());
} }
TEST(attributes, gelu_v0_op) { TEST(attributes, gelu_v0_op) {
NodeBuilder::get_ops().register_factory<opset2::Gelu>(); NodeBuilder::get_ops().register_factory<ov::op::v0::Gelu>();
const auto data_input = make_shared<op::Parameter>(element::f32, Shape{1, 2, 3}); const auto data_input = make_shared<ov::op::v0::Parameter>(element::f32, Shape{1, 2, 3});
const auto gelu = make_shared<opset2::Gelu>(data_input); const auto gelu = make_shared<ov::op::v0::Gelu>(data_input);
NodeBuilder builder(gelu, {data_input}); NodeBuilder builder(gelu, {data_input});
const auto expected_attr_count = 0; const auto expected_attr_count = 0;
EXPECT_NO_THROW(auto g_gelu = ov::as_type_ptr<op::v7::DFT>(builder.create())); EXPECT_NO_THROW(auto g_gelu = ov::as_type_ptr<op::v7::DFT>(builder.create()));

View File

@ -2,21 +2,20 @@
// SPDX-License-Identifier: Apache-2.0 // SPDX-License-Identifier: Apache-2.0
// //
#include "openvino/op/generate_proposals.hpp"
#include <gtest/gtest.h>
#include <vector> #include <vector>
#include "common_test_utils/visitor.hpp" #include "visitors/visitors.hpp"
#include "gtest/gtest.h"
#include "ngraph/ngraph.hpp"
#include "ngraph/op/util/attr_types.hpp"
#include "ngraph/opsets/opset9.hpp"
using namespace std; using namespace std;
using namespace ngraph; using namespace ov;
using ngraph::test::NodeBuilder; using ov::test::NodeBuilder;
using ngraph::test::ValueMap;
using GenerateProposals = opset9::GenerateProposals; using GenerateProposals = ov::op::v9::GenerateProposals;
using Attrs = opset9::GenerateProposals::Attributes; using Attrs = ov::op::v9::GenerateProposals::Attributes;
TEST(attributes, generate_proposals) { TEST(attributes, generate_proposals) {
NodeBuilder::get_ops().register_factory<GenerateProposals>(); NodeBuilder::get_ops().register_factory<GenerateProposals>();
@ -29,10 +28,10 @@ TEST(attributes, generate_proposals) {
attrs.normalized = true; attrs.normalized = true;
attrs.nms_eta = 1.0f; attrs.nms_eta = 1.0f;
auto im_info = std::make_shared<op::Parameter>(element::f32, Shape{1, 4}); auto im_info = std::make_shared<ov::op::v0::Parameter>(element::f32, Shape{1, 4});
auto anchors = std::make_shared<op::Parameter>(element::f32, Shape{200, 336, 3, 4}); auto anchors = std::make_shared<ov::op::v0::Parameter>(element::f32, Shape{200, 336, 3, 4});
auto deltas = std::make_shared<op::Parameter>(element::f32, Shape{1, 12, 200, 336}); auto deltas = std::make_shared<ov::op::v0::Parameter>(element::f32, Shape{1, 12, 200, 336});
auto scores = std::make_shared<op::Parameter>(element::f32, Shape{1, 3, 200, 336}); auto scores = std::make_shared<ov::op::v0::Parameter>(element::f32, Shape{1, 3, 200, 336});
auto proposals = std::make_shared<GenerateProposals>(im_info, anchors, deltas, scores, attrs); auto proposals = std::make_shared<GenerateProposals>(im_info, anchors, deltas, scores, attrs);

View File

@ -2,9 +2,10 @@
// SPDX-License-Identifier: Apache-2.0 // SPDX-License-Identifier: Apache-2.0
// //
#include "binary_ops.hpp" #include "openvino/op/greater.hpp"
#include "ngraph/opsets/opset1.hpp"
using Type = ::testing::Types<BinaryOperatorType<ngraph::opset1::Greater, ngraph::element::f32>>; #include "binary_ops.hpp"
using Type = ::testing::Types<BinaryOperatorType<ov::op::v1::Greater, ov::element::f32>>;
INSTANTIATE_TYPED_TEST_SUITE_P(visitor_with_auto_broadcast, BinaryOperatorVisitor, Type, BinaryOperatorTypeName); INSTANTIATE_TYPED_TEST_SUITE_P(visitor_with_auto_broadcast, BinaryOperatorVisitor, Type, BinaryOperatorTypeName);

View File

@ -3,8 +3,8 @@
// //
#include "binary_ops.hpp" #include "binary_ops.hpp"
#include "ngraph/opsets/opset1.hpp" #include "openvino/op/greater_eq.hpp"
using Type = ::testing::Types<BinaryOperatorType<ngraph::opset1::GreaterEqual, ngraph::element::f32>>; using Type = ::testing::Types<BinaryOperatorType<ov::op::v1::GreaterEqual, ov::element::f32>>;
INSTANTIATE_TYPED_TEST_SUITE_P(visitor_with_auto_broadcast, BinaryOperatorVisitor, Type, BinaryOperatorTypeName); INSTANTIATE_TYPED_TEST_SUITE_P(visitor_with_auto_broadcast, BinaryOperatorVisitor, Type, BinaryOperatorTypeName);

View File

@ -2,24 +2,24 @@
// SPDX-License-Identifier: Apache-2.0 // SPDX-License-Identifier: Apache-2.0
// //
#include "common_test_utils/visitor.hpp" #include "openvino/op/grid_sample.hpp"
#include "gtest/gtest.h"
#include "ngraph/ngraph.hpp" #include <gtest/gtest.h>
#include "ngraph/op/util/attr_types.hpp"
#include "openvino/opsets/opset9.hpp" #include "visitors/visitors.hpp"
using namespace std; using namespace std;
using namespace ov; using namespace ov;
using ngraph::test::NodeBuilder; using ov::test::NodeBuilder;
TEST(attributes, grid_sample_defaults) { TEST(attributes, grid_sample_defaults) {
NodeBuilder::get_ops().register_factory<opset9::GridSample>(); NodeBuilder::get_ops().register_factory<ov::op::v9::GridSample>();
const auto data = make_shared<opset9::Parameter>(element::f32, Shape{1, 3, 10, 10}); const auto data = make_shared<ov::op::v0::Parameter>(element::f32, Shape{1, 3, 10, 10});
const auto grid = make_shared<opset9::Parameter>(element::f32, Shape{1, 5, 5, 2}); const auto grid = make_shared<ov::op::v0::Parameter>(element::f32, Shape{1, 5, 5, 2});
const auto op = make_shared<opset9::GridSample>(data, grid, opset9::GridSample::Attributes{}); const auto op = make_shared<ov::op::v9::GridSample>(data, grid, ov::op::v9::GridSample::Attributes{});
NodeBuilder builder(op, {data, grid}); NodeBuilder builder(op, {data, grid});
EXPECT_NO_THROW(auto g_op = ov::as_type_ptr<opset9::GridSample>(builder.create())); EXPECT_NO_THROW(auto g_op = ov::as_type_ptr<ov::op::v9::GridSample>(builder.create()));
const auto expected_attr_count = 3; const auto expected_attr_count = 3;
EXPECT_EQ(builder.get_value_map_size(), expected_attr_count); EXPECT_EQ(builder.get_value_map_size(), expected_attr_count);

View File

@ -2,25 +2,25 @@
// SPDX-License-Identifier: Apache-2.0 // SPDX-License-Identifier: Apache-2.0
// //
#include "common_test_utils/visitor.hpp" #include "openvino/op/grn.hpp"
#include "gtest/gtest.h"
#include "ngraph/ngraph.hpp" #include <gtest/gtest.h>
#include "ngraph/op/util/attr_types.hpp"
#include "ngraph/opsets/opset1.hpp" #include "visitors/visitors.hpp"
using namespace std; using namespace std;
using namespace ngraph; using namespace ov;
using ngraph::test::NodeBuilder; using ov::test::NodeBuilder;
TEST(attributes, grn_op) { TEST(attributes, grn_op) {
NodeBuilder::get_ops().register_factory<opset1::GRN>(); NodeBuilder::get_ops().register_factory<ov::op::v0::GRN>();
auto data = make_shared<op::Parameter>(element::f32, Shape{2, 3, 4, 5}); auto data = make_shared<ov::op::v0::Parameter>(element::f32, Shape{2, 3, 4, 5});
float bias = 1.25f; float bias = 1.25f;
auto grn = make_shared<opset1::GRN>(data, bias); auto grn = make_shared<ov::op::v0::GRN>(data, bias);
NodeBuilder builder(grn, {data}); NodeBuilder builder(grn, {data});
auto g_grn = ov::as_type_ptr<opset1::GRN>(builder.create()); auto g_grn = ov::as_type_ptr<ov::op::v0::GRN>(builder.create());
const auto expected_attr_count = 1; const auto expected_attr_count = 1;
EXPECT_EQ(builder.get_value_map_size(), expected_attr_count); EXPECT_EQ(builder.get_value_map_size(), expected_attr_count);

View File

@ -2,29 +2,25 @@
// SPDX-License-Identifier: Apache-2.0 // SPDX-License-Identifier: Apache-2.0
// //
#include "common_test_utils/visitor.hpp" #include "openvino/op/group_conv.hpp"
#include "gtest/gtest.h"
#include "ngraph/ngraph.hpp" #include <gtest/gtest.h>
#include "ngraph/op/util/attr_types.hpp"
#include "ngraph/opsets/opset1.hpp" #include "visitors/visitors.hpp"
#include "ngraph/opsets/opset3.hpp"
#include "ngraph/opsets/opset4.hpp"
#include "ngraph/opsets/opset5.hpp"
using namespace std; using namespace std;
using namespace ngraph; using namespace ov;
using ngraph::test::NodeBuilder; using ov::test::NodeBuilder;
using ngraph::test::ValueMap;
TEST(attributes, group_conv_op) { TEST(attributes, group_conv_op) {
NodeBuilder::get_ops().register_factory<opset1::GroupConvolution>(); NodeBuilder::get_ops().register_factory<ov::op::v1::GroupConvolution>();
auto data = make_shared<op::Parameter>(element::f32, Shape{1, 12, 224, 224}); auto data = make_shared<ov::op::v0::Parameter>(element::f32, Shape{1, 12, 224, 224});
auto filters = make_shared<op::Parameter>(element::f32, Shape{4, 1, 3, 5, 5}); auto filters = make_shared<ov::op::v0::Parameter>(element::f32, Shape{4, 1, 3, 5, 5});
auto strides = Strides{1, 1}; auto strides = Strides{1, 1};
auto pads_begin = CoordinateDiff{1, 2}; auto pads_begin = CoordinateDiff{1, 2};
auto pads_end = CoordinateDiff{1, 2}; auto pads_end = CoordinateDiff{1, 2};
auto dilations = Strides{1, 1}; auto dilations = Strides{1, 1};
auto group_conv = make_shared<opset1::GroupConvolution>(data, auto group_conv = make_shared<op::v1::GroupConvolution>(data,
filters, filters,
strides, strides,
pads_begin, pads_begin,
@ -32,7 +28,7 @@ TEST(attributes, group_conv_op) {
dilations, dilations,
op::PadType::VALID); op::PadType::VALID);
NodeBuilder builder(group_conv, {data, filters}); NodeBuilder builder(group_conv, {data, filters});
auto g_group_conv = ov::as_type_ptr<opset1::GroupConvolution>(builder.create()); auto g_group_conv = ov::as_type_ptr<op::v1::GroupConvolution>(builder.create());
EXPECT_EQ(g_group_conv->get_strides(), group_conv->get_strides()); EXPECT_EQ(g_group_conv->get_strides(), group_conv->get_strides());
EXPECT_EQ(g_group_conv->get_pads_begin(), group_conv->get_pads_begin()); EXPECT_EQ(g_group_conv->get_pads_begin(), group_conv->get_pads_begin());
EXPECT_EQ(g_group_conv->get_pads_end(), group_conv->get_pads_end()); EXPECT_EQ(g_group_conv->get_pads_end(), group_conv->get_pads_end());
@ -41,10 +37,10 @@ TEST(attributes, group_conv_op) {
} }
TEST(attributes, group_conv_backprop_data_op) { TEST(attributes, group_conv_backprop_data_op) {
NodeBuilder::get_ops().register_factory<opset1::GroupConvolutionBackpropData>(); NodeBuilder::get_ops().register_factory<op::v1::GroupConvolutionBackpropData>();
const auto data = make_shared<op::Parameter>(element::f32, Shape{1, 20, 224, 224}); const auto data = make_shared<ov::op::v0::Parameter>(element::f32, Shape{1, 20, 224, 224});
const auto filter = make_shared<op::Parameter>(element::f32, Shape{4, 5, 2, 3, 3}); const auto filter = make_shared<ov::op::v0::Parameter>(element::f32, Shape{4, 5, 2, 3, 3});
const auto output_shape = make_shared<op::Parameter>(element::i32, Shape{2}); const auto output_shape = make_shared<ov::op::v0::Parameter>(element::i32, Shape{2});
const auto strides = Strides{2, 1}; const auto strides = Strides{2, 1};
const auto pads_begin = CoordinateDiff{3, 4}; const auto pads_begin = CoordinateDiff{3, 4};
@ -53,7 +49,7 @@ TEST(attributes, group_conv_backprop_data_op) {
const auto auto_pad = op::PadType::EXPLICIT; const auto auto_pad = op::PadType::EXPLICIT;
const auto output_padding = CoordinateDiff{3, 4}; const auto output_padding = CoordinateDiff{3, 4};
const auto gcbd = make_shared<opset1::GroupConvolutionBackpropData>(data, const auto gcbd = make_shared<op::v1::GroupConvolutionBackpropData>(data,
filter, filter,
output_shape, output_shape,
strides, strides,
@ -63,7 +59,7 @@ TEST(attributes, group_conv_backprop_data_op) {
auto_pad, auto_pad,
output_padding); output_padding);
NodeBuilder builder(gcbd, {data, filter}); NodeBuilder builder(gcbd, {data, filter});
const auto g_gcbd = ov::as_type_ptr<opset1::GroupConvolutionBackpropData>(builder.create()); const auto g_gcbd = ov::as_type_ptr<op::v1::GroupConvolutionBackpropData>(builder.create());
EXPECT_EQ(g_gcbd->get_strides(), gcbd->get_strides()); EXPECT_EQ(g_gcbd->get_strides(), gcbd->get_strides());
EXPECT_EQ(g_gcbd->get_pads_begin(), gcbd->get_pads_begin()); EXPECT_EQ(g_gcbd->get_pads_begin(), gcbd->get_pads_begin());

View File

@ -2,25 +2,26 @@
// SPDX-License-Identifier: Apache-2.0 // SPDX-License-Identifier: Apache-2.0
// //
#include "common_test_utils/visitor.hpp" #include "openvino/op/group_normalization.hpp"
#include "gtest/gtest.h"
#include "ngraph/ngraph.hpp" #include <gtest/gtest.h>
#include "ngraph/op/util/attr_types.hpp"
#include "openvino/opsets/opset12.hpp" #include "openvino/op/unique.hpp"
#include "visitors/visitors.hpp"
using namespace std; using namespace std;
using namespace ov; using namespace ov;
using ngraph::test::NodeBuilder; using ov::test::NodeBuilder;
TEST(attributes, group_normalization) { TEST(attributes, group_normalization) {
NodeBuilder::get_ops().register_factory<opset12::Unique>(); NodeBuilder::get_ops().register_factory<ov::op::v10::Unique>();
const auto data = make_shared<opset12::Parameter>(element::f32, Shape{1, 3, 10, 10}); const auto data = make_shared<ov::op::v0::Parameter>(element::f32, Shape{1, 3, 10, 10});
const auto scale = make_shared<opset12::Parameter>(element::f32, Shape{3}); const auto scale = make_shared<ov::op::v0::Parameter>(element::f32, Shape{3});
const auto bias = make_shared<opset12::Parameter>(element::f32, Shape{3}); const auto bias = make_shared<ov::op::v0::Parameter>(element::f32, Shape{3});
const auto op = make_shared<opset12::GroupNormalization>(data, scale, bias, 3, 0.00001f); const auto op = make_shared<ov::op::v12::GroupNormalization>(data, scale, bias, 3, 0.00001f);
NodeBuilder builder(op); NodeBuilder builder(op);
auto g_unique = ov::as_type_ptr<opset12::GroupNormalization>(builder.create()); auto g_unique = ov::as_type_ptr<ov::op::v12::GroupNormalization>(builder.create());
constexpr auto expected_attr_count = 2; constexpr auto expected_attr_count = 2;
EXPECT_EQ(builder.get_value_map_size(), expected_attr_count); EXPECT_EQ(builder.get_value_map_size(), expected_attr_count);

View File

@ -2,33 +2,29 @@
// SPDX-License-Identifier: Apache-2.0 // SPDX-License-Identifier: Apache-2.0
// //
#include "common_test_utils/visitor.hpp" #include "openvino/op/gru_cell.hpp"
#include "gtest/gtest.h"
#include "ngraph/ngraph.hpp" #include <gtest/gtest.h>
#include "ngraph/op/util/attr_types.hpp"
#include "ngraph/opsets/opset1.hpp" #include "visitors/visitors.hpp"
#include "ngraph/opsets/opset3.hpp"
#include "ngraph/opsets/opset4.hpp"
#include "ngraph/opsets/opset5.hpp"
using namespace std; using namespace std;
using namespace ngraph; using namespace ov;
using ngraph::test::NodeBuilder; using ov::test::NodeBuilder;
using ngraph::test::ValueMap;
TEST(attributes, gru_cell_op) { TEST(attributes, gru_cell_op) {
NodeBuilder::get_ops().register_factory<opset5::GRUCell>(); NodeBuilder::get_ops().register_factory<ov::op::v3::GRUCell>();
auto X = make_shared<op::Parameter>(element::f32, Shape{2, 3}); auto X = make_shared<ov::op::v0::Parameter>(element::f32, Shape{2, 3});
auto W = make_shared<op::Parameter>(element::f32, Shape{9, 3}); auto W = make_shared<ov::op::v0::Parameter>(element::f32, Shape{9, 3});
auto R = make_shared<op::Parameter>(element::f32, Shape{9, 3}); auto R = make_shared<ov::op::v0::Parameter>(element::f32, Shape{9, 3});
const auto initial_hidden_state = make_shared<op::Parameter>(element::f32, Shape{2, 3}); const auto initial_hidden_state = make_shared<ov::op::v0::Parameter>(element::f32, Shape{2, 3});
const auto hidden_size = 3; const auto hidden_size = 3;
const std::vector<std::string> activations = {"tanh", "sigmoid"}; const std::vector<std::string> activations = {"tanh", "sigmoid"};
auto activations_alpha = std::vector<float>{1.0, 1.5}; auto activations_alpha = std::vector<float>{1.0, 1.5};
auto activations_beta = std::vector<float>{2.0, 1.0}; auto activations_beta = std::vector<float>{2.0, 1.0};
const float clip = 0.5f; const float clip = 0.5f;
const auto gru_cell = make_shared<opset5::GRUCell>(X, const auto gru_cell = make_shared<ov::op::v3::GRUCell>(X,
initial_hidden_state, initial_hidden_state,
W, W,
R, R,
@ -39,7 +35,7 @@ TEST(attributes, gru_cell_op) {
clip, clip,
false); false);
NodeBuilder builder(gru_cell, {X, initial_hidden_state, W, R}); NodeBuilder builder(gru_cell, {X, initial_hidden_state, W, R});
auto g_gru_cell = ov::as_type_ptr<opset5::GRUCell>(builder.create()); auto g_gru_cell = ov::as_type_ptr<ov::op::v3::GRUCell>(builder.create());
EXPECT_EQ(g_gru_cell->get_hidden_size(), gru_cell->get_hidden_size()); EXPECT_EQ(g_gru_cell->get_hidden_size(), gru_cell->get_hidden_size());
EXPECT_EQ(g_gru_cell->get_activations(), gru_cell->get_activations()); EXPECT_EQ(g_gru_cell->get_activations(), gru_cell->get_activations());
@ -50,19 +46,19 @@ TEST(attributes, gru_cell_op) {
} }
TEST(attributes, gru_cell_op2) { TEST(attributes, gru_cell_op2) {
NodeBuilder::get_ops().register_factory<opset5::GRUCell>(); NodeBuilder::get_ops().register_factory<ov::op::v3::GRUCell>();
auto X = make_shared<op::Parameter>(element::f32, Shape{2, 3}); auto X = make_shared<ov::op::v0::Parameter>(element::f32, Shape{2, 3});
auto W = make_shared<op::Parameter>(element::f32, Shape{9, 3}); auto W = make_shared<ov::op::v0::Parameter>(element::f32, Shape{9, 3});
auto R = make_shared<op::Parameter>(element::f32, Shape{9, 3}); auto R = make_shared<ov::op::v0::Parameter>(element::f32, Shape{9, 3});
const auto initial_hidden_state = make_shared<op::Parameter>(element::f32, Shape{2, 3}); const auto initial_hidden_state = make_shared<ov::op::v0::Parameter>(element::f32, Shape{2, 3});
auto B = make_shared<op::Parameter>(element::f32, Shape{9}); auto B = make_shared<ov::op::v0::Parameter>(element::f32, Shape{9});
const auto hidden_size = 3; const auto hidden_size = 3;
const std::vector<std::string> activations = {"tanh", "sigmoid"}; const std::vector<std::string> activations = {"tanh", "sigmoid"};
auto activations_alpha = std::vector<float>{1.0, 1.5}; auto activations_alpha = std::vector<float>{1.0, 1.5};
auto activations_beta = std::vector<float>{2.0, 1.0}; auto activations_beta = std::vector<float>{2.0, 1.0};
const float clip = 0.5f; const float clip = 0.5f;
const auto gru_cell = make_shared<opset5::GRUCell>(X, const auto gru_cell = make_shared<ov::op::v3::GRUCell>(X,
initial_hidden_state, initial_hidden_state,
W, W,
R, R,
@ -74,7 +70,7 @@ TEST(attributes, gru_cell_op2) {
clip, clip,
false); false);
NodeBuilder builder(gru_cell, {X, initial_hidden_state, W, R, B}); NodeBuilder builder(gru_cell, {X, initial_hidden_state, W, R, B});
auto g_gru_cell = ov::as_type_ptr<opset5::GRUCell>(builder.create()); auto g_gru_cell = ov::as_type_ptr<ov::op::v3::GRUCell>(builder.create());
EXPECT_EQ(g_gru_cell->get_hidden_size(), gru_cell->get_hidden_size()); EXPECT_EQ(g_gru_cell->get_hidden_size(), gru_cell->get_hidden_size());
EXPECT_EQ(g_gru_cell->get_activations(), gru_cell->get_activations()); EXPECT_EQ(g_gru_cell->get_activations(), gru_cell->get_activations());

View File

@ -2,22 +2,18 @@
// SPDX-License-Identifier: Apache-2.0 // SPDX-License-Identifier: Apache-2.0
// //
#include "common_test_utils/visitor.hpp" #include "openvino/op/gru_sequence.hpp"
#include "gtest/gtest.h"
#include "ngraph/ngraph.hpp" #include <gtest/gtest.h>
#include "ngraph/op/util/attr_types.hpp"
#include "ngraph/opsets/opset1.hpp" #include "visitors/visitors.hpp"
#include "ngraph/opsets/opset3.hpp"
#include "ngraph/opsets/opset4.hpp"
#include "ngraph/opsets/opset5.hpp"
using namespace std; using namespace std;
using namespace ngraph; using namespace ov;
using ngraph::test::NodeBuilder; using ov::test::NodeBuilder;
using ngraph::test::ValueMap;
TEST(attributes, gru_sequence_op) { TEST(attributes, gru_sequence_op) {
NodeBuilder::get_ops().register_factory<opset5::GRUSequence>(); NodeBuilder::get_ops().register_factory<ov::op::v5::GRUSequence>();
const size_t batch_size = 4; const size_t batch_size = 4;
const size_t num_directions = 2; const size_t num_directions = 2;
@ -25,15 +21,16 @@ TEST(attributes, gru_sequence_op) {
const size_t input_size = 16; const size_t input_size = 16;
const size_t hidden_size = 64; const size_t hidden_size = 64;
const auto X = make_shared<op::Parameter>(element::f32, Shape{batch_size, seq_length, input_size}); const auto X = make_shared<ov::op::v0::Parameter>(element::f32, Shape{batch_size, seq_length, input_size});
const auto initial_hidden_state = const auto initial_hidden_state =
make_shared<op::Parameter>(element::f32, Shape{batch_size, num_directions, hidden_size}); make_shared<ov::op::v0::Parameter>(element::f32, Shape{batch_size, num_directions, hidden_size});
const auto initial_cell_state = const auto initial_cell_state =
make_shared<op::Parameter>(element::f32, Shape{batch_size, num_directions, hidden_size}); make_shared<ov::op::v0::Parameter>(element::f32, Shape{batch_size, num_directions, hidden_size});
const auto sequence_lengths = make_shared<op::Parameter>(element::i32, Shape{batch_size}); const auto sequence_lengths = make_shared<ov::op::v0::Parameter>(element::i32, Shape{batch_size});
const auto W = make_shared<op::Parameter>(element::f32, Shape{num_directions, 3 * hidden_size, input_size}); const auto W = make_shared<ov::op::v0::Parameter>(element::f32, Shape{num_directions, 3 * hidden_size, input_size});
const auto R = make_shared<op::Parameter>(element::f32, Shape{num_directions, 3 * hidden_size, hidden_size}); const auto R =
const auto B = make_shared<op::Parameter>(element::f32, Shape{num_directions, 3 * hidden_size}); make_shared<ov::op::v0::Parameter>(element::f32, Shape{num_directions, 3 * hidden_size, hidden_size});
const auto B = make_shared<ov::op::v0::Parameter>(element::f32, Shape{num_directions, 3 * hidden_size});
const auto gru_direction = op::RecurrentSequenceDirection::BIDIRECTIONAL; const auto gru_direction = op::RecurrentSequenceDirection::BIDIRECTIONAL;
const std::vector<float> activations_alpha = {1, 2}; const std::vector<float> activations_alpha = {1, 2};
@ -41,7 +38,7 @@ TEST(attributes, gru_sequence_op) {
const std::vector<std::string> activations = {"tanh", "sigmoid"}; const std::vector<std::string> activations = {"tanh", "sigmoid"};
const float clip_threshold = 0.5f; const float clip_threshold = 0.5f;
const auto gru_sequence = make_shared<opset5::GRUSequence>(X, const auto gru_sequence = make_shared<ov::op::v5::GRUSequence>(X,
initial_hidden_state, initial_hidden_state,
sequence_lengths, sequence_lengths,
W, W,
@ -54,7 +51,7 @@ TEST(attributes, gru_sequence_op) {
activations_beta, activations_beta,
clip_threshold); clip_threshold);
NodeBuilder builder(gru_sequence, {X, initial_hidden_state, sequence_lengths, W, R, B}); NodeBuilder builder(gru_sequence, {X, initial_hidden_state, sequence_lengths, W, R, B});
auto g_gru_sequence = ov::as_type_ptr<opset5::GRUSequence>(builder.create()); auto g_gru_sequence = ov::as_type_ptr<ov::op::v5::GRUSequence>(builder.create());
EXPECT_EQ(g_gru_sequence->get_hidden_size(), gru_sequence->get_hidden_size()); EXPECT_EQ(g_gru_sequence->get_hidden_size(), gru_sequence->get_hidden_size());
EXPECT_EQ(g_gru_sequence->get_activations(), gru_sequence->get_activations()); EXPECT_EQ(g_gru_sequence->get_activations(), gru_sequence->get_activations());

View File

@ -2,25 +2,25 @@
// SPDX-License-Identifier: Apache-2.0 // SPDX-License-Identifier: Apache-2.0
// //
#include "common_test_utils/visitor.hpp" #include "openvino/op/hard_sigmoid.hpp"
#include "gtest/gtest.h"
#include "ngraph/ngraph.hpp" #include <gtest/gtest.h>
#include "ngraph/op/util/attr_types.hpp"
#include "ngraph/opsets/opset1.hpp" #include "visitors/visitors.hpp"
using namespace std; using namespace std;
using namespace ngraph; using namespace ov;
using ngraph::test::NodeBuilder; using ov::test::NodeBuilder;
TEST(attributes, hardsigmoid_op) { TEST(attributes, hardsigmoid_op) {
NodeBuilder::get_ops().register_factory<opset1::HardSigmoid>(); NodeBuilder::get_ops().register_factory<ov::op::v0::HardSigmoid>();
const auto data = make_shared<op::Parameter>(element::f32, Shape{2, 5}); const auto data = make_shared<ov::op::v0::Parameter>(element::f32, Shape{2, 5});
const auto alpha = make_shared<op::Parameter>(element::f32, Shape{}); const auto alpha = make_shared<ov::op::v0::Parameter>(element::f32, Shape{});
const auto beta = make_shared<op::Parameter>(element::f32, Shape{}); const auto beta = make_shared<ov::op::v0::Parameter>(element::f32, Shape{});
const auto hardsigmoid = make_shared<opset1::HardSigmoid>(data, alpha, beta); const auto hardsigmoid = make_shared<ov::op::v0::HardSigmoid>(data, alpha, beta);
NodeBuilder builder(hardsigmoid, {data, alpha, beta}); NodeBuilder builder(hardsigmoid, {data, alpha, beta});
EXPECT_NO_THROW(auto g_hardsigmoid = ov::as_type_ptr<opset1::HardSigmoid>(builder.create())); EXPECT_NO_THROW(auto g_hardsigmoid = ov::as_type_ptr<ov::op::v0::HardSigmoid>(builder.create()));
const auto expected_attr_count = 0; const auto expected_attr_count = 0;
EXPECT_EQ(builder.get_value_map_size(), expected_attr_count); EXPECT_EQ(builder.get_value_map_size(), expected_attr_count);

View File

@ -2,8 +2,10 @@
// SPDX-License-Identifier: Apache-2.0 // SPDX-License-Identifier: Apache-2.0
// //
#include "openvino/op/hsigmoid.hpp"
#include "unary_ops.hpp" #include "unary_ops.hpp"
using Type = ::testing::Types<UnaryOperatorType<ov::op::v5::HSigmoid, ngraph::element::f32>>; using Type = ::testing::Types<UnaryOperatorType<ov::op::v5::HSigmoid, ov::element::f32>>;
INSTANTIATE_TYPED_TEST_SUITE_P(visitor_without_atrribute, UnaryOperatorVisitor, Type, UnaryOperatorTypeName); INSTANTIATE_TYPED_TEST_SUITE_P(visitor_without_atrribute, UnaryOperatorVisitor, Type, UnaryOperatorTypeName);

View File

@ -2,8 +2,10 @@
// SPDX-License-Identifier: Apache-2.0 // SPDX-License-Identifier: Apache-2.0
// //
#include "openvino/op/hswish.hpp"
#include "unary_ops.hpp" #include "unary_ops.hpp"
using Type = ::testing::Types<UnaryOperatorType<ov::op::v4::HSwish, ngraph::element::f32>>; using Type = ::testing::Types<UnaryOperatorType<ov::op::v4::HSwish, ov::element::f32>>;
INSTANTIATE_TYPED_TEST_SUITE_P(visitor_without_atrribute, UnaryOperatorVisitor, Type, UnaryOperatorTypeName); INSTANTIATE_TYPED_TEST_SUITE_P(visitor_without_atrribute, UnaryOperatorVisitor, Type, UnaryOperatorTypeName);

View File

@ -2,19 +2,21 @@
// SPDX-License-Identifier: Apache-2.0 // SPDX-License-Identifier: Apache-2.0
// //
#include "common_test_utils/visitor.hpp" #include "openvino/op/idft.hpp"
#include "gtest/gtest.h"
#include "ngraph/ngraph.hpp" #include <gtest/gtest.h>
#include "ngraph/op/util/attr_types.hpp"
#include "openvino/op/constant.hpp"
#include "visitors/visitors.hpp"
using namespace std; using namespace std;
using namespace ngraph; using namespace ov;
using ngraph::test::NodeBuilder; using ov::test::NodeBuilder;
TEST(attributes, idft_op) { TEST(attributes, idft_op) {
NodeBuilder::get_ops().register_factory<op::v7::IDFT>(); NodeBuilder::get_ops().register_factory<op::v7::IDFT>();
auto data = make_shared<op::Parameter>(element::f32, Shape{2, 10, 10, 2}); auto data = make_shared<ov::op::v0::Parameter>(element::f32, Shape{2, 10, 10, 2});
auto axes = op::Constant::create<int64_t>(element::Type_t::i64, Shape{1}, {2}); auto axes = ov::op::v0::Constant::create<int64_t>(element::Type_t::i64, Shape{1}, {2});
auto idft = make_shared<op::v7::IDFT>(data, axes); auto idft = make_shared<op::v7::IDFT>(data, axes);
NodeBuilder builder(idft, {data, axes}); NodeBuilder builder(idft, {data, axes});
@ -26,9 +28,9 @@ TEST(attributes, idft_op) {
TEST(attributes, idft_op_signal) { TEST(attributes, idft_op_signal) {
NodeBuilder::get_ops().register_factory<op::v7::IDFT>(); NodeBuilder::get_ops().register_factory<op::v7::IDFT>();
auto data = make_shared<op::Parameter>(element::f32, Shape{2, 10, 10, 2}); auto data = make_shared<ov::op::v0::Parameter>(element::f32, Shape{2, 10, 10, 2});
auto axes = op::Constant::create<int64_t>(element::Type_t::i64, Shape{1}, {2}); auto axes = ov::op::v0::Constant::create<int64_t>(element::Type_t::i64, Shape{1}, {2});
auto signal = op::Constant::create<int64_t>(element::Type_t::i64, Shape{1}, {20}); auto signal = ov::op::v0::Constant::create<int64_t>(element::Type_t::i64, Shape{1}, {20});
auto idft = make_shared<op::v7::IDFT>(data, axes, signal); auto idft = make_shared<op::v7::IDFT>(data, axes, signal);
NodeBuilder builder(idft, {data, axes, signal}); NodeBuilder builder(idft, {data, axes, signal});

View File

@ -2,35 +2,35 @@
// SPDX-License-Identifier: Apache-2.0 // SPDX-License-Identifier: Apache-2.0
// //
#include "openvino/op/if.hpp"
#include <gtest/gtest.h>
#include "common_test_utils/graph_comparator.hpp" #include "common_test_utils/graph_comparator.hpp"
#include "common_test_utils/visitor.hpp" #include "openvino/op/constant.hpp"
#include "gtest/gtest.h" #include "openvino/op/multiply.hpp"
#include "ngraph/ngraph.hpp" #include "visitors/visitors.hpp"
#include "ngraph/op/util/attr_types.hpp"
#include "ngraph/opsets/opset8.hpp"
using namespace std; using namespace std;
using namespace ngraph; using namespace ov;
using namespace ngraph::opset8; using ov::test::NodeBuilder;
using ngraph::test::NodeBuilder;
using ngraph::test::ValueMap;
TEST(attributes, if_op) { TEST(attributes, if_op) {
NodeBuilder::get_ops().register_factory<If>(); NodeBuilder::get_ops().register_factory<ov::op::v8::If>();
auto X = make_shared<Parameter>(element::f32, Shape{1, 2, 2}); auto X = make_shared<ov::op::v0::Parameter>(element::f32, Shape{1, 2, 2});
auto Y = make_shared<Parameter>(element::f32, Shape{1, 2, 2}); auto Y = make_shared<ov::op::v0::Parameter>(element::f32, Shape{1, 2, 2});
auto cond = std::make_shared<Constant>(element::boolean, Shape{1}, true); auto cond = std::make_shared<ov::op::v0::Constant>(element::boolean, Shape{1}, true);
auto cond2 = std::make_shared<Constant>(element::boolean, Shape{1}, false); auto cond2 = std::make_shared<ov::op::v0::Constant>(element::boolean, Shape{1}, false);
auto Xt = make_shared<op::Parameter>(element::f32, PartialShape::dynamic()); auto Xt = make_shared<ov::op::v0::Parameter>(element::f32, PartialShape::dynamic());
auto Yt = make_shared<op::Parameter>(element::f32, PartialShape::dynamic()); auto Yt = make_shared<ov::op::v0::Parameter>(element::f32, PartialShape::dynamic());
auto Xe = make_shared<op::Parameter>(element::f32, PartialShape::dynamic()); auto Xe = make_shared<ov::op::v0::Parameter>(element::f32, PartialShape::dynamic());
auto Ye = make_shared<op::Parameter>(element::f32, PartialShape::dynamic()); auto Ye = make_shared<ov::op::v0::Parameter>(element::f32, PartialShape::dynamic());
auto then_op = std::make_shared<op::v1::Multiply>(Xt, Yt); auto then_op = std::make_shared<op::v1::Multiply>(Xt, Yt);
auto res0 = make_shared<op::Result>(then_op); auto res0 = make_shared<ov::op::v0::Result>(then_op);
auto res1 = make_shared<op::Result>(Xe); auto res1 = make_shared<ov::op::v0::Result>(Xe);
auto then_body = make_shared<ngraph::Function>(OutputVector{res0}, ParameterVector{Xt, Yt}); auto then_body = make_shared<ov::Model>(OutputVector{res0}, ParameterVector{Xt, Yt});
auto else_body = make_shared<ngraph::Function>(OutputVector{res1}, ParameterVector{Xe}); auto else_body = make_shared<ov::Model>(OutputVector{res1}, ParameterVector{Xe});
auto if_op = make_shared<If>(cond); auto if_op = make_shared<ov::op::v8::If>(cond);
if_op->set_then_body(then_body); if_op->set_then_body(then_body);
if_op->set_else_body(else_body); if_op->set_else_body(else_body);
if_op->set_input(X, Xt, Xe); if_op->set_input(X, Xt, Xe);
@ -38,7 +38,7 @@ TEST(attributes, if_op) {
if_op->set_output(res0, res1); if_op->set_output(res0, res1);
if_op->validate_and_infer_types(); if_op->validate_and_infer_types();
NodeBuilder builder(if_op); NodeBuilder builder(if_op);
auto g_if = ov::as_type_ptr<If>(builder.create()); auto g_if = ov::as_type_ptr<ov::op::v8::If>(builder.create());
EXPECT_EQ(g_if->get_input_descriptions(0), if_op->get_input_descriptions(0)); EXPECT_EQ(g_if->get_input_descriptions(0), if_op->get_input_descriptions(0));
EXPECT_EQ(g_if->get_input_descriptions(1), if_op->get_input_descriptions(1)); EXPECT_EQ(g_if->get_input_descriptions(1), if_op->get_input_descriptions(1));
EXPECT_EQ(g_if->get_output_descriptions(0), if_op->get_output_descriptions(0)); EXPECT_EQ(g_if->get_output_descriptions(0), if_op->get_output_descriptions(0));

View File

@ -2,27 +2,24 @@
// SPDX-License-Identifier: Apache-2.0 // SPDX-License-Identifier: Apache-2.0
// //
#include "common_test_utils/visitor.hpp" #include "openvino/op/interpolate.hpp"
#include "gtest/gtest.h"
#include "ngraph/ngraph.hpp" #include <gtest/gtest.h>
#include "ngraph/op/util/attr_types.hpp"
#include "ngraph/opsets/opset1.hpp" #include "openvino/op/constant.hpp"
#include "ngraph/opsets/opset11.hpp" #include "openvino/op/parameter.hpp"
#include "ngraph/opsets/opset3.hpp" #include "visitors/visitors.hpp"
#include "ngraph/opsets/opset4.hpp"
#include "ngraph/opsets/opset5.hpp"
using namespace std; using namespace std;
using namespace ngraph; using namespace ov;
using ngraph::test::NodeBuilder; using ov::test::NodeBuilder;
using ngraph::test::ValueMap;
TEST(attributes, interpolate_op1) { TEST(attributes, interpolate_op1) {
NodeBuilder::get_ops().register_factory<opset1::Interpolate>(); NodeBuilder::get_ops().register_factory<ov::op::v0::Interpolate>();
auto img = make_shared<op::Parameter>(element::f32, Shape{1, 3, 32, 32}); auto img = make_shared<ov::op::v0::Parameter>(element::f32, Shape{1, 3, 32, 32});
auto out_shape = make_shared<op::Parameter>(element::i32, Shape{2}); auto out_shape = make_shared<ov::op::v0::Parameter>(element::i32, Shape{2});
op::v0::InterpolateAttrs interp_atrs; op::v0::Interpolate::Attributes interp_atrs;
interp_atrs.axes = AxisSet{1, 2}; interp_atrs.axes = AxisSet{1, 2};
interp_atrs.mode = "cubic"; interp_atrs.mode = "cubic";
interp_atrs.align_corners = true; interp_atrs.align_corners = true;
@ -30,9 +27,9 @@ TEST(attributes, interpolate_op1) {
interp_atrs.pads_begin = vector<size_t>{0, 0}; interp_atrs.pads_begin = vector<size_t>{0, 0};
interp_atrs.pads_end = vector<size_t>{0, 0}; interp_atrs.pads_end = vector<size_t>{0, 0};
auto interpolate = make_shared<opset1::Interpolate>(img, out_shape, interp_atrs); auto interpolate = make_shared<ov::op::v0::Interpolate>(img, out_shape, interp_atrs);
NodeBuilder builder(interpolate, {img, out_shape}); NodeBuilder builder(interpolate, {img, out_shape});
auto g_interpolate = ov::as_type_ptr<opset1::Interpolate>(builder.create()); auto g_interpolate = ov::as_type_ptr<ov::op::v0::Interpolate>(builder.create());
const auto i_attrs = interpolate->get_attrs(); const auto i_attrs = interpolate->get_attrs();
const auto g_i_attrs = g_interpolate->get_attrs(); const auto g_i_attrs = g_interpolate->get_attrs();
@ -46,9 +43,9 @@ TEST(attributes, interpolate_op1) {
} }
TEST(attributes, interpolate_op4) { TEST(attributes, interpolate_op4) {
NodeBuilder::get_ops().register_factory<opset4::Interpolate>(); NodeBuilder::get_ops().register_factory<ov::op::v4::Interpolate>();
auto img = make_shared<op::Parameter>(element::f32, Shape{1, 3, 32, 32}); auto img = make_shared<ov::op::v0::Parameter>(element::f32, Shape{1, 3, 32, 32});
auto out_shape = make_shared<op::Parameter>(element::i32, Shape{2}); auto out_shape = make_shared<ov::op::v0::Parameter>(element::i32, Shape{2});
auto scales = op::v0::Constant::create(element::f32, {1}, {1.0}); auto scales = op::v0::Constant::create(element::f32, {1}, {1.0});
op::v4::Interpolate::InterpolateAttrs attrs; op::v4::Interpolate::InterpolateAttrs attrs;
@ -61,9 +58,9 @@ TEST(attributes, interpolate_op4) {
attrs.antialias = true; attrs.antialias = true;
attrs.cube_coeff = -0.75; attrs.cube_coeff = -0.75;
auto interpolate = make_shared<opset4::Interpolate>(img, out_shape, scales, attrs); auto interpolate = make_shared<ov::op::v4::Interpolate>(img, out_shape, scales, attrs);
NodeBuilder builder(interpolate, {img, out_shape, scales}); NodeBuilder builder(interpolate, {img, out_shape, scales});
auto g_interpolate = ov::as_type_ptr<opset4::Interpolate>(builder.create()); auto g_interpolate = ov::as_type_ptr<ov::op::v4::Interpolate>(builder.create());
const auto i_attrs = interpolate->get_attrs(); const auto i_attrs = interpolate->get_attrs();
const auto g_i_attrs = g_interpolate->get_attrs(); const auto g_i_attrs = g_interpolate->get_attrs();
@ -79,8 +76,8 @@ TEST(attributes, interpolate_op4) {
} }
TEST(attributes, interpolate_op11) { TEST(attributes, interpolate_op11) {
NodeBuilder::get_ops().register_factory<opset11::Interpolate>(); NodeBuilder::get_ops().register_factory<ov::op::v11::Interpolate>();
const auto img = make_shared<op::Parameter>(element::f32, Shape{1, 3, 32, 32}); const auto img = make_shared<ov::op::v0::Parameter>(element::f32, Shape{1, 3, 32, 32});
const auto scales = op::v0::Constant::create(element::f32, {4}, {1.0, 1.0, 2.0, 2.0}); const auto scales = op::v0::Constant::create(element::f32, {4}, {1.0, 1.0, 2.0, 2.0});
op::v11::Interpolate::InterpolateAttrs attrs; op::v11::Interpolate::InterpolateAttrs attrs;
@ -93,9 +90,9 @@ TEST(attributes, interpolate_op11) {
attrs.antialias = true; attrs.antialias = true;
attrs.cube_coeff = -0.75; attrs.cube_coeff = -0.75;
auto interpolate = make_shared<opset11::Interpolate>(img, scales, attrs); auto interpolate = make_shared<ov::op::v11::Interpolate>(img, scales, attrs);
NodeBuilder builder(interpolate, {img, scales}); NodeBuilder builder(interpolate, {img, scales});
auto g_interpolate = ov::as_type_ptr<opset11::Interpolate>(builder.create()); auto g_interpolate = ov::as_type_ptr<ov::op::v11::Interpolate>(builder.create());
const auto i_attrs = interpolate->get_attrs(); const auto i_attrs = interpolate->get_attrs();
const auto g_i_attrs = g_interpolate->get_attrs(); const auto g_i_attrs = g_interpolate->get_attrs();

View File

@ -2,14 +2,16 @@
// SPDX-License-Identifier: Apache-2.0 // SPDX-License-Identifier: Apache-2.0
// //
#include "common_test_utils/visitor.hpp" #include "openvino/op/irdft.hpp"
#include "gtest/gtest.h"
#include "ngraph/ngraph.hpp" #include <gtest/gtest.h>
#include "ngraph/op/util/attr_types.hpp"
#include "openvino/op/constant.hpp"
#include "visitors/visitors.hpp"
using namespace std; using namespace std;
using namespace ngraph; using namespace ov;
using ngraph::test::NodeBuilder; using ov::test::NodeBuilder;
TEST(attributes, irdft_op) { TEST(attributes, irdft_op) {
NodeBuilder::get_ops().register_factory<op::v9::IRDFT>(); NodeBuilder::get_ops().register_factory<op::v9::IRDFT>();
@ -27,7 +29,7 @@ TEST(attributes, irdft_op) {
TEST(attributes, irdft_op_signal) { TEST(attributes, irdft_op_signal) {
NodeBuilder::get_ops().register_factory<op::v9::IRDFT>(); NodeBuilder::get_ops().register_factory<op::v9::IRDFT>();
auto data = make_shared<op::v0::Parameter>(element::f32, Shape{2, 10, 10, 2}); auto data = make_shared<op::v0::Parameter>(element::f32, Shape{2, 10, 10, 2});
auto signal = op::Constant::create<int64_t>(element::Type_t::i64, Shape{1}, {20}); auto signal = op::v0::Constant::create<int64_t>(element::Type_t::i64, Shape{1}, {20});
auto axes = op::v0::Constant::create<int64_t>(element::i64, Shape{1}, {2}); auto axes = op::v0::Constant::create<int64_t>(element::i64, Shape{1}, {2});
auto irdft = make_shared<op::v9::IRDFT>(data, axes, signal); auto irdft = make_shared<op::v9::IRDFT>(data, axes, signal);

View File

@ -2,9 +2,10 @@
// SPDX-License-Identifier: Apache-2.0 // SPDX-License-Identifier: Apache-2.0
// //
#include "openvino/opsets/opset10.hpp" #include "openvino/op/is_finite.hpp"
#include "unary_ops.hpp" #include "unary_ops.hpp"
using Types = ::testing::Types<UnaryOperatorType<ov::op::v10::IsFinite, ngraph::element::f32>>; using Types = ::testing::Types<UnaryOperatorType<ov::op::v10::IsFinite, ov::element::f32>>;
INSTANTIATE_TYPED_TEST_SUITE_P(visitor_without_attribute, UnaryOperatorVisitor, Types, UnaryOperatorTypeName); INSTANTIATE_TYPED_TEST_SUITE_P(visitor_without_attribute, UnaryOperatorVisitor, Types, UnaryOperatorTypeName);

View File

@ -2,25 +2,25 @@
// SPDX-License-Identifier: Apache-2.0 // SPDX-License-Identifier: Apache-2.0
// //
#include "common_test_utils/visitor.hpp" #include "openvino/op/is_inf.hpp"
#include "gtest/gtest.h"
#include "ngraph/ngraph.hpp" #include <gtest/gtest.h>
#include "ngraph/op/util/attr_types.hpp"
#include "openvino/opsets/opset10.hpp" #include "openvino/op/parameter.hpp"
#include "visitors/visitors.hpp"
using namespace std; using namespace std;
using namespace ov; using namespace ov;
using ngraph::test::NodeBuilder; using ov::test::NodeBuilder;
using namespace ov::opset10;
TEST(attributes, is_inf_defaults) { TEST(attributes, is_inf_defaults) {
NodeBuilder::get_ops().register_factory<IsInf>(); NodeBuilder::get_ops().register_factory<ov::op::v10::IsInf>();
const auto data = make_shared<Parameter>(element::f32, Shape{1, 3, 10, 10}); const auto data = make_shared<ov::op::v0::Parameter>(element::f32, Shape{1, 3, 10, 10});
const auto is_inf = make_shared<IsInf>(data); const auto is_inf = make_shared<ov::op::v10::IsInf>(data);
NodeBuilder builder(is_inf); NodeBuilder builder(is_inf);
auto g_is_inf = ov::as_type_ptr<IsInf>(builder.create()); auto g_is_inf = ov::as_type_ptr<ov::op::v10::IsInf>(builder.create());
const auto is_inf_attrs = is_inf->get_attributes(); const auto is_inf_attrs = is_inf->get_attributes();
const auto g_is_inf_attrs = g_is_inf->get_attributes(); const auto g_is_inf_attrs = g_is_inf->get_attributes();
@ -30,15 +30,15 @@ TEST(attributes, is_inf_defaults) {
} }
TEST(attributes, is_inf_positive_only) { TEST(attributes, is_inf_positive_only) {
NodeBuilder::get_ops().register_factory<IsInf>(); NodeBuilder::get_ops().register_factory<ov::op::v10::IsInf>();
const auto data = make_shared<Parameter>(element::f32, Shape{1, 3, 10, 10}); const auto data = make_shared<ov::op::v0::Parameter>(element::f32, Shape{1, 3, 10, 10});
IsInf::Attributes attributes{}; ov::op::v10::IsInf::Attributes attributes{};
attributes.detect_negative = false; attributes.detect_negative = false;
const auto is_inf = make_shared<IsInf>(data, attributes); const auto is_inf = make_shared<ov::op::v10::IsInf>(data, attributes);
NodeBuilder builder(is_inf); NodeBuilder builder(is_inf);
auto g_is_inf = ov::as_type_ptr<IsInf>(builder.create()); auto g_is_inf = ov::as_type_ptr<ov::op::v10::IsInf>(builder.create());
const auto is_inf_attrs = is_inf->get_attributes(); const auto is_inf_attrs = is_inf->get_attributes();
const auto g_is_inf_attrs = g_is_inf->get_attributes(); const auto g_is_inf_attrs = g_is_inf->get_attributes();
@ -48,15 +48,15 @@ TEST(attributes, is_inf_positive_only) {
} }
TEST(attributes, is_inf_negative_only) { TEST(attributes, is_inf_negative_only) {
NodeBuilder::get_ops().register_factory<IsInf>(); NodeBuilder::get_ops().register_factory<ov::op::v10::IsInf>();
const auto data = make_shared<Parameter>(element::f32, Shape{1, 3, 10, 10}); const auto data = make_shared<ov::op::v0::Parameter>(element::f32, Shape{1, 3, 10, 10});
IsInf::Attributes attributes{}; ov::op::v10::IsInf::Attributes attributes{};
attributes.detect_positive = false; attributes.detect_positive = false;
const auto is_inf = make_shared<IsInf>(data, attributes); const auto is_inf = make_shared<ov::op::v10::IsInf>(data, attributes);
NodeBuilder builder(is_inf); NodeBuilder builder(is_inf);
auto g_is_inf = ov::as_type_ptr<IsInf>(builder.create()); auto g_is_inf = ov::as_type_ptr<ov::op::v10::IsInf>(builder.create());
const auto is_inf_attrs = is_inf->get_attributes(); const auto is_inf_attrs = is_inf->get_attributes();
const auto g_is_inf_attrs = g_is_inf->get_attributes(); const auto g_is_inf_attrs = g_is_inf->get_attributes();
@ -66,16 +66,16 @@ TEST(attributes, is_inf_negative_only) {
} }
TEST(attributes, is_inf_detect_none) { TEST(attributes, is_inf_detect_none) {
NodeBuilder::get_ops().register_factory<IsInf>(); NodeBuilder::get_ops().register_factory<ov::op::v10::IsInf>();
const auto data = make_shared<Parameter>(element::f32, Shape{1, 3, 10, 10}); const auto data = make_shared<ov::op::v0::Parameter>(element::f32, Shape{1, 3, 10, 10});
IsInf::Attributes attributes{}; ov::op::v10::IsInf::Attributes attributes{};
attributes.detect_negative = false; attributes.detect_negative = false;
attributes.detect_positive = false; attributes.detect_positive = false;
const auto is_inf = make_shared<IsInf>(data, attributes); const auto is_inf = make_shared<ov::op::v10::IsInf>(data, attributes);
NodeBuilder builder(is_inf); NodeBuilder builder(is_inf);
auto g_is_inf = ov::as_type_ptr<IsInf>(builder.create()); auto g_is_inf = ov::as_type_ptr<ov::op::v10::IsInf>(builder.create());
const auto is_inf_attrs = is_inf->get_attributes(); const auto is_inf_attrs = is_inf->get_attributes();
const auto g_is_inf_attrs = g_is_inf->get_attributes(); const auto g_is_inf_attrs = g_is_inf->get_attributes();

View File

@ -2,7 +2,10 @@
// SPDX-License-Identifier: Apache-2.0 // SPDX-License-Identifier: Apache-2.0
// //
#include "openvino/op/is_nan.hpp"
#include "unary_ops.hpp" #include "unary_ops.hpp"
using Types = ::testing::Types<UnaryOperatorType<ov::op::v10::IsNaN, ov::element::f32>>; using Types = ::testing::Types<UnaryOperatorType<ov::op::v10::IsNaN, ov::element::f32>>;
INSTANTIATE_TYPED_TEST_SUITE_P(visitor_without_attribute, UnaryOperatorVisitor, Types, UnaryOperatorTypeName); INSTANTIATE_TYPED_TEST_SUITE_P(visitor_without_attribute, UnaryOperatorVisitor, Types, UnaryOperatorTypeName);

View File

@ -2,9 +2,10 @@
// SPDX-License-Identifier: Apache-2.0 // SPDX-License-Identifier: Apache-2.0
// //
#include "binary_ops.hpp" #include "openvino/op/less.hpp"
#include "ngraph/opsets/opset1.hpp"
using Type = ::testing::Types<BinaryOperatorType<ngraph::opset1::Less, ngraph::element::f32>>; #include "binary_ops.hpp"
using Type = ::testing::Types<BinaryOperatorType<ov::op::v1::Less, ov::element::f32>>;
INSTANTIATE_TYPED_TEST_SUITE_P(visitor_with_auto_broadcast, BinaryOperatorVisitor, Type, BinaryOperatorTypeName); INSTANTIATE_TYPED_TEST_SUITE_P(visitor_with_auto_broadcast, BinaryOperatorVisitor, Type, BinaryOperatorTypeName);

View File

@ -3,8 +3,8 @@
// //
#include "binary_ops.hpp" #include "binary_ops.hpp"
#include "ngraph/opsets/opset1.hpp" #include "openvino/op/less_eq.hpp"
using Type = ::testing::Types<BinaryOperatorType<ngraph::opset1::LessEqual, ngraph::element::f32>>; using Type = ::testing::Types<BinaryOperatorType<ov::op::v1::LessEqual, ov::element::f32>>;
INSTANTIATE_TYPED_TEST_SUITE_P(visitor_with_auto_broadcast, BinaryOperatorVisitor, Type, BinaryOperatorTypeName); INSTANTIATE_TYPED_TEST_SUITE_P(visitor_with_auto_broadcast, BinaryOperatorVisitor, Type, BinaryOperatorTypeName);

View File

@ -2,9 +2,11 @@
// SPDX-License-Identifier: Apache-2.0 // SPDX-License-Identifier: Apache-2.0
// //
#include "openvino/op/log.hpp"
#include "unary_ops.hpp" #include "unary_ops.hpp"
using Types = ::testing::Types<UnaryOperatorType<ov::op::v0::Log, ngraph::element::f32>, using Types = ::testing::Types<UnaryOperatorType<ov::op::v0::Log, ov::element::f32>,
UnaryOperatorType<ov::op::v0::Log, ngraph::element::f16>>; UnaryOperatorType<ov::op::v0::Log, ov::element::f16>>;
INSTANTIATE_TYPED_TEST_SUITE_P(visitor_without_attribute, UnaryOperatorVisitor, Types, UnaryOperatorTypeName); INSTANTIATE_TYPED_TEST_SUITE_P(visitor_without_attribute, UnaryOperatorVisitor, Types, UnaryOperatorTypeName);

View File

@ -2,29 +2,25 @@
// SPDX-License-Identifier: Apache-2.0 // SPDX-License-Identifier: Apache-2.0
// //
#include "common_test_utils/visitor.hpp" #include "openvino/op/log_softmax.hpp"
#include "gtest/gtest.h"
#include "ngraph/ngraph.hpp" #include <gtest/gtest.h>
#include "ngraph/op/util/attr_types.hpp"
#include "ngraph/opsets/opset1.hpp" #include "visitors/visitors.hpp"
#include "ngraph/opsets/opset3.hpp"
#include "ngraph/opsets/opset4.hpp"
#include "ngraph/opsets/opset5.hpp"
using namespace std; using namespace std;
using namespace ngraph; using namespace ov;
using ngraph::test::NodeBuilder; using ov::test::NodeBuilder;
using ngraph::test::ValueMap;
TEST(attributes, logsoftmax_op) { TEST(attributes, logsoftmax_op) {
NodeBuilder::get_ops().register_factory<opset5::LogSoftmax>(); NodeBuilder::get_ops().register_factory<ov::op::v5::LogSoftmax>();
auto data = make_shared<op::Parameter>(element::f32, Shape{3, 2, 3}); auto data = make_shared<ov::op::v0::Parameter>(element::f32, Shape{3, 2, 3});
int64_t axis = 2; int64_t axis = 2;
const auto logsoftmax = make_shared<opset5::LogSoftmax>(data, axis); const auto logsoftmax = make_shared<ov::op::v5::LogSoftmax>(data, axis);
NodeBuilder builder(logsoftmax, {data}); NodeBuilder builder(logsoftmax, {data});
auto g_logsoftmax = ov::as_type_ptr<opset5::LogSoftmax>(builder.create()); auto g_logsoftmax = ov::as_type_ptr<ov::op::v5::LogSoftmax>(builder.create());
const auto expected_attr_count = 1; const auto expected_attr_count = 1;
EXPECT_EQ(builder.get_value_map_size(), expected_attr_count); EXPECT_EQ(builder.get_value_map_size(), expected_attr_count);

View File

@ -2,9 +2,10 @@
// SPDX-License-Identifier: Apache-2.0 // SPDX-License-Identifier: Apache-2.0
// //
#include "binary_ops.hpp" #include "openvino/op/logical_and.hpp"
#include "ngraph/opsets/opset1.hpp"
using Type = ::testing::Types<BinaryOperatorType<ngraph::opset1::LogicalAnd, ngraph::element::boolean>>; #include "binary_ops.hpp"
using Type = ::testing::Types<BinaryOperatorType<ov::op::v1::LogicalAnd, ov::element::boolean>>;
INSTANTIATE_TYPED_TEST_SUITE_P(visitor_with_auto_broadcast, BinaryOperatorVisitor, Type, BinaryOperatorTypeName); INSTANTIATE_TYPED_TEST_SUITE_P(visitor_with_auto_broadcast, BinaryOperatorVisitor, Type, BinaryOperatorTypeName);

View File

@ -2,8 +2,10 @@
// SPDX-License-Identifier: Apache-2.0 // SPDX-License-Identifier: Apache-2.0
// //
#include "openvino/op/logical_not.hpp"
#include "unary_ops.hpp" #include "unary_ops.hpp"
using Type = ::testing::Types<UnaryOperatorType<ov::op::v1::LogicalNot, ngraph::element::boolean>>; using Type = ::testing::Types<UnaryOperatorType<ov::op::v1::LogicalNot, ov::element::boolean>>;
INSTANTIATE_TYPED_TEST_SUITE_P(visitor_without_attribute, UnaryOperatorVisitor, Type, UnaryOperatorTypeName); INSTANTIATE_TYPED_TEST_SUITE_P(visitor_without_attribute, UnaryOperatorVisitor, Type, UnaryOperatorTypeName);

View File

@ -2,9 +2,10 @@
// SPDX-License-Identifier: Apache-2.0 // SPDX-License-Identifier: Apache-2.0
// //
#include "binary_ops.hpp" #include "openvino/op/logical_or.hpp"
#include "ngraph/opsets/opset1.hpp"
using Type = ::testing::Types<BinaryOperatorType<ngraph::opset1::LogicalOr, ngraph::element::boolean>>; #include "binary_ops.hpp"
using Type = ::testing::Types<BinaryOperatorType<ov::op::v1::LogicalOr, ov::element::boolean>>;
INSTANTIATE_TYPED_TEST_SUITE_P(visitor_with_auto_broadcast, BinaryOperatorVisitor, Type, BinaryOperatorTypeName); INSTANTIATE_TYPED_TEST_SUITE_P(visitor_with_auto_broadcast, BinaryOperatorVisitor, Type, BinaryOperatorTypeName);

View File

@ -2,9 +2,10 @@
// SPDX-License-Identifier: Apache-2.0 // SPDX-License-Identifier: Apache-2.0
// //
#include "binary_ops.hpp" #include "openvino/op/logical_xor.hpp"
#include "ngraph/opsets/opset1.hpp"
using Type = ::testing::Types<BinaryOperatorType<ngraph::opset1::LogicalXor, ngraph::element::boolean>>; #include "binary_ops.hpp"
using Type = ::testing::Types<BinaryOperatorType<ov::op::v1::LogicalXor, ov::element::boolean>>;
INSTANTIATE_TYPED_TEST_SUITE_P(visitor_with_auto_broadcast, BinaryOperatorVisitor, Type, BinaryOperatorTypeName); INSTANTIATE_TYPED_TEST_SUITE_P(visitor_with_auto_broadcast, BinaryOperatorVisitor, Type, BinaryOperatorTypeName);

View File

@ -2,33 +2,29 @@
// SPDX-License-Identifier: Apache-2.0 // SPDX-License-Identifier: Apache-2.0
// //
#include "common_test_utils/visitor.hpp" #include "openvino/op/lrn.hpp"
#include "gtest/gtest.h"
#include "ngraph/ngraph.hpp" #include <gtest/gtest.h>
#include "ngraph/op/util/attr_types.hpp"
#include "ngraph/opsets/opset1.hpp" #include "visitors/visitors.hpp"
#include "ngraph/opsets/opset3.hpp"
#include "ngraph/opsets/opset4.hpp"
#include "ngraph/opsets/opset5.hpp"
using namespace std; using namespace std;
using namespace ngraph; using namespace ov;
using ngraph::test::NodeBuilder; using ov::test::NodeBuilder;
using ngraph::test::ValueMap;
TEST(attributes, lrn_op) { TEST(attributes, lrn_op) {
NodeBuilder::get_ops().register_factory<opset1::LRN>(); NodeBuilder::get_ops().register_factory<ov::op::v0::LRN>();
const auto arg = make_shared<op::Parameter>(element::f32, Shape{1, 2, 3, 4}); const auto arg = make_shared<ov::op::v0::Parameter>(element::f32, Shape{1, 2, 3, 4});
const auto axes = make_shared<op::Parameter>(element::i32, Shape{2}); const auto axes = make_shared<ov::op::v0::Parameter>(element::i32, Shape{2});
const double alpha = 1.1; const double alpha = 1.1;
const double beta = 2.2; const double beta = 2.2;
const double bias = 3.3; const double bias = 3.3;
const size_t size = 4; const size_t size = 4;
const auto lrn = make_shared<opset1::LRN>(arg, axes, alpha, beta, bias, size); const auto lrn = make_shared<ov::op::v0::LRN>(arg, axes, alpha, beta, bias, size);
NodeBuilder builder(lrn, {arg, axes}); NodeBuilder builder(lrn, {arg, axes});
auto g_lrn = ov::as_type_ptr<opset1::LRN>(builder.create()); auto g_lrn = ov::as_type_ptr<ov::op::v0::LRN>(builder.create());
EXPECT_EQ(g_lrn->get_alpha(), lrn->get_alpha()); EXPECT_EQ(g_lrn->get_alpha(), lrn->get_alpha());
EXPECT_EQ(g_lrn->get_beta(), lrn->get_beta()); EXPECT_EQ(g_lrn->get_beta(), lrn->get_beta());

View File

@ -2,28 +2,24 @@
// SPDX-License-Identifier: Apache-2.0 // SPDX-License-Identifier: Apache-2.0
// //
#include "common_test_utils/visitor.hpp" #include "openvino/op/lstm_cell.hpp"
#include "gtest/gtest.h"
#include "ngraph/ngraph.hpp" #include <gtest/gtest.h>
#include "ngraph/op/util/attr_types.hpp"
#include "ngraph/opsets/opset1.hpp" #include "visitors/visitors.hpp"
#include "ngraph/opsets/opset3.hpp"
#include "ngraph/opsets/opset4.hpp"
#include "ngraph/opsets/opset5.hpp"
using namespace std; using namespace std;
using namespace ngraph; using namespace ov;
using ngraph::test::NodeBuilder; using ov::test::NodeBuilder;
using ngraph::test::ValueMap;
TEST(attributes, lstm_cell_v0_op) { TEST(attributes, lstm_cell_v0_op) {
NodeBuilder::get_ops().register_factory<opset1::LSTMCell>(); NodeBuilder::get_ops().register_factory<ov::op::v0::LSTMCell>();
auto X = make_shared<op::Parameter>(element::f32, Shape{2, 3}); auto X = make_shared<ov::op::v0::Parameter>(element::f32, Shape{2, 3});
auto H = make_shared<op::Parameter>(element::f32, Shape{2, 3}); auto H = make_shared<ov::op::v0::Parameter>(element::f32, Shape{2, 3});
auto W = make_shared<op::Parameter>(element::f32, Shape{12, 3}); auto W = make_shared<ov::op::v0::Parameter>(element::f32, Shape{12, 3});
auto R = make_shared<op::Parameter>(element::f32, Shape{12, 3}); auto R = make_shared<ov::op::v0::Parameter>(element::f32, Shape{12, 3});
const auto initial_hidden_state = make_shared<op::Parameter>(element::f32, Shape{2, 3}); const auto initial_hidden_state = make_shared<ov::op::v0::Parameter>(element::f32, Shape{2, 3});
const auto initial_cell_state = make_shared<op::Parameter>(element::f32, Shape{2, 3}); const auto initial_cell_state = make_shared<ov::op::v0::Parameter>(element::f32, Shape{2, 3});
const auto hidden_size = 3; const auto hidden_size = 3;
auto weights_format = ov::op::LSTMWeightsFormat::IFCO; auto weights_format = ov::op::LSTMWeightsFormat::IFCO;
@ -32,7 +28,7 @@ TEST(attributes, lstm_cell_v0_op) {
auto activations_beta = std::vector<float>{2.0, 1.0}; auto activations_beta = std::vector<float>{2.0, 1.0};
const float clip = 0.5f; const float clip = 0.5f;
auto input_forget = false; auto input_forget = false;
const auto lstm_cell = make_shared<opset1::LSTMCell>(X, const auto lstm_cell = make_shared<ov::op::v0::LSTMCell>(X,
initial_hidden_state, initial_hidden_state,
initial_cell_state, initial_cell_state,
W, W,
@ -45,7 +41,7 @@ TEST(attributes, lstm_cell_v0_op) {
clip, clip,
input_forget); input_forget);
NodeBuilder builder(lstm_cell, {X, initial_hidden_state, initial_cell_state, W, R}); NodeBuilder builder(lstm_cell, {X, initial_hidden_state, initial_cell_state, W, R});
auto g_lstm_cell = ov::as_type_ptr<opset1::LSTMCell>(builder.create()); auto g_lstm_cell = ov::as_type_ptr<ov::op::v0::LSTMCell>(builder.create());
EXPECT_EQ(g_lstm_cell->get_hidden_size(), lstm_cell->get_hidden_size()); EXPECT_EQ(g_lstm_cell->get_hidden_size(), lstm_cell->get_hidden_size());
EXPECT_EQ(g_lstm_cell->get_activations(), lstm_cell->get_activations()); EXPECT_EQ(g_lstm_cell->get_activations(), lstm_cell->get_activations());
@ -55,20 +51,20 @@ TEST(attributes, lstm_cell_v0_op) {
} }
TEST(attributes, lstm_cell_v4_op) { TEST(attributes, lstm_cell_v4_op) {
NodeBuilder::get_ops().register_factory<opset4::LSTMCell>(); NodeBuilder::get_ops().register_factory<ov::op::v4::LSTMCell>();
auto X = make_shared<op::Parameter>(element::f32, Shape{2, 3}); auto X = make_shared<ov::op::v0::Parameter>(element::f32, Shape{2, 3});
auto H = make_shared<op::Parameter>(element::f32, Shape{2, 3}); auto H = make_shared<ov::op::v0::Parameter>(element::f32, Shape{2, 3});
auto W = make_shared<op::Parameter>(element::f32, Shape{12, 3}); auto W = make_shared<ov::op::v0::Parameter>(element::f32, Shape{12, 3});
auto R = make_shared<op::Parameter>(element::f32, Shape{12, 3}); auto R = make_shared<ov::op::v0::Parameter>(element::f32, Shape{12, 3});
const auto initial_hidden_state = make_shared<op::Parameter>(element::f32, Shape{2, 3}); const auto initial_hidden_state = make_shared<ov::op::v0::Parameter>(element::f32, Shape{2, 3});
const auto initial_cell_state = make_shared<op::Parameter>(element::f32, Shape{2, 3}); const auto initial_cell_state = make_shared<ov::op::v0::Parameter>(element::f32, Shape{2, 3});
const auto hidden_size = 3; const auto hidden_size = 3;
const std::vector<std::string> activations = {"tanh", "sigmoid", "tanh"}; const std::vector<std::string> activations = {"tanh", "sigmoid", "tanh"};
auto activations_alpha = std::vector<float>{1.0, 1.5}; auto activations_alpha = std::vector<float>{1.0, 1.5};
auto activations_beta = std::vector<float>{2.0, 1.0}; auto activations_beta = std::vector<float>{2.0, 1.0};
const float clip = 0.5f; const float clip = 0.5f;
const auto lstm_cell = make_shared<opset4::LSTMCell>(X, const auto lstm_cell = make_shared<ov::op::v4::LSTMCell>(X,
initial_hidden_state, initial_hidden_state,
initial_cell_state, initial_cell_state,
W, W,
@ -79,7 +75,7 @@ TEST(attributes, lstm_cell_v4_op) {
activations_beta, activations_beta,
clip); clip);
NodeBuilder builder(lstm_cell, {X, initial_hidden_state, initial_cell_state, W, R}); NodeBuilder builder(lstm_cell, {X, initial_hidden_state, initial_cell_state, W, R});
auto g_lstm_cell = ov::as_type_ptr<opset4::LSTMCell>(builder.create()); auto g_lstm_cell = ov::as_type_ptr<ov::op::v4::LSTMCell>(builder.create());
EXPECT_EQ(g_lstm_cell->get_hidden_size(), lstm_cell->get_hidden_size()); EXPECT_EQ(g_lstm_cell->get_hidden_size(), lstm_cell->get_hidden_size());
EXPECT_EQ(g_lstm_cell->get_activations(), lstm_cell->get_activations()); EXPECT_EQ(g_lstm_cell->get_activations(), lstm_cell->get_activations());
@ -89,21 +85,21 @@ TEST(attributes, lstm_cell_v4_op) {
} }
TEST(attributes, lstm_cell_v4_op2) { TEST(attributes, lstm_cell_v4_op2) {
NodeBuilder::get_ops().register_factory<opset4::LSTMCell>(); NodeBuilder::get_ops().register_factory<ov::op::v4::LSTMCell>();
auto X = make_shared<op::Parameter>(element::f32, Shape{2, 3}); auto X = make_shared<ov::op::v0::Parameter>(element::f32, Shape{2, 3});
auto H = make_shared<op::Parameter>(element::f32, Shape{2, 3}); auto H = make_shared<ov::op::v0::Parameter>(element::f32, Shape{2, 3});
auto W = make_shared<op::Parameter>(element::f32, Shape{12, 3}); auto W = make_shared<ov::op::v0::Parameter>(element::f32, Shape{12, 3});
auto R = make_shared<op::Parameter>(element::f32, Shape{12, 3}); auto R = make_shared<ov::op::v0::Parameter>(element::f32, Shape{12, 3});
const auto initial_hidden_state = make_shared<op::Parameter>(element::f32, Shape{2, 3}); const auto initial_hidden_state = make_shared<ov::op::v0::Parameter>(element::f32, Shape{2, 3});
const auto initial_cell_state = make_shared<op::Parameter>(element::f32, Shape{2, 3}); const auto initial_cell_state = make_shared<ov::op::v0::Parameter>(element::f32, Shape{2, 3});
auto B = make_shared<op::Parameter>(element::f32, Shape{12}); auto B = make_shared<ov::op::v0::Parameter>(element::f32, Shape{12});
const auto hidden_size = 3; const auto hidden_size = 3;
const std::vector<std::string> activations = {"tanh", "sigmoid", "tanh"}; const std::vector<std::string> activations = {"tanh", "sigmoid", "tanh"};
auto activations_alpha = std::vector<float>{1.0, 1.5}; auto activations_alpha = std::vector<float>{1.0, 1.5};
auto activations_beta = std::vector<float>{2.0, 1.0}; auto activations_beta = std::vector<float>{2.0, 1.0};
const float clip = 0.5f; const float clip = 0.5f;
const auto lstm_cell = make_shared<opset4::LSTMCell>(X, const auto lstm_cell = make_shared<ov::op::v4::LSTMCell>(X,
initial_hidden_state, initial_hidden_state,
initial_cell_state, initial_cell_state,
W, W,
@ -115,7 +111,7 @@ TEST(attributes, lstm_cell_v4_op2) {
activations_beta, activations_beta,
clip); clip);
NodeBuilder builder(lstm_cell, {X, initial_hidden_state, initial_cell_state, W, R, B}); NodeBuilder builder(lstm_cell, {X, initial_hidden_state, initial_cell_state, W, R, B});
auto g_lstm_cell = ov::as_type_ptr<opset4::LSTMCell>(builder.create()); auto g_lstm_cell = ov::as_type_ptr<ov::op::v4::LSTMCell>(builder.create());
EXPECT_EQ(g_lstm_cell->get_hidden_size(), lstm_cell->get_hidden_size()); EXPECT_EQ(g_lstm_cell->get_hidden_size(), lstm_cell->get_hidden_size());
EXPECT_EQ(g_lstm_cell->get_activations(), lstm_cell->get_activations()); EXPECT_EQ(g_lstm_cell->get_activations(), lstm_cell->get_activations());

View File

@ -2,22 +2,18 @@
// SPDX-License-Identifier: Apache-2.0 // SPDX-License-Identifier: Apache-2.0
// //
#include "common_test_utils/visitor.hpp" #include "openvino/op/lstm_sequence.hpp"
#include "gtest/gtest.h"
#include "ngraph/ngraph.hpp" #include <gtest/gtest.h>
#include "ngraph/op/util/attr_types.hpp"
#include "ngraph/opsets/opset1.hpp" #include "visitors/visitors.hpp"
#include "ngraph/opsets/opset3.hpp"
#include "ngraph/opsets/opset4.hpp"
#include "ngraph/opsets/opset5.hpp"
using namespace std; using namespace std;
using namespace ngraph; using namespace ov;
using ngraph::test::NodeBuilder; using ov::test::NodeBuilder;
using ngraph::test::ValueMap;
TEST(attributes, lstm_sequence_op) { TEST(attributes, lstm_sequence_op) {
NodeBuilder::get_ops().register_factory<opset5::LSTMSequence>(); NodeBuilder::get_ops().register_factory<ov::op::v5::LSTMSequence>();
const size_t batch_size = 4; const size_t batch_size = 4;
const size_t num_directions = 2; const size_t num_directions = 2;
@ -25,15 +21,16 @@ TEST(attributes, lstm_sequence_op) {
const size_t input_size = 16; const size_t input_size = 16;
const size_t hidden_size = 64; const size_t hidden_size = 64;
const auto X = make_shared<op::Parameter>(element::f32, Shape{batch_size, seq_length, input_size}); const auto X = make_shared<ov::op::v0::Parameter>(element::f32, Shape{batch_size, seq_length, input_size});
const auto initial_hidden_state = const auto initial_hidden_state =
make_shared<op::Parameter>(element::f32, Shape{batch_size, num_directions, hidden_size}); make_shared<ov::op::v0::Parameter>(element::f32, Shape{batch_size, num_directions, hidden_size});
const auto initial_cell_state = const auto initial_cell_state =
make_shared<op::Parameter>(element::f32, Shape{batch_size, num_directions, hidden_size}); make_shared<ov::op::v0::Parameter>(element::f32, Shape{batch_size, num_directions, hidden_size});
const auto sequence_lengths = make_shared<op::Parameter>(element::i32, Shape{batch_size}); const auto sequence_lengths = make_shared<ov::op::v0::Parameter>(element::i32, Shape{batch_size});
const auto W = make_shared<op::Parameter>(element::f32, Shape{num_directions, 4 * hidden_size, input_size}); const auto W = make_shared<ov::op::v0::Parameter>(element::f32, Shape{num_directions, 4 * hidden_size, input_size});
const auto R = make_shared<op::Parameter>(element::f32, Shape{num_directions, 4 * hidden_size, hidden_size}); const auto R =
const auto B = make_shared<op::Parameter>(element::f32, Shape{num_directions, 4 * hidden_size}); make_shared<ov::op::v0::Parameter>(element::f32, Shape{num_directions, 4 * hidden_size, hidden_size});
const auto B = make_shared<ov::op::v0::Parameter>(element::f32, Shape{num_directions, 4 * hidden_size});
const auto lstm_direction = op::RecurrentSequenceDirection::BIDIRECTIONAL; const auto lstm_direction = op::RecurrentSequenceDirection::BIDIRECTIONAL;
const std::vector<float> activations_alpha = {1, 2, 3}; const std::vector<float> activations_alpha = {1, 2, 3};
@ -41,7 +38,7 @@ TEST(attributes, lstm_sequence_op) {
const std::vector<std::string> activations = {"tanh", "sigmoid", "tanh"}; const std::vector<std::string> activations = {"tanh", "sigmoid", "tanh"};
const float clip_threshold = 0.5f; const float clip_threshold = 0.5f;
const auto lstm_sequence = make_shared<opset5::LSTMSequence>(X, const auto lstm_sequence = make_shared<ov::op::v5::LSTMSequence>(X,
initial_hidden_state, initial_hidden_state,
initial_cell_state, initial_cell_state,
sequence_lengths, sequence_lengths,
@ -55,7 +52,7 @@ TEST(attributes, lstm_sequence_op) {
activations, activations,
clip_threshold); clip_threshold);
NodeBuilder builder(lstm_sequence, {X, initial_hidden_state, initial_cell_state, sequence_lengths, W, R, B}); NodeBuilder builder(lstm_sequence, {X, initial_hidden_state, initial_cell_state, sequence_lengths, W, R, B});
auto g_lstm_sequence = ov::as_type_ptr<opset5::LSTMSequence>(builder.create()); auto g_lstm_sequence = ov::as_type_ptr<ov::op::v5::LSTMSequence>(builder.create());
EXPECT_EQ(g_lstm_sequence->get_hidden_size(), lstm_sequence->get_hidden_size()); EXPECT_EQ(g_lstm_sequence->get_hidden_size(), lstm_sequence->get_hidden_size());
EXPECT_EQ(g_lstm_sequence->get_activations(), lstm_sequence->get_activations()); EXPECT_EQ(g_lstm_sequence->get_activations(), lstm_sequence->get_activations());
@ -66,7 +63,7 @@ TEST(attributes, lstm_sequence_op) {
} }
TEST(attributes, lstm_sequence_v1_op) { TEST(attributes, lstm_sequence_v1_op) {
NodeBuilder::get_ops().register_factory<opset5::LSTMSequence>(); NodeBuilder::get_ops().register_factory<ov::op::v5::LSTMSequence>();
const size_t batch_size = 4; const size_t batch_size = 4;
const size_t num_directions = 2; const size_t num_directions = 2;
@ -74,16 +71,17 @@ TEST(attributes, lstm_sequence_v1_op) {
const size_t input_size = 16; const size_t input_size = 16;
const size_t hidden_size = 64; const size_t hidden_size = 64;
const auto X = make_shared<op::Parameter>(element::f32, Shape{batch_size, seq_length, input_size}); const auto X = make_shared<ov::op::v0::Parameter>(element::f32, Shape{batch_size, seq_length, input_size});
const auto initial_hidden_state = const auto initial_hidden_state =
make_shared<op::Parameter>(element::f32, Shape{batch_size, num_directions, hidden_size}); make_shared<ov::op::v0::Parameter>(element::f32, Shape{batch_size, num_directions, hidden_size});
const auto initial_cell_state = const auto initial_cell_state =
make_shared<op::Parameter>(element::f32, Shape{batch_size, num_directions, hidden_size}); make_shared<ov::op::v0::Parameter>(element::f32, Shape{batch_size, num_directions, hidden_size});
const auto sequence_lengths = make_shared<op::Parameter>(element::i32, Shape{batch_size}); const auto sequence_lengths = make_shared<ov::op::v0::Parameter>(element::i32, Shape{batch_size});
const auto W = make_shared<op::Parameter>(element::f32, Shape{num_directions, 4 * hidden_size, input_size}); const auto W = make_shared<ov::op::v0::Parameter>(element::f32, Shape{num_directions, 4 * hidden_size, input_size});
const auto R = make_shared<op::Parameter>(element::f32, Shape{num_directions, 4 * hidden_size, hidden_size}); const auto R =
const auto B = make_shared<op::Parameter>(element::f32, Shape{num_directions, 4 * hidden_size}); make_shared<ov::op::v0::Parameter>(element::f32, Shape{num_directions, 4 * hidden_size, hidden_size});
const auto P = make_shared<op::Parameter>(element::f32, Shape{num_directions, 3 * hidden_size}); const auto B = make_shared<ov::op::v0::Parameter>(element::f32, Shape{num_directions, 4 * hidden_size});
const auto P = make_shared<ov::op::v0::Parameter>(element::f32, Shape{num_directions, 3 * hidden_size});
const auto lstm_direction = op::RecurrentSequenceDirection::BIDIRECTIONAL; const auto lstm_direction = op::RecurrentSequenceDirection::BIDIRECTIONAL;
const ov::op::LSTMWeightsFormat weights_format = ov::op::LSTMWeightsFormat::FICO; const ov::op::LSTMWeightsFormat weights_format = ov::op::LSTMWeightsFormat::FICO;
@ -93,7 +91,7 @@ TEST(attributes, lstm_sequence_v1_op) {
const float clip_threshold = 0.5f; const float clip_threshold = 0.5f;
const bool input_forget = true; const bool input_forget = true;
const auto lstm_sequence = make_shared<opset1::LSTMSequence>(X, const auto lstm_sequence = make_shared<ov::op::v0::LSTMSequence>(X,
initial_hidden_state, initial_hidden_state,
initial_cell_state, initial_cell_state,
sequence_lengths, sequence_lengths,
@ -110,7 +108,7 @@ TEST(attributes, lstm_sequence_v1_op) {
clip_threshold, clip_threshold,
input_forget); input_forget);
NodeBuilder builder(lstm_sequence, {X, initial_hidden_state, initial_cell_state, sequence_lengths, W, R, B, P}); NodeBuilder builder(lstm_sequence, {X, initial_hidden_state, initial_cell_state, sequence_lengths, W, R, B, P});
auto g_lstm_sequence = ov::as_type_ptr<opset1::LSTMSequence>(builder.create()); auto g_lstm_sequence = ov::as_type_ptr<ov::op::v0::LSTMSequence>(builder.create());
EXPECT_EQ(g_lstm_sequence->get_hidden_size(), lstm_sequence->get_hidden_size()); EXPECT_EQ(g_lstm_sequence->get_hidden_size(), lstm_sequence->get_hidden_size());
EXPECT_EQ(g_lstm_sequence->get_activations(), lstm_sequence->get_activations()); EXPECT_EQ(g_lstm_sequence->get_activations(), lstm_sequence->get_activations());

View File

@ -2,102 +2,98 @@
// SPDX-License-Identifier: Apache-2.0 // SPDX-License-Identifier: Apache-2.0
// //
#include "common_test_utils/visitor.hpp" #include "openvino/op/matmul.hpp"
#include "gtest/gtest.h"
#include "ngraph/ngraph.hpp" #include <gtest/gtest.h>
#include "ngraph/op/util/attr_types.hpp"
#include "ngraph/opsets/opset1.hpp" #include "visitors/visitors.hpp"
#include "ngraph/opsets/opset3.hpp"
#include "ngraph/opsets/opset4.hpp"
#include "ngraph/opsets/opset5.hpp"
using namespace std; using namespace std;
using namespace ngraph; using namespace ov;
using ngraph::test::NodeBuilder; using ov::test::NodeBuilder;
using ngraph::test::ValueMap;
TEST(attributes, matmul_op) { TEST(attributes, matmul_op) {
NodeBuilder::get_ops().register_factory<opset1::MatMul>(); NodeBuilder::get_ops().register_factory<ov::op::v0::MatMul>();
auto A = make_shared<op::Parameter>(element::f32, Shape{0, 2}); auto A = make_shared<ov::op::v0::Parameter>(element::f32, Shape{0, 2});
auto B = make_shared<op::Parameter>(element::f32, Shape{2, 0}); auto B = make_shared<ov::op::v0::Parameter>(element::f32, Shape{2, 0});
bool transpose_a = true; bool transpose_a = true;
bool transpose_b = true; bool transpose_b = true;
auto matmul = make_shared<opset1::MatMul>(A, B, transpose_a, transpose_b); auto matmul = make_shared<ov::op::v0::MatMul>(A, B, transpose_a, transpose_b);
NodeBuilder builder(matmul, {A, B}); NodeBuilder builder(matmul, {A, B});
auto g_matmul = ov::as_type_ptr<opset1::MatMul>(builder.create()); auto g_matmul = ov::as_type_ptr<ov::op::v0::MatMul>(builder.create());
EXPECT_EQ(g_matmul->get_transpose_a(), matmul->get_transpose_a()); EXPECT_EQ(g_matmul->get_transpose_a(), matmul->get_transpose_a());
EXPECT_EQ(g_matmul->get_transpose_b(), matmul->get_transpose_b()); EXPECT_EQ(g_matmul->get_transpose_b(), matmul->get_transpose_b());
} }
TEST(attributes, matmul_op2) { TEST(attributes, matmul_op2) {
NodeBuilder::get_ops().register_factory<opset1::MatMul>(); NodeBuilder::get_ops().register_factory<ov::op::v0::MatMul>();
auto A = make_shared<op::Parameter>(element::f32, Shape{10, 2}); auto A = make_shared<ov::op::v0::Parameter>(element::f32, Shape{10, 2});
auto B = make_shared<op::Parameter>(element::f32, Shape{2, 1}); auto B = make_shared<ov::op::v0::Parameter>(element::f32, Shape{2, 1});
bool transpose_a = false; bool transpose_a = false;
bool transpose_b = false; bool transpose_b = false;
auto matmul = make_shared<opset1::MatMul>(A, B, transpose_a, transpose_b); auto matmul = make_shared<ov::op::v0::MatMul>(A, B, transpose_a, transpose_b);
NodeBuilder builder(matmul, {A, B}); NodeBuilder builder(matmul, {A, B});
auto g_matmul = ov::as_type_ptr<opset1::MatMul>(builder.create()); auto g_matmul = ov::as_type_ptr<ov::op::v0::MatMul>(builder.create());
EXPECT_EQ(g_matmul->get_transpose_a(), matmul->get_transpose_a()); EXPECT_EQ(g_matmul->get_transpose_a(), matmul->get_transpose_a());
EXPECT_EQ(g_matmul->get_transpose_b(), matmul->get_transpose_b()); EXPECT_EQ(g_matmul->get_transpose_b(), matmul->get_transpose_b());
} }
TEST(attributes, matmul_op3) { TEST(attributes, matmul_op3) {
NodeBuilder::get_ops().register_factory<opset1::MatMul>(); NodeBuilder::get_ops().register_factory<ov::op::v0::MatMul>();
auto A = make_shared<op::Parameter>(element::f32, Shape{2, 10}); auto A = make_shared<ov::op::v0::Parameter>(element::f32, Shape{2, 10});
auto B = make_shared<op::Parameter>(element::f32, Shape{2, 1}); auto B = make_shared<ov::op::v0::Parameter>(element::f32, Shape{2, 1});
bool transpose_a = true; bool transpose_a = true;
bool transpose_b = false; bool transpose_b = false;
auto matmul = make_shared<opset1::MatMul>(A, B, transpose_a, transpose_b); auto matmul = make_shared<ov::op::v0::MatMul>(A, B, transpose_a, transpose_b);
NodeBuilder builder(matmul, {A, B}); NodeBuilder builder(matmul, {A, B});
auto g_matmul = ov::as_type_ptr<opset1::MatMul>(builder.create()); auto g_matmul = ov::as_type_ptr<ov::op::v0::MatMul>(builder.create());
EXPECT_EQ(g_matmul->get_transpose_a(), matmul->get_transpose_a()); EXPECT_EQ(g_matmul->get_transpose_a(), matmul->get_transpose_a());
EXPECT_EQ(g_matmul->get_transpose_b(), matmul->get_transpose_b()); EXPECT_EQ(g_matmul->get_transpose_b(), matmul->get_transpose_b());
} }
TEST(attributes, matmul_op4) { TEST(attributes, matmul_op4) {
NodeBuilder::get_ops().register_factory<opset1::MatMul>(); NodeBuilder::get_ops().register_factory<ov::op::v0::MatMul>();
auto A = make_shared<op::Parameter>(element::f32, Shape{2, 3, 2}); auto A = make_shared<ov::op::v0::Parameter>(element::f32, Shape{2, 3, 2});
auto B = make_shared<op::Parameter>(element::f32, Shape{3, 2, 2, 1}); auto B = make_shared<ov::op::v0::Parameter>(element::f32, Shape{3, 2, 2, 1});
auto matmul = make_shared<opset1::MatMul>(A, B); auto matmul = make_shared<ov::op::v0::MatMul>(A, B);
NodeBuilder builder(matmul, {A, B}); NodeBuilder builder(matmul, {A, B});
auto g_matmul = ov::as_type_ptr<opset1::MatMul>(builder.create()); auto g_matmul = ov::as_type_ptr<ov::op::v0::MatMul>(builder.create());
EXPECT_EQ(g_matmul->get_transpose_a(), matmul->get_transpose_a()); EXPECT_EQ(g_matmul->get_transpose_a(), matmul->get_transpose_a());
EXPECT_EQ(g_matmul->get_transpose_b(), matmul->get_transpose_b()); EXPECT_EQ(g_matmul->get_transpose_b(), matmul->get_transpose_b());
} }
TEST(attributes, matmul_op5) { TEST(attributes, matmul_op5) {
NodeBuilder::get_ops().register_factory<opset1::MatMul>(); NodeBuilder::get_ops().register_factory<ov::op::v0::MatMul>();
auto A = make_shared<op::Parameter>(element::f32, Shape{2}); auto A = make_shared<ov::op::v0::Parameter>(element::f32, Shape{2});
auto B = make_shared<op::Parameter>(element::f32, Shape{2, 10}); auto B = make_shared<ov::op::v0::Parameter>(element::f32, Shape{2, 10});
auto matmul = make_shared<opset1::MatMul>(A, B); auto matmul = make_shared<ov::op::v0::MatMul>(A, B);
NodeBuilder builder(matmul, {A, B}); NodeBuilder builder(matmul, {A, B});
auto g_matmul = ov::as_type_ptr<opset1::MatMul>(builder.create()); auto g_matmul = ov::as_type_ptr<ov::op::v0::MatMul>(builder.create());
EXPECT_EQ(g_matmul->get_transpose_a(), matmul->get_transpose_a()); EXPECT_EQ(g_matmul->get_transpose_a(), matmul->get_transpose_a());
EXPECT_EQ(g_matmul->get_transpose_b(), matmul->get_transpose_b()); EXPECT_EQ(g_matmul->get_transpose_b(), matmul->get_transpose_b());
} }
TEST(attributes, matmul_op6) { TEST(attributes, matmul_op6) {
NodeBuilder::get_ops().register_factory<opset1::MatMul>(); NodeBuilder::get_ops().register_factory<ov::op::v0::MatMul>();
auto A = make_shared<op::Parameter>(element::f32, Shape{1, 2048}); auto A = make_shared<ov::op::v0::Parameter>(element::f32, Shape{1, 2048});
auto B = make_shared<op::Parameter>(element::f32, Shape{2048, 1000}); auto B = make_shared<ov::op::v0::Parameter>(element::f32, Shape{2048, 1000});
auto matmul = make_shared<opset1::MatMul>(A, B); auto matmul = make_shared<ov::op::v0::MatMul>(A, B);
NodeBuilder builder(matmul, {A, B}); NodeBuilder builder(matmul, {A, B});
auto g_matmul = ov::as_type_ptr<opset1::MatMul>(builder.create()); auto g_matmul = ov::as_type_ptr<ov::op::v0::MatMul>(builder.create());
EXPECT_EQ(g_matmul->get_transpose_a(), matmul->get_transpose_a()); EXPECT_EQ(g_matmul->get_transpose_a(), matmul->get_transpose_a());
EXPECT_EQ(g_matmul->get_transpose_b(), matmul->get_transpose_b()); EXPECT_EQ(g_matmul->get_transpose_b(), matmul->get_transpose_b());

View File

@ -2,42 +2,37 @@
// SPDX-License-Identifier: Apache-2.0 // SPDX-License-Identifier: Apache-2.0
// //
#include "common_test_utils/visitor.hpp" #include "openvino/op/matrix_nms.hpp"
#include "gtest/gtest.h"
#include "ngraph/ngraph.hpp" #include <gtest/gtest.h>
#include "ngraph/op/util/attr_types.hpp"
#include "ngraph/opsets/opset1.hpp" #include "visitors/visitors.hpp"
#include "ngraph/opsets/opset3.hpp"
#include "ngraph/opsets/opset4.hpp"
#include "ngraph/opsets/opset5.hpp"
#include "ngraph/opsets/opset8.hpp"
using namespace std; using namespace std;
using namespace ngraph; using namespace ov;
using ngraph::test::NodeBuilder; using ov::test::NodeBuilder;
using ngraph::test::ValueMap;
TEST(attributes, matrix_nms_v8_op_custom_attributes) { TEST(attributes, matrix_nms_v8_op_custom_attributes) {
NodeBuilder::get_ops().register_factory<opset8::MatrixNms>(); NodeBuilder::get_ops().register_factory<ov::op::v8::MatrixNms>();
auto boxes = make_shared<op::Parameter>(element::f32, Shape{1, 1, 4}); auto boxes = make_shared<ov::op::v0::Parameter>(element::f32, Shape{1, 1, 4});
auto scores = make_shared<op::Parameter>(element::f32, Shape{1, 1, 1}); auto scores = make_shared<ov::op::v0::Parameter>(element::f32, Shape{1, 1, 1});
opset8::MatrixNms::Attributes attrs; ov::op::v8::MatrixNms::Attributes attrs;
attrs.sort_result_type = opset8::MatrixNms::SortResultType::SCORE; attrs.sort_result_type = ov::op::v8::MatrixNms::SortResultType::SCORE;
attrs.output_type = ngraph::element::i32; attrs.output_type = ov::element::i32;
attrs.nms_top_k = 100; attrs.nms_top_k = 100;
attrs.keep_top_k = 10; attrs.keep_top_k = 10;
attrs.sort_result_across_batch = true; attrs.sort_result_across_batch = true;
attrs.score_threshold = 0.1f; attrs.score_threshold = 0.1f;
attrs.background_class = 2; attrs.background_class = 2;
attrs.decay_function = opset8::MatrixNms::DecayFunction::GAUSSIAN; attrs.decay_function = ov::op::v8::MatrixNms::DecayFunction::GAUSSIAN;
attrs.gaussian_sigma = 0.2f; attrs.gaussian_sigma = 0.2f;
attrs.post_threshold = 0.3f; attrs.post_threshold = 0.3f;
attrs.normalized = false; attrs.normalized = false;
auto nms = make_shared<opset8::MatrixNms>(boxes, scores, attrs); auto nms = make_shared<ov::op::v8::MatrixNms>(boxes, scores, attrs);
NodeBuilder builder(nms, {boxes, scores}); NodeBuilder builder(nms, {boxes, scores});
auto g_nms = ov::as_type_ptr<opset8::MatrixNms>(builder.create()); auto g_nms = ov::as_type_ptr<ov::op::v8::MatrixNms>(builder.create());
const auto expected_attr_count = 11; const auto expected_attr_count = 11;
EXPECT_EQ(builder.get_value_map_size(), expected_attr_count); EXPECT_EQ(builder.get_value_map_size(), expected_attr_count);
@ -70,13 +65,13 @@ TEST(attributes, matrix_nms_v8_op_custom_attributes) {
} }
TEST(attributes, matrix_nms_v8_op_default_attributes) { TEST(attributes, matrix_nms_v8_op_default_attributes) {
NodeBuilder::get_ops().register_factory<opset8::MatrixNms>(); NodeBuilder::get_ops().register_factory<ov::op::v8::MatrixNms>();
auto boxes = make_shared<op::Parameter>(element::f32, Shape{1, 1, 4}); auto boxes = make_shared<ov::op::v0::Parameter>(element::f32, Shape{1, 1, 4});
auto scores = make_shared<op::Parameter>(element::f32, Shape{1, 1, 1}); auto scores = make_shared<ov::op::v0::Parameter>(element::f32, Shape{1, 1, 1});
auto nms = make_shared<opset8::MatrixNms>(boxes, scores, opset8::MatrixNms::Attributes()); auto nms = make_shared<ov::op::v8::MatrixNms>(boxes, scores, ov::op::v8::MatrixNms::Attributes());
NodeBuilder builder(nms, {boxes, scores}); NodeBuilder builder(nms, {boxes, scores});
auto g_nms = ov::as_type_ptr<opset8::MatrixNms>(builder.create()); auto g_nms = ov::as_type_ptr<ov::op::v8::MatrixNms>(builder.create());
const auto expected_attr_count = 11; const auto expected_attr_count = 11;
EXPECT_EQ(builder.get_value_map_size(), expected_attr_count); EXPECT_EQ(builder.get_value_map_size(), expected_attr_count);

View File

@ -2,21 +2,19 @@
// SPDX-License-Identifier: Apache-2.0 // SPDX-License-Identifier: Apache-2.0
// //
#include "common_test_utils/visitor.hpp" #include "openvino/op/max_pool.hpp"
#include "gtest/gtest.h"
#include "ngraph/ngraph.hpp" #include <gtest/gtest.h>
#include "ngraph/op/util/attr_types.hpp"
#include "ngraph/opsets/opset1.hpp" #include "visitors/visitors.hpp"
#include "ngraph/opsets/opset8.hpp"
using namespace std; using namespace std;
using namespace ngraph; using namespace ov;
using ngraph::test::NodeBuilder; using ov::test::NodeBuilder;
using ngraph::test::ValueMap;
TEST(attributes, max_pool_op) { TEST(attributes, max_pool_op) {
NodeBuilder::get_ops().register_factory<opset1::MaxPool>(); NodeBuilder::get_ops().register_factory<ov::op::v1::MaxPool>();
auto data = make_shared<op::Parameter>(element::f32, Shape{64, 3, 5}); auto data = make_shared<ov::op::v0::Parameter>(element::f32, Shape{64, 3, 5});
auto strides = Strides{2}; auto strides = Strides{2};
auto pads_begin = Shape{1}; auto pads_begin = Shape{1};
@ -25,9 +23,10 @@ TEST(attributes, max_pool_op) {
auto rounding_mode = op::RoundingType::FLOOR; auto rounding_mode = op::RoundingType::FLOOR;
auto auto_pad = op::PadType::EXPLICIT; auto auto_pad = op::PadType::EXPLICIT;
auto max_pool = make_shared<opset1::MaxPool>(data, strides, pads_begin, pads_end, kernel, rounding_mode, auto_pad); auto max_pool =
make_shared<ov::op::v1::MaxPool>(data, strides, pads_begin, pads_end, kernel, rounding_mode, auto_pad);
NodeBuilder builder(max_pool, {data}); NodeBuilder builder(max_pool, {data});
auto g_max_pool = ov::as_type_ptr<opset1::MaxPool>(builder.create()); auto g_max_pool = ov::as_type_ptr<ov::op::v1::MaxPool>(builder.create());
EXPECT_EQ(g_max_pool->get_strides(), max_pool->get_strides()); EXPECT_EQ(g_max_pool->get_strides(), max_pool->get_strides());
EXPECT_EQ(g_max_pool->get_pads_begin(), max_pool->get_pads_begin()); EXPECT_EQ(g_max_pool->get_pads_begin(), max_pool->get_pads_begin());
@ -38,8 +37,8 @@ TEST(attributes, max_pool_op) {
} }
TEST(attributes, max_pool_v8_op) { TEST(attributes, max_pool_v8_op) {
NodeBuilder::get_ops().register_factory<opset8::MaxPool>(); NodeBuilder::get_ops().register_factory<ov::op::v8::MaxPool>();
const auto data = make_shared<op::Parameter>(element::i32, Shape{1, 3, 37, 37}); const auto data = make_shared<ov::op::v0::Parameter>(element::i32, Shape{1, 3, 37, 37});
const auto strides = Strides{1, 1}; const auto strides = Strides{1, 1};
const auto dilations = Strides{1, 1}; const auto dilations = Strides{1, 1};
@ -50,7 +49,7 @@ TEST(attributes, max_pool_v8_op) {
const auto auto_pad = op::PadType::EXPLICIT; const auto auto_pad = op::PadType::EXPLICIT;
const element::Type& index_element_type = element::i32; const element::Type& index_element_type = element::i32;
const auto max_pool = make_shared<opset8::MaxPool>(data, const auto max_pool = make_shared<ov::op::v8::MaxPool>(data,
strides, strides,
dilations, dilations,
pads_begin, pads_begin,
@ -60,7 +59,7 @@ TEST(attributes, max_pool_v8_op) {
auto_pad, auto_pad,
index_element_type); index_element_type);
NodeBuilder builder(max_pool, {data}); NodeBuilder builder(max_pool, {data});
auto g_max_pool = ov::as_type_ptr<opset8::MaxPool>(builder.create()); auto g_max_pool = ov::as_type_ptr<ov::op::v8::MaxPool>(builder.create());
EXPECT_EQ(g_max_pool->get_strides(), max_pool->get_strides()); EXPECT_EQ(g_max_pool->get_strides(), max_pool->get_strides());
EXPECT_EQ(g_max_pool->get_dilations(), max_pool->get_dilations()); EXPECT_EQ(g_max_pool->get_dilations(), max_pool->get_dilations());

View File

@ -2,9 +2,10 @@
// SPDX-License-Identifier: Apache-2.0 // SPDX-License-Identifier: Apache-2.0
// //
#include "binary_ops.hpp" #include "openvino/op/maximum.hpp"
#include "ngraph/opsets/opset1.hpp"
using Type = ::testing::Types<BinaryOperatorType<ngraph::opset1::Maximum, ngraph::element::f32>>; #include "binary_ops.hpp"
using Type = ::testing::Types<BinaryOperatorType<ov::op::v1::Maximum, ov::element::f32>>;
INSTANTIATE_TYPED_TEST_SUITE_P(visitor_with_auto_broadcast, BinaryOperatorVisitor, Type, BinaryOperatorTypeName); INSTANTIATE_TYPED_TEST_SUITE_P(visitor_with_auto_broadcast, BinaryOperatorVisitor, Type, BinaryOperatorTypeName);

View File

@ -2,9 +2,10 @@
// SPDX-License-Identifier: Apache-2.0 // SPDX-License-Identifier: Apache-2.0
// //
#include "binary_ops.hpp" #include "openvino/op/minimum.hpp"
#include "ngraph/opsets/opset1.hpp"
using Type = ::testing::Types<BinaryOperatorType<ngraph::opset1::Minimum, ngraph::element::f32>>; #include "binary_ops.hpp"
using Type = ::testing::Types<BinaryOperatorType<ov::op::v1::Minimum, ov::element::f32>>;
INSTANTIATE_TYPED_TEST_SUITE_P(visitor_with_auto_broadcast, BinaryOperatorVisitor, Type, BinaryOperatorTypeName); INSTANTIATE_TYPED_TEST_SUITE_P(visitor_with_auto_broadcast, BinaryOperatorVisitor, Type, BinaryOperatorTypeName);

View File

@ -2,23 +2,22 @@
// SPDX-License-Identifier: Apache-2.0 // SPDX-License-Identifier: Apache-2.0
// //
#include "common_test_utils/visitor.hpp" #include "openvino/op/mish.hpp"
#include "gtest/gtest.h" #include "gtest/gtest.h"
#include "ngraph/ngraph.hpp" #include "visitors/visitors.hpp"
#include "ngraph/op/util/attr_types.hpp"
#include "ngraph/opsets/opset4.hpp"
using namespace std; using namespace std;
using namespace ngraph; using namespace ov;
using ngraph::test::NodeBuilder; using ov::test::NodeBuilder;
TEST(attributes, mish_op) { TEST(attributes, mish_op) {
NodeBuilder::get_ops().register_factory<opset4::Mish>(); NodeBuilder::get_ops().register_factory<ov::op::v4::Mish>();
const auto A = make_shared<op::Parameter>(element::f32, Shape{5, 2}); const auto A = make_shared<ov::op::v0::Parameter>(element::f32, Shape{5, 2});
const auto mish = make_shared<opset4::Mish>(A); const auto mish = make_shared<ov::op::v4::Mish>(A);
NodeBuilder builder(mish, {A}); NodeBuilder builder(mish, {A});
EXPECT_NO_THROW(auto g_mish = ov::as_type_ptr<opset4::Mish>(builder.create())); EXPECT_NO_THROW(auto g_mish = ov::as_type_ptr<ov::op::v4::Mish>(builder.create()));
const auto expected_attr_count = 0; const auto expected_attr_count = 0;
EXPECT_EQ(builder.get_value_map_size(), expected_attr_count); EXPECT_EQ(builder.get_value_map_size(), expected_attr_count);

Some files were not shown because too many files have changed in this diff Show More