diff --git a/testdata/dnn/tflite/generate.py b/testdata/dnn/tflite/generate.py index 975abeb4d..c14b128dc 100644 --- a/testdata/dnn/tflite/generate.py +++ b/testdata/dnn/tflite/generate.py @@ -154,3 +154,10 @@ def keras_to_tf(model, input_shape): leakyRelu, inp = keras_to_tf(leakyRelu, (1, 7, 7, 5)) save_tflite_model(leakyRelu, inp, 'leakyRelu') + +@tf.function(input_signature=[tf.TensorSpec(shape=[2, 1, 1, 4], dtype=tf.float32)]) +def strided_slice(x): + return x[-1:, ..., ::2] + +inp = np.random.standard_normal((2, 1, 1, 4)).astype(np.float32) +save_tflite_model(strided_slice, inp, 'strided_slice') diff --git a/testdata/dnn/tflite/strided_slice.tflite b/testdata/dnn/tflite/strided_slice.tflite new file mode 100644 index 000000000..1fee759fc Binary files /dev/null and b/testdata/dnn/tflite/strided_slice.tflite differ diff --git a/testdata/dnn/tflite/strided_slice_inp.npy b/testdata/dnn/tflite/strided_slice_inp.npy new file mode 100644 index 000000000..3467fc362 Binary files /dev/null and b/testdata/dnn/tflite/strided_slice_inp.npy differ diff --git a/testdata/dnn/tflite/strided_slice_out_Identity.npy b/testdata/dnn/tflite/strided_slice_out_Identity.npy new file mode 100644 index 000000000..339210877 Binary files /dev/null and b/testdata/dnn/tflite/strided_slice_out_Identity.npy differ