diff options
author | TensorFlower Gardener <gardener@tensorflow.org> | 2018-08-10 11:56:12 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-08-10 11:56:12 -0700 |
commit | 295c89b42ede95996c39f3dfe9612c140b648885 (patch) | |
tree | 857e976e1db9434e245c3e63f3b556d728facf98 /tensorflow/contrib/training | |
parent | fea9d07d1e34d5330a13024cb42d9bc460869905 (diff) | |
parent | 523cf0cc83205652fbd8be579c52b18f97c9de54 (diff) |
Merge pull request #20413 from Horstage:master
PiperOrigin-RevId: 208244350
Diffstat (limited to 'tensorflow/contrib/training')
-rw-r--r-- | tensorflow/contrib/training/python/training/sequence_queueing_state_saver.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/tensorflow/contrib/training/python/training/sequence_queueing_state_saver.py b/tensorflow/contrib/training/python/training/sequence_queueing_state_saver.py index 39d75a0806..53e4f23a7c 100644 --- a/tensorflow/contrib/training/python/training/sequence_queueing_state_saver.py +++ b/tensorflow/contrib/training/python/training/sequence_queueing_state_saver.py @@ -988,14 +988,14 @@ class SequenceQueueingStateSaver(object): assert isinstance(sequences, dict) assert isinstance(context, dict) assert isinstance(states, dict) - self._name_to_index = dict( - (name, ix) + self._name_to_index = { + name: ix for (ix, name) in enumerate([ "__length", "__total_length", "__next_key", "__sequence", "__sequence_count" ] + ["__sequence__%s" % k for k in sequences.keys()] + [ "__context__%s" % k for k in context.keys() - ] + ["__state__%s" % k for k in states.keys()])) + ] + ["__state__%s" % k for k in states.keys()])} self._index_to_name = [ name for (name, _) in sorted( |