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

Revise abs (#3931)

Browse Source 
* remove type_prop test file for abs operator

* add abs operator to unary_ops

* remove abs type_prop from CMakeList
This commit is contained in:
Piotr Szmelczynski 2021-01-21 16:58:21 +01:00 committed by GitHub
parent 31dec9addf
commit 9857f6368d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 1 additions and 66 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
ngraph/test/CMakeLists.txt
View File

@ -104,7 +104,6 @@ set(SRC
shape.cpp
specialize_function.cpp
tensor.cpp
type_prop/abs.cpp
type_prop/assign.cpp
type_prop/avg_pool.cpp
type_prop/batch_norm.cpp

64
ngraph/test/type_prop/abs.cpp
View File

@ -1,64 +0,0 @@
//*****************************************************************************
// Copyright 2017-2021 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 "gtest/gtest.h"
#include "ngraph/ngraph.hpp"
using namespace std;
using namespace ngraph;
TEST(type_prop, abs_basic_shape_inference)
{
Shape data_shape{2, 2};
const auto param = make_shared<op::Parameter>(element::f32, data_shape);
const auto op = make_shared<op::Abs>(param);
ASSERT_EQ(op->get_shape(), (data_shape));
ASSERT_EQ(op->get_element_type(), element::f32);
}
TEST(type_prop, abs_incompatible_input_type)
{
Shape data_shape{3, 3};
const auto param = make_shared<op::Parameter>(element::boolean, data_shape);
ASSERT_THROW(make_shared<op::Abs>(param), ngraph::NodeValidationFailure);
}
TEST(type_prop, abs_dynamic_shape_2D)
{
const PartialShape data_shape{Dimension::dynamic(), 2};
const auto param = make_shared<op::Parameter>(element::f32, data_shape);
const auto op = make_shared<op::Abs>(param);
ASSERT_TRUE(op->get_output_partial_shape(0).same_scheme({Dimension::dynamic(), 2}));
ASSERT_EQ(op->get_element_type(), element::f32);
}
TEST(type_prop, abs_dynamic_shape_3D)
{
const PartialShape data_shape{Dimension::dynamic(), Dimension::dynamic(), 3};
const auto param = make_shared<op::Parameter>(element::f32, data_shape);
const auto op = make_shared<op::Abs>(param);
ASSERT_TRUE(op->get_output_partial_shape(0).same_scheme(
{Dimension::dynamic(), Dimension::dynamic(), 3}));
ASSERT_EQ(op->get_element_type(), element::f32);
}
TEST(type_prop, abs_dynamic_ok)
{
const auto param = make_shared<op::Parameter>(element::f32, PartialShape::dynamic());
auto ap = make_shared<op::Abs>(param);
ASSERT_EQ(ap->get_output_element_type(0), element::f32);
ASSERT_TRUE(ap->get_output_partial_shape(0).same_scheme(PartialShape::dynamic()));
}

2
ngraph/test/type_prop/unary_ops.cpp
View File

@ -107,6 +107,6 @@ REGISTER_TYPED_TEST_CASE_P(UnaryOperator,
dynamic_rank_input_shape_3D,
dynamic_rank_input_shape_full);
using Types = ::testing::Types<op::Acos, op::Asin>;
using Types = ::testing::Types<op::Acos, op::Asin, op::Abs>;
INSTANTIATE_TYPED_TEST_CASE_P(type_prop, UnaryOperator, Types);