Fix typo exporting the wrong tensor for TFLite model

This commit is contained in:
Alexandre Lissy 2018-11-07 16:52:57 +01:00
parent 1c71263790
commit d6642da05b

View File

@ -1828,7 +1828,7 @@ def create_inference_graph(batch_size=1, n_steps=16, use_new_decoder=False, tfli
else:
logits = tf.identity(logits, name='logits')
new_state_c = tf.identity(new_state_c, name='new_state_c')
new_state_h = tf.identity(new_state_c, name='new_state_h')
new_state_h = tf.identity(new_state_h, name='new_state_h')
return (
{