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

Added default value for broadcast to SquearedDifference operation (#4490)

Browse Source
This commit is contained in:
Gleb Kazantaev authored and GitHub committed 2021-02-26 07:44:08 +03:00
1 parent dcce0f549c
commit 53a96f4a5a
2 files changed
+8 -4
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

ngraph/core/src/op/squared_difference.cpp
+1 -1
View File
@@ -31,7 +31,7 @@ constexpr NodeTypeInfo op::SquaredDifference::type_info;
op::SquaredDifference::SquaredDifference()
: FusedOp()
, m_autobroadcast()
, m_autobroadcast(AutoBroadcastType::NUMPY)
{
}
ngraph/test/type_prop/squared_difference.cpp
+7 -3
View File
@@ -37,7 +37,11 @@ TEST(type_prop, squared_difference)
EXPECT_HAS_SUBSTRING(error.what(), std::string("Argument shapes are inconsistent"));
}
const auto clamp = make_shared<op::SquaredDifference>(x1, x3);
EXPECT_EQ(clamp->get_element_type(), element::f64);
EXPECT_EQ(clamp->get_shape(), (Shape{2, 2}));
const auto squared_diff = make_shared<op::SquaredDifference>(x1, x3);
EXPECT_EQ(squared_diff->get_element_type(), element::f64);
EXPECT_EQ(squared_diff->get_shape(), (Shape{2, 2}));
EXPECT_EQ(squared_diff->get_autob(), op::AutoBroadcastType::NUMPY);
const auto squared_diff_no_args = make_shared<op::SquaredDifference>();
EXPECT_EQ(squared_diff_no_args->get_autob(), op::AutoBroadcastType::NUMPY);
}