commit 80bbe517a6b82a84eb39979dd1881d844c7d3310
author Haoliang Zhang <haoliang@google.com> Wed Mar 30 15:24:39 2022 -0700
committer TensorFlower Gardener <gardener@tensorflow.org> Wed Mar 30 15:29:05 2022 -0700
tree 50fb44c7cdf025a22304d4f92b30a3ad588636b7
parent fa6895a1564f14f488c5019039e3f0addef49f32

In slice.cc, also set the output to dynamic if the input tensor has unspecified dimension. For example, if the input shape is [-1, 10], with begin=[0,0] and size=[5,5]. The current logic will fail at Prepare.

PiperOrigin-RevId: 438410823

tensorflow/lite/kernels/slice.cc

diff --git a/tensorflow/lite/kernels/slice.cc b/tensorflow/lite/kernels/slice.cc
index a68c3a8..4efaaaf 100644
--- a/tensorflow/lite/kernels/slice.cc
+++ b/tensorflow/lite/kernels/slice.cc
@@ -136,8 +136,9 @@
                      "Slice op only supports 1D-5D input arrays.");
 
   // Postpone allocation of output if any of the indexing tensors is not
-  // constant
-  if (!(IsConstantTensor(begin) && IsConstantTensor(size))) {
+  // constant, or the input tensor has dynamic dimension.
+  if (!(IsConstantTensor(begin) && IsConstantTensor(size)) ||
+      HasUnspecifiedDimension(input)) {
     SetTensorToDynamic(output);
     return kTfLiteOk;
   }