From 0bbad002df387541c97fac168fbb33eab44c4f50 Mon Sep 17 00:00:00 2001 From: liym27 <33742067+liym27@users.noreply.github.com> Date: Mon, 13 Apr 2020 11:47:11 +0800 Subject: [PATCH] [dy2static] Revert to usage of 'assign' in test_transformer (#23777) * Add the support of bool list for assign value, test=develop * Fix the assign op test case for bool dtype, test=develop * Revert to usage of 'assign' in test_transformer. test=develop Co-authored-by: wawltor --- .../unittests/dygraph_to_static/transformer_dygraph_model.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/python/paddle/fluid/tests/unittests/dygraph_to_static/transformer_dygraph_model.py b/python/paddle/fluid/tests/unittests/dygraph_to_static/transformer_dygraph_model.py index 27b24e120d7..6098cb3d213 100644 --- a/python/paddle/fluid/tests/unittests/dygraph_to_static/transformer_dygraph_model.py +++ b/python/paddle/fluid/tests/unittests/dygraph_to_static/transformer_dygraph_model.py @@ -584,8 +584,9 @@ class Transformer(Layer): [[0.] + [-inf] * (beam_size - 1)] * batch_size, dtype="float32")) - finished = fluid.layers.fill_constant( - shape=[batch_size, beam_size], value=0, dtype="bool") + finished = to_variable( + np.full( + [batch_size, beam_size], 0, dtype="bool")) trg_word = layers.fill_constant( shape=[batch_size * beam_size, 1], dtype="int64", value=bos_id) -- GitLab