commit 73bf1036696ddd9a7030035e5326223b737ac236
author Robert Neale <rneale@google.com> Tue Jun 07 12:10:32 2022 -0700
committer TensorFlower Gardener <gardener@tensorflow.org> Tue Jun 07 12:27:20 2022 -0700
tree a0f285a41fa35f86d7af799ff17eb4f58dbaed67
parent cbef1510b8bf0e7d0d9a3bb5a678d2d118822979

Dynamic TF Lite tensors using the TF Lite shim were not getting resized correctly. This fixes that problem and adds test for it on the byte_splitter op.

PiperOrigin-RevId: 453494714

tensorflow/lite/kernels/shim/tflite_op_shim.cc

diff --git a/tensorflow/lite/kernels/shim/tflite_op_shim.cc b/tensorflow/lite/kernels/shim/tflite_op_shim.cc
index 683b697..4f8b0e7 100644
--- a/tensorflow/lite/kernels/shim/tflite_op_shim.cc
+++ b/tensorflow/lite/kernels/shim/tflite_op_shim.cc
@@ -97,7 +97,8 @@
   if (tflite_tensor == nullptr)
     return absl::InternalError(
         absl::StrCat("output tensor is null during invocation. idx: ", idx));
-  if (tflite_tensor->data.raw == nullptr) {
+  if (tflite_tensor->data.raw == nullptr ||
+      tflite_tensor->allocation_type == kTfLiteDynamic) {
     TfLiteIntArray* output_shape_array =
         ShapeToTfLiteShape(output_shape.value());
     context_->ResizeTensor(context_, tflite_tensor, output_shape_array);