24 lines
970 B
Diff
24 lines
970 B
Diff
|
From 6841e522a3e7d48706a02e8819836e809f738682 Mon Sep 17 00:00:00 2001
|
||
|
|
From: Mihai Maruseac <mihaimaruseac@google.com>
|
||
|
|
Date: Wed, 28 Apr 2021 15:13:03 -0700
|
||
|
|
Subject: [PATCH] Prevent division by 0
|
||
|
|
|
||
|
|
PiperOrigin-RevId: 370995582
|
||
|
|
Change-Id: I670ffaf52d1ff8823ec31ea5f438f9125b402223
|
||
|
|
---
|
||
|
|
tensorflow/lite/kernels/svdf.cc | 1 +
|
||
|
|
1 file changed, 1 insertion(+)
|
||
|
|
|
||
|
|
diff --git a/tensorflow/lite/kernels/svdf.cc b/tensorflow/lite/kernels/svdf.cc
|
||
|
|
index 8f5c9a86bff5c..73024fd1e587a 100644
|
||
|
|
--- a/tensorflow/lite/kernels/svdf.cc
|
||
|
|
+++ b/tensorflow/lite/kernels/svdf.cc
|
||
|
|
@@ -99,6 +99,7 @@ TfLiteStatus Prepare(TfLiteContext* context, TfLiteNode* node) {
|
||
|
|
const int rank = params->rank;
|
||
|
|
const int batch_size = input->dims->data[0];
|
||
|
|
const int num_filters = weights_feature->dims->data[0];
|
||
|
|
+ TF_LITE_ENSURE(context, rank != 0);
|
||
|
|
TF_LITE_ENSURE_EQ(context, num_filters % rank, 0);
|
||
|
|
const int num_units = num_filters / rank;
|
||
|
|
const int memory_size = weights_time->dims->data[1];
|