27 lines
960 B
Diff
27 lines
960 B
Diff
|
From 8e45822aa0b9f5df4b4c64f221e64dc930a70a9d Mon Sep 17 00:00:00 2001
|
||
|
|
From: Mihai Maruseac <mihaimaruseac@google.com>
|
||
|
|
Date: Tue, 27 Apr 2021 17:46:10 -0700
|
||
|
|
Subject: [PATCH] Handle one more division by 0 in TFLite.
|
||
|
|
|
||
|
|
---
|
||
|
|
tensorflow/lite/kernels/gather_nd.cc | 3 +++
|
||
|
|
1 file changed, 3 insertions(+)
|
||
|
|
|
||
|
|
diff --git a/tensorflow/lite/kernels/gather_nd.cc b/tensorflow/lite/kernels/gather_nd.cc
|
||
|
|
index 288f7deb..5c9990a0 100644
|
||
|
|
--- a/tensorflow/lite/kernels/gather_nd.cc
|
||
|
|
+++ b/tensorflow/lite/kernels/gather_nd.cc
|
||
|
|
@@ -155,6 +155,9 @@ TfLiteStatus Eval(TfLiteContext* context, TfLiteNode* node) {
|
||
|
|
const TfLiteTensor* indices = GetInput(context, node, kIndices);
|
||
|
|
TfLiteTensor* output = GetOutput(context, node, kOutputTensor);
|
||
|
|
|
||
|
|
+ // Prevent division by 0 in the helper
|
||
|
|
+ TF_LITE_ENSURE(context, NumElements(params) > 0);
|
||
|
|
+
|
||
|
|
switch (indices->type) {
|
||
|
|
case kTfLiteInt32:
|
||
|
|
return EvalGatherNd<int32_t>(context, params, indices, output);
|
||
|
|
--
|
||
|
|
2.23.0
|
||
|
|
|