From 309fb111931bb7aae3e716594e5c53ac4976e76a Mon Sep 17 00:00:00 2001 From: "harumitsu.nobuta" Date: Tue, 20 Feb 2018 17:29:29 +0900 Subject: [PATCH] explicit dtype converting --- tensorflow/contrib/seq2seq/python/ops/beam_search_decoder.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tensorflow/contrib/seq2seq/python/ops/beam_search_decoder.py b/tensorflow/contrib/seq2seq/python/ops/beam_search_decoder.py index d6184d6..554eb24e 100644 --- a/tensorflow/contrib/seq2seq/python/ops/beam_search_decoder.py +++ b/tensorflow/contrib/seq2seq/python/ops/beam_search_decoder.py @@ -724,7 +724,7 @@ def _mask_probs(probs, eos_token, finished): eos_token, vocab_size, dtype=probs.dtype, - on_value=0., + on_value=ops.convert_to_tensor(0., dtype=probs.dtype), off_value=probs.dtype.min) finished_probs = array_ops.tile( array_ops.reshape(finished_row, [1, 1, -1]), -- 2.7.4