diff options
| author |  Shanqing Cai <cais@google.com> | 2018-03-12 19:33:52 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-03-12 19:37:39 -0700 |
| commit | 7144571f2fc59c8705e4e3d7b922fa0ebf44f3fa (patch) | |
| tree | b14683f826541c183c1bb783265e13b565469fbb /tensorflow/contrib/seq2seq | |
| parent | 2bda52d485c9715dcd17f49526cea7890e091cb8 (diff) | |
Merge changes from github.
PiperOrigin-RevId: 188817194
Diffstat (limited to 'tensorflow/contrib/seq2seq')
| -rw-r--r-- | tensorflow/contrib/seq2seq/python/ops/beam_search_decoder.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/contrib/seq2seq/python/ops/beam_search_decoder.py b/tensorflow/contrib/seq2seq/python/ops/beam_search_decoder.py index 6e57ccd6dd..03fe31abf7 100644 --- a/tensorflow/contrib/seq2seq/python/ops/beam_search_decoder.py +++ b/tensorflow/contrib/seq2seq/python/ops/beam_search_decoder.py @@ -722,7 +722,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]), |