From 0b5b1975760e0dcfbced63cd9890a0ab87f72f30 Mon Sep 17 00:00:00 2001 From: TFLM-bot Date: Fri, 10 Apr 2026 14:23:02 +0000 Subject: [PATCH] Sync from upstream TF. --- tensorflow/lite/kernels/internal/reference/concatenation.h | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tensorflow/lite/kernels/internal/reference/concatenation.h b/tensorflow/lite/kernels/internal/reference/concatenation.h index 4a82d7c502d..915492b1e92 100644 --- a/tensorflow/lite/kernels/internal/reference/concatenation.h +++ b/tensorflow/lite/kernels/internal/reference/concatenation.h @@ -17,6 +17,7 @@ limitations under the License. #define TENSORFLOW_LITE_KERNELS_INTERNAL_REFERENCE_CONCATENATION_H_ #include +#include #include "tensorflow/lite/kernels/internal/common.h" #include "tensorflow/lite/kernels/internal/compatibility.h" @@ -109,7 +110,7 @@ inline void Concatenation(const ConcatenationParams& params, // not garbage. // Note: output_shape.FlatSize() gives number of elements (nibbles). // Bytes needed: (elements + 1) / 2. - memset(output_ptr, 0, (output_shape.FlatSize() + 1) / 2); + memset(output_ptr, 0, (static_cast(output_shape.FlatSize()) + 1) / 2); int64_t output_offset = 0; for (int k = 0; k < outer_size; k++) {