experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

add a e2e test for keras lstm fusion.

Browse Source 
PiperOrigin-RevId: 299474392
Change-Id: I1cb636a046e893154552061289667a9a5e49680d
This commit is contained in:
Renjie Liu 2020-03-06 17:49:09 -08:00 committed by TensorFlower Gardener
parent 3874b28288
commit 56ee67f7b1
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/lite/testing/model_coverage/model_coverage_lib.py
View File

@ -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)