Fix typo exporting the wrong tensor for TFLite model
This commit is contained in:
parent
1c71263790
commit
d6642da05b
@ -1828,7 +1828,7 @@ def create_inference_graph(batch_size=1, n_steps=16, use_new_decoder=False, tfli
|
|||||||
else:
|
else:
|
||||||
logits = tf.identity(logits, name='logits')
|
logits = tf.identity(logits, name='logits')
|
||||||
new_state_c = tf.identity(new_state_c, name='new_state_c')
|
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 (
|
return (
|
||||||
{
|
{
|
||||||
|
Loading…
x
Reference in New Issue
Block a user