We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 1b4f614 commit 5baed55Copy full SHA for 5baed55
tests/tflm/tflite_export/test_write.py
@@ -94,6 +94,8 @@ def test_tflite_fb_write(hybrid_quant_output):
94
open("tflm_test_model.tflite", "wb").write(model_content)
95
test_model = tf.lite.Interpreter('tflm_test_model.tflite')
96
test_model.allocate_tensors()
97
+ input_data = np.array(np.ones([4,1]), dtype=np.float32)
98
+ test_model.set_tensor(test_model.get_input_details()[0]['index'], input_data)
99
test_model.invoke()
100
101
print("0 :", test_model.get_tensor(0))
0 commit comments