25 lines
1.2 KiB
Diff
25 lines
1.2 KiB
Diff
From 15691e456c7dc9bd6be203b09765b063bf4a380c Mon Sep 17 00:00:00 2001
|
|
From: Mihai Maruseac <mihaimaruseac@google.com>
|
|
Date: Tue, 27 Jul 2021 18:40:39 -0700
|
|
Subject: [PATCH] Prevent dereferencing of null pointers in TFLite's `add.cc`.
|
|
|
|
PiperOrigin-RevId: 387244946
|
|
Change-Id: I56094233327fbd8439b92e1dbb1262176e00eeb9
|
|
---
|
|
tensorflow/lite/kernels/internal/optimized/optimized_ops.h | 2 +-
|
|
1 file changed, 1 insertion(+), 1 deletion(-)
|
|
|
|
diff --git a/tensorflow/lite/kernels/internal/optimized/optimized_ops.h b/tensorflow/lite/kernels/internal/optimized/optimized_ops.h
|
|
index 241405a6bbae1..296b3923a257a 100644
|
|
--- a/tensorflow/lite/kernels/internal/optimized/optimized_ops.h
|
|
+++ b/tensorflow/lite/kernels/internal/optimized/optimized_ops.h
|
|
@@ -265,7 +265,7 @@ inline void BinaryBroadcastFiveFold(const ArithmeticParams& unswitched_params,
|
|
// We have broadcast y2*y3*y4 of input2 data y1 times, and now move on.
|
|
input2_data_reset = input2_data_ptr;
|
|
}
|
|
- } else {
|
|
+ } else if (input1_data_ptr != nullptr) {
|
|
// Special case of y4 == 1, in which the innermost loop is a single
|
|
// element and can be combined with the next (y3) as an inner broadcast.
|
|
//
|