From 24f4d46e56a4274ea0a6458748830ad824d96626 Mon Sep 17 00:00:00 2001 From: TFLM-bot Date: Thu, 29 Jan 2026 14:13:44 +0000 Subject: [PATCH] Sync from upstream TF. --- .../kernels/internal/reference/broadcast_to.h | 15 ++++++++++----- 1 file changed, 10 insertions(+), 5 deletions(-) diff --git a/tensorflow/lite/kernels/internal/reference/broadcast_to.h b/tensorflow/lite/kernels/internal/reference/broadcast_to.h index 0cd03db926d..66d86af4a50 100644 --- a/tensorflow/lite/kernels/internal/reference/broadcast_to.h +++ b/tensorflow/lite/kernels/internal/reference/broadcast_to.h @@ -29,11 +29,14 @@ void BroadcastImpl(const NdArrayDesc& input_desc, const char* input_data, const int type_size) { // Copy data from input to output. if (dim == last_broadcasting_dim) { - int copy_size = output_desc.strides[dim] * type_size; + size_t copy_size = + static_cast(output_desc.strides[dim]) * type_size; const char* data_src = - input_data + SubscriptToIndex(input_desc, indexes) * type_size; + input_data + + static_cast(SubscriptToIndex(input_desc, indexes)) * type_size; char* data_dst = - output_data + SubscriptToIndex(output_desc, indexes) * type_size; + output_data + + static_cast(SubscriptToIndex(output_desc, indexes)) * type_size; for (int i = 0; i < output_desc.extents[dim]; ++i, data_dst += copy_size) { memcpy(data_dst, data_src, copy_size); } @@ -50,9 +53,11 @@ void BroadcastImpl(const NdArrayDesc& input_desc, const char* input_data, // Duplicate data in output tensor. indexes[dim] = 0; if (input_desc.extents[dim] != output_desc.extents[dim]) { - int copy_size = output_desc.strides[dim] * type_size; + size_t copy_size = + static_cast(output_desc.strides[dim] * type_size); char* data_src = - output_data + SubscriptToIndex(output_desc, indexes) * type_size; + output_data + + static_cast(SubscriptToIndex(output_desc, indexes)) * type_size; char* data_dst = data_src + copy_size; for (int i = 1; i < output_desc.extents[dim]; ++i, data_dst += copy_size) { memcpy(data_dst, data_src, copy_size);