binding data to the first input tensor

uTensor · Nov 29, 2019 · 5baed55 · 5baed55
1 parent 1b4f614
commit 5baed55
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tests/tflm/tflite_export/test_write.py b/tests/tflm/tflite_export/test_write.py
@@ -94,6 +94,8 @@ def test_tflite_fb_write(hybrid_quant_output):
     open("tflm_test_model.tflite", "wb").write(model_content)
     test_model = tf.lite.Interpreter('tflm_test_model.tflite')
     test_model.allocate_tensors()
+    input_data = np.array(np.ones([4,1]), dtype=np.float32)
+    test_model.set_tensor(test_model.get_input_details()[0]['index'], input_data)
     test_model.invoke()
 
     print("0 :", test_model.get_tensor(0))