add a e2e test for keras lstm fusion.

PiperOrigin-RevId: 299474392 Change-Id: I1cb636a046e893154552061289667a9a5e49680d
khaled-besrour · Mar 7, 2020 · 56ee67f · 56ee67f
1 parent 3874b28
commit 56ee67f
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tensorflow/lite/testing/model_coverage/model_coverage_lib.py b/tensorflow/lite/testing/model_coverage/model_coverage_lib.py
@@ -380,6 +380,8 @@ def compare_models_v2(tflite_model,
       tf_results = [tf_results[list(tf_results.keys())[0]]]
     else:
       tf_results = [tf_results[tflite_label] for tflite_label in tflite_labels]
+  else:
+    tf_results = [tf_results]
 
   for tf_result, tflite_result in zip(tf_results, tflite_results):
     np.testing.assert_almost_equal(tf_result, tflite_result, tolerance)