Skip to content

GitLab

Commit e3f8ea22 authored by Hongkun Yu's avatar Hongkun Yu Committed by A. Unique TensorFlower
Browse files

Internal change 

PiperOrigin-RevId: 345383199
parent abcb643f
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
official/nlp/modeling/ops/beam_search.py
View file @ e3f8ea22
...@@ -398,7 +398,7 @@ class SequenceBeamSearch(tf.Module): ...@@ -398,7 +398,7 @@ class SequenceBeamSearch(tf.Module):
raise TypeError( raise TypeError(
"initial_cache element for key '%s' has dtype %s that does not " "initial_cache element for key '%s' has dtype %s that does not "
"match SequenceBeamSearch's dtype of %s. Value: %s" % "match SequenceBeamSearch's dtype of %s. Value: %s" %
(key, value.dtype.name, self.dtype.name, inner_value)) (key, inner_value.dtype.name, self.dtype.name, inner_value))
# Current loop index (starts at 0) # Current loop index (starts at 0)
cur_index = tf.constant(0) cur_index = tf.constant(0)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment