MLBEDSW-3561: Support const STRING input tensors
Added handling of input tensors with constant string data.
Signed-off-by: Louis Verhaard <louis.verhaard@arm.com>
Change-Id: Ieb5164a9d56d580ad08ea834bf2cbb7288cd9539
diff --git a/ethosu/vela/test/test_tflite_reader.py b/ethosu/vela/test/test_tflite_reader.py
index 14c9b20..a69e8d3 100644
--- a/ethosu/vela/test/test_tflite_reader.py
+++ b/ethosu/vela/test/test_tflite_reader.py
@@ -84,9 +84,9 @@
string_buffer_testdata = [
(np.array([np.random.randint(256) for _ in range(100)], dtype=np.uint8), [3, 5]),
- (np.array([np.random.randint(256) for _ in range(100)], dtype=np.int16), [10, 10]),
- (np.array([np.random.randint(256) for _ in range(100)], dtype=np.float32), [100]),
- (np.array([], dtype=np.int8), [30]),
+ (np.array([np.random.randint(256) for _ in range(100)], dtype=np.uint8), [10, 10]),
+ (np.array([np.random.randint(256) for _ in range(100)], dtype=np.uint8), []),
+ (np.array([], dtype=np.uint8), [30]),
]
@pytest.mark.parametrize("buffer, tens_shape", string_buffer_testdata)
@@ -111,4 +111,4 @@
subgraph = TFLiteSubgraph(graph, tfl_sg)
tens = subgraph.parse_tensor(tens_data)
- assert tens.values is None
+ assert np.array_equal(tens.values, buffer)