update tests

xmos · Mar 21, 2024 · 5524f83 · 5524f83
1 parent ba5c1f5
commit 5524f83
Show file tree

Hide file tree

Showing 6 changed files with 3 additions and 2 deletions.
diff --git a/integration_tests/models/16x8/test_concatenate/generate.py b/integration_tests/models/16x8/test_concatenate/generate.py
@@ -6,7 +6,7 @@
 
 def generate_concatenate_model(input_shapes, axis):
     input_data = [tf.keras.Input(shape=input_shape, dtype=np.float32, batch_size=1) for input_shape in input_shapes]
-    concatenated_output = tf.keras.layers.Concatenate(axis=axis)(input_data)
+    concatenated_output = tf.concat(input_data, axis=axis)
     model = tf.keras.Model(inputs=input_data, outputs=concatenated_output)
     converter = tfl.TFLiteConverter.from_keras_model(model)
     def representative_dataset_gen():
@@ -28,6 +28,7 @@ def representative_dataset_gen():
 
 generate_concatenate_model([(64), (64)], 0)
 generate_concatenate_model([(2, 3), (2, 3)], 1)
-generate_concatenate_model([(2, 3, 5), (2, 3, 5)], 0)
+generate_concatenate_model([(2, 3, 5), (2, 3, 5)], -1)
 generate_concatenate_model([(2, 6, 5, 2), (2, 6, 5, 2)], 1)
+generate_concatenate_model([(2, 6, 5, 2), (2, 6, 5, 2)], 3)
 
diff --git a/integration_tests/models/16x8/test_concatenate/test_concatenate_0.tflite b/integration_tests/models/16x8/test_concatenate/test_concatenate_0.tflite
diff --git a/integration_tests/models/16x8/test_concatenate/test_concatenate_1.tflite b/integration_tests/models/16x8/test_concatenate/test_concatenate_1.tflite
diff --git a/integration_tests/models/16x8/test_concatenate/test_concatenate_2.tflite b/integration_tests/models/16x8/test_concatenate/test_concatenate_2.tflite
diff --git a/integration_tests/models/16x8/test_concatenate/test_concatenate_3.tflite b/integration_tests/models/16x8/test_concatenate/test_concatenate_3.tflite
diff --git a/integration_tests/models/16x8/test_concatenate/test_concatenate_4.tflite b/integration_tests/models/16x8/test_concatenate/test_concatenate_4.tflite