experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Add sanity check for resize-bilinear input shape.

Browse Source 
PiperOrigin-RevId: 245618186
This commit is contained in:
Renjie Liu 2019-04-28 00:53:50 -07:00 committed by TensorFlower Gardener
parent d3e78f98ba
commit 3acc8eaf60
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
tensorflow/lite/kernels/resize_bilinear.cc
View File

@ -40,9 +40,12 @@ TfLiteStatus ResizeOutputTensor(TfLiteContext* context,
const TfLiteTensor* input, const TfLiteTensor* input,
const TfLiteTensor* size, const TfLiteTensor* size,
TfLiteTensor* output) { TfLiteTensor* output) {
const int32* size_data = GetTensorData<int32>(size);
// Sanity check, the up/down sampling size should always be positive.
TF_LITE_ENSURE(context, size_data[0] > 0);
TF_LITE_ENSURE(context, size_data[1] > 0);
TfLiteIntArray* output_size = TfLiteIntArrayCreate(4); TfLiteIntArray* output_size = TfLiteIntArrayCreate(4);
output_size->data[0] = input->dims->data[0]; output_size->data[0] = input->dims->data[0];
const int32* size_data = GetTensorData<int32>(size);
output_size->data[1] = size_data[0]; output_size->data[1] = size_data[0];
output_size->data[2] = size_data[1]; output_size->data[2] = size_data[1];
output_size->data[3] = input->dims->data[3]; output_size->data[3] = input->dims->data[3];