From bf4b21fa3d09ac7936acec3b77ffaa5ed2cf3a59 Mon Sep 17 00:00:00 2001 From: Leo Zhao <48052473+LeoZhao-Intel@users.noreply.github.com> Date: Tue, 30 Apr 2019 02:37:56 +0800 Subject: [PATCH] fix assertion failure issue when test_analyzer_bert uses ngraph (#17148) resolve #17147 test=develop --- paddle/fluid/operators/ngraph/ops/softmax_op.h | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/paddle/fluid/operators/ngraph/ops/softmax_op.h b/paddle/fluid/operators/ngraph/ops/softmax_op.h index 174b7a91a8d..6eb84703998 100644 --- a/paddle/fluid/operators/ngraph/ops/softmax_op.h +++ b/paddle/fluid/operators/ngraph/ops/softmax_op.h @@ -35,7 +35,7 @@ std::shared_ptr GetSoftmax(std::shared_ptr x) { auto x_max = std::make_shared(x, ngraph::AxisSet{1}); auto x_max_bcast = std::make_shared( - x_max, x_shape, ngraph::AxisSet{1}); + x_max, x_2d_shape, ngraph::AxisSet{1}); auto x_shifted = x - x_max_bcast; auto x_clipped = paddle::operators::ngraphs::ElementwiseScalar( -- GitLab