//***************************************************************************** // Copyright 2017-2019 Intel Corporation // // Licensed under the Apache License, Version 2.0 (the "License"); // you may not use this file except in compliance with the License. // You may obtain a copy of the License at // // http://www.apache.org/licenses/LICENSE-2.0 // // Unless required by applicable law or agreed to in writing, software // distributed under the License is distributed on an "AS IS" BASIS, // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. // See the License for the specific language governing permissions and // limitations under the License. //***************************************************************************** #include <memory> #include "gtest/gtest.h" #include "ngraph/file_util.hpp" #include "ngraph/graph_util.hpp" #include "ngraph/log.hpp" #include "ngraph/ngraph.hpp" #include "ngraph/op/add.hpp" #include "ngraph/op/constant.hpp" #include "ngraph/op/divide.hpp" #include "ngraph/op/multiply.hpp" #include "ngraph/op/product.hpp" #include "ngraph/op/sqrt.hpp" #include "ngraph/op/subtract.hpp" #include "ngraph/op/sum.hpp" #include "ngraph/pass/manager.hpp" #include "ngraph/pass/visualize_tree.hpp" #include "ngraph/pass/zero_dim_tensor_elimination.hpp" #include "util/test_tools.hpp" using namespace ngraph; using namespace std; TEST(zero_dim_tensor_elimination, zero_sum) { Shape zero_shape{0}; auto A = std::make_shared<op::Parameter>(element::i32, zero_shape); auto abs_node = std::make_shared<op::Abs>(A); auto sum_node = std::make_shared<op::Sum>(abs_node, AxisSet{0}); auto constant = std::make_shared<op::Constant>(element::i32, zero_shape, std::vector<string>{}); auto f = std::make_shared<Function>(NodeVector{sum_node, constant}, ParameterVector{A}); pass::Manager pass_manager; pass_manager.register_pass<pass::VisualizeTree>("zero_sum_before.png"); pass_manager.register_pass<ngraph::pass::ZeroDimTensorElimination>(); pass_manager.register_pass<pass::VisualizeTree>("zero_sum_after.png"); EXPECT_EQ(count_ops_of_type<op::Sum>(f), 1); pass_manager.run_passes(f); EXPECT_EQ(count_ops_of_type<op::Sum>(f), 0); } TEST(zero_dim_tensor_elimination, zero_product) { Shape zero_shape{0}; auto A = std::make_shared<op::Parameter>(element::i32, zero_shape); auto abs_node = std::make_shared<op::Abs>(A); auto product_node = std::make_shared<op::Product>(abs_node, AxisSet{0}); auto constant = std::make_shared<op::Constant>(element::i32, zero_shape, std::vector<string>{}); auto f = std::make_shared<Function>(NodeVector{product_node, constant}, ParameterVector{A}); pass::Manager pass_manager; pass_manager.register_pass<pass::VisualizeTree>("zero_product_before.png"); pass_manager.register_pass<ngraph::pass::ZeroDimTensorElimination>(); pass_manager.register_pass<pass::VisualizeTree>("zero_product_after.png"); EXPECT_EQ(count_ops_of_type<op::Product>(f), 1); pass_manager.run_passes(f); EXPECT_EQ(count_ops_of_type<op::Product>(f), 0); } TEST(zero_dim_tensor_elimination, zero_min) { Shape zero_shape{0}; auto A = std::make_shared<op::Parameter>(element::i32, zero_shape); auto abs_node = std::make_shared<op::Abs>(A); auto min_node = std::make_shared<op::Min>(abs_node, AxisSet{0}); auto constant = std::make_shared<op::Constant>(element::i32, zero_shape, std::vector<string>{}); auto f = std::make_shared<Function>(NodeVector{min_node, constant}, ParameterVector{A}); pass::Manager pass_manager; pass_manager.register_pass<pass::VisualizeTree>("zero_min_before.png"); pass_manager.register_pass<ngraph::pass::ZeroDimTensorElimination>(); pass_manager.register_pass<pass::VisualizeTree>("zero_min_after.png"); EXPECT_EQ(count_ops_of_type<op::Min>(f), 1); pass_manager.run_passes(f); EXPECT_EQ(count_ops_of_type<op::Min>(f), 0); } TEST(zero_dim_tensor_elimination, zero_max) { Shape zero_shape{0}; auto A = std::make_shared<op::Parameter>(element::i32, zero_shape); auto abs_node = std::make_shared<op::Abs>(A); auto max_node = std::make_shared<op::Max>(abs_node, AxisSet{0}); auto constant = std::make_shared<op::Constant>(element::i32, zero_shape, std::vector<string>{}); auto f = std::make_shared<Function>(NodeVector{max_node, constant}, ParameterVector{A}); pass::Manager pass_manager; pass_manager.register_pass<pass::VisualizeTree>("zero_max_before.png"); pass_manager.register_pass<ngraph::pass::ZeroDimTensorElimination>(); pass_manager.register_pass<pass::VisualizeTree>("zero_max_after.png"); EXPECT_EQ(count_ops_of_type<op::Max>(f), 1); pass_manager.run_passes(f); EXPECT_EQ(count_ops_of_type<op::Max>(f), 0); } TEST(zero_dim_tensor_elimination, zero_const_conv) { Shape zero_shape{0}; auto A = std::make_shared<op::Parameter>(element::f32, Shape{1, 1, 0}); auto weights = std::make_shared<op::Parameter>(element::f32, Shape{1, 1, 4}); auto convolution = std::make_shared<op::Convolution>( A, weights, Strides{1}, Strides{1}, CoordinateDiff{2}, CoordinateDiff{2}); auto abs_node = std::make_shared<op::Abs>(convolution); auto constant = std::make_shared<op::Constant>(element::i32, zero_shape, std::vector<string>{}); auto f = std::make_shared<Function>(NodeVector{abs_node, constant}, ParameterVector{A, weights}); pass::Manager pass_manager; pass_manager.register_pass<pass::VisualizeTree>("zero_const_conv_before.png"); pass_manager.register_pass<ngraph::pass::ZeroDimTensorElimination>(); pass_manager.register_pass<pass::VisualizeTree>("zero_const_conv_after.png"); EXPECT_EQ(count_ops_of_type<op::Convolution>(f), 1); pass_manager.run_passes(f); EXPECT_EQ(count_ops_of_type<op::Convolution>(f), 0); } TEST(zero_dim_tensor_elimination, zero_const_avg_pool) { Shape zero_shape{0}; auto A = std::make_shared<op::Parameter>(element::f32, Shape{1, 1, 0}); auto avg_pool = std::make_shared<op::AvgPool>(A, Shape{1}, Strides{1}, Shape{2}, Shape{2}, true); auto abs_node = std::make_shared<op::Abs>(avg_pool); auto constant = std::make_shared<op::Constant>(element::i32, zero_shape, std::vector<string>{}); auto f = std::make_shared<Function>(NodeVector{abs_node, constant}, ParameterVector{A}); pass::Manager pass_manager; pass_manager.register_pass<pass::VisualizeTree>("zero_const_avg_pool_before.png"); pass_manager.register_pass<ngraph::pass::ZeroDimTensorElimination>(); pass_manager.register_pass<pass::VisualizeTree>("zero_const_avg_pool_after.png"); EXPECT_EQ(count_ops_of_type<op::AvgPool>(f), 1); pass_manager.run_passes(f); EXPECT_EQ(count_ops_of_type<op::AvgPool>(f), 0); } TEST(zero_dim_tensor_elimination, zero_const_pad) { Shape zero_shape{0}; auto A = std::make_shared<op::Parameter>(element::f32, zero_shape); auto B = std::make_shared<op::Parameter>(element::f32, Shape{}); auto pad = std::make_shared<op::Pad>(A, B, CoordinateDiff{2}, CoordinateDiff{2}); auto abs_node = std::make_shared<op::Abs>(pad); auto constant = std::make_shared<op::Constant>(element::i32, zero_shape, std::vector<string>{}); auto f = std::make_shared<Function>(NodeVector{abs_node, constant}, ParameterVector{A, B}); pass::Manager pass_manager; pass_manager.register_pass<pass::VisualizeTree>("zero_const_pad_before.png"); pass_manager.register_pass<ngraph::pass::ZeroDimTensorElimination>(); pass_manager.register_pass<pass::VisualizeTree>("zero_const_pad_after.png"); EXPECT_EQ(count_ops_of_type<op::Broadcast>(f), 0); pass_manager.run_passes(f); EXPECT_EQ(count_ops_of_type<op::Broadcast>(f), 1); } TEST(zero_dim_tensor_elimination, zero_const_slice) { Shape zero_shape{0}; auto A = std::make_shared<op::Parameter>(element::f32, zero_shape); auto B = std::make_shared<op::Parameter>(element::f32, Shape{}); auto slice = make_shared<op::Slice>(A, Coordinate{0}, Coordinate{0}); auto pad = std::make_shared<op::Pad>(A, B, CoordinateDiff{2}, CoordinateDiff{2}); auto abs_node = std::make_shared<op::Abs>(pad); auto constant = std::make_shared<op::Constant>(element::i32, zero_shape, std::vector<string>{}); auto f = std::make_shared<Function>(NodeVector{abs_node, constant}, ParameterVector{A, B}); pass::Manager pass_manager; pass_manager.register_pass<pass::VisualizeTree>("zero_const_slice_before.png"); pass_manager.register_pass<ngraph::pass::ZeroDimTensorElimination>(); pass_manager.register_pass<pass::VisualizeTree>("zero_const_slice_after.png"); EXPECT_EQ(count_ops_of_type<op::Broadcast>(f), 0); EXPECT_EQ(count_ops_of_type<op::Slice>(f), 0); pass_manager.run_passes(f); EXPECT_EQ(count_ops_of_type<op::Broadcast>(f), 1); EXPECT_EQ(count_ops_of_type<op::Slice>(f), 0); } TEST(zero_dim_tensor_elimination, zero_argmax) { auto A = std::make_shared<op::Parameter>(element::f32, Shape{0, 2, 3}); auto argmax = make_shared<op::ArgMax>(A, 1, element::i32); auto f = std::make_shared<Function>(NodeVector{argmax}, ParameterVector{A}); pass::Manager pass_manager; pass_manager.register_pass<pass::VisualizeTree>("zero_argmax_before.png"); pass_manager.register_pass<ngraph::pass::ZeroDimTensorElimination>(); pass_manager.register_pass<pass::VisualizeTree>("zero_argmax_after.png"); EXPECT_EQ(count_ops_of_type<op::ArgMax>(f), 1); pass_manager.run_passes(f); EXPECT_EQ(count_ops_of_type<op::ArgMax>(f), 0); EXPECT_EQ(f->get_results().at(0)->get_shape(), (Shape{0, 3})); } TEST(zero_dim_tensor_elimination, zero_argmin) { auto A = std::make_shared<op::Parameter>(element::f32, Shape{0, 2, 3}); auto argmin = make_shared<op::ArgMin>(A, 1, element::i32); auto f = std::make_shared<Function>(NodeVector{argmin}, ParameterVector{A}); pass::Manager pass_manager; pass_manager.register_pass<pass::VisualizeTree>("zero_argmin_before.png"); pass_manager.register_pass<ngraph::pass::ZeroDimTensorElimination>(); pass_manager.register_pass<pass::VisualizeTree>("zero_argmin_after.png"); EXPECT_EQ(count_ops_of_type<op::ArgMin>(f), 1); pass_manager.run_passes(f); EXPECT_EQ(count_ops_of_type<op::ArgMin>(f), 0); EXPECT_EQ(f->get_results().at(0)->get_shape(), (Shape{0, 3})); } TEST(zero_dim_tensor_elimination, pass_property) { auto pass = std::make_shared<ngraph::pass::ZeroDimTensorElimination>(); ASSERT_EQ(true, pass->get_property(pass::PassProperty::REQUIRE_STATIC_SHAPE)); ASSERT_EQ(false, pass->get_property(pass::PassProperty::CHANGE_DYNAMIC_STATE)); }