Pad dynamically allocated tensors with XNN_EXTRA_BYTES

Since XNNPack can now be applied to all models, these buffers may be passed to XNNPack. PiperOrigin-RevId: 676764594
osmaelo · Sep 20, 2024 · 37c5bd1 · 37c5bd1
1 parent 598ebd5
commit 37c5bd1
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tensorflow/lite/core/c/common.cc b/tensorflow/lite/core/c/common.cc
@@ -295,7 +295,8 @@ TfLiteStatus TfLiteTensorResizeMaybeCopy(size_t num_bytes, TfLiteTensor* tensor,
 #ifdef TF_LITE_TENSORFLOW_PROFILER
  tflite::PauseHeapMonitoring(/*pause=*/true);
 #endif
- size_t alloc_bytes = num_bytes;
+ // This buffer may be consumed by XNNPack.
+ size_t alloc_bytes = num_bytes + /*XNN_EXTRA_BYTES=*/16;
  // TODO(b/145340303): Tensor data should be aligned.
  if (!tensor->data.data) {
  tensor->data.data = (char*)malloc(alloc_bytes);