提交 bf4b21fa 编写于 作者: L Leo Zhao 提交者: baojun

fix assertion failure issue when test_analyzer_bert uses ngraph (#17148)

resolve #17147
test=develop
上级 626922d3
...@@ -35,7 +35,7 @@ std::shared_ptr<ngraph::Node> GetSoftmax(std::shared_ptr<ngraph::Node> x) { ...@@ -35,7 +35,7 @@ std::shared_ptr<ngraph::Node> GetSoftmax(std::shared_ptr<ngraph::Node> x) {
auto x_max = std::make_shared<ngraph::op::Max>(x, ngraph::AxisSet{1}); auto x_max = std::make_shared<ngraph::op::Max>(x, ngraph::AxisSet{1});
auto x_max_bcast = std::make_shared<ngraph::op::Broadcast>( auto x_max_bcast = std::make_shared<ngraph::op::Broadcast>(
x_max, x_shape, ngraph::AxisSet{1}); x_max, x_2d_shape, ngraph::AxisSet{1});
auto x_shifted = x - x_max_bcast; auto x_shifted = x - x_max_bcast;
auto x_clipped = auto x_clipped =
paddle::operators::ngraphs::ElementwiseScalar<ngraph::op::Maximum>( paddle::operators::ngraphs::ElementwiseScalar<ngraph::op::Maximum>(
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册