commit 22442927fe96b1b7756afe62ddeec9831572eeb0
author A. Unique TensorFlower <gardener@tensorflow.org> Thu Jan 16 11:40:06 2020 -0800
committer TensorFlower Gardener <gardener@tensorflow.org> Thu Jan 16 11:43:43 2020 -0800
tree 3c80e38fa37ecfb604c7ed992cb99a06d9676e50
parent ae0b693cb439dcc0af5f396e0a41705564d2d00d

Add type checking when preparing interpter inputs.

PiperOrigin-RevId: 290110585
Change-Id: Ie8e1968397a5dcb861b3664a1b860d79a7385377

tensorflow/lite/delegates/gpu/common/testing/interpreter_utils.cc

diff --git a/tensorflow/lite/delegates/gpu/common/testing/interpreter_utils.cc b/tensorflow/lite/delegates/gpu/common/testing/interpreter_utils.cc
index ed95de9..cbd62fa 100644
--- a/tensorflow/lite/delegates/gpu/common/testing/interpreter_utils.cc
+++ b/tensorflow/lite/delegates/gpu/common/testing/interpreter_utils.cc
@@ -47,8 +47,12 @@
     return InternalError("Unable to allocate TfLite tensors");
   }
   for (int i = 0; i < inputs.size(); ++i) {
+    DCHECK_EQ(interpreter->tensor(interpreter->inputs()[i])->type,
+              kTfLiteFloat32);
     float* tflite_data =
         interpreter->typed_tensor<float>(interpreter->inputs()[i]);
+    DCHECK_EQ(inputs[i].data.size() * sizeof(float),
+              interpreter->tensor(interpreter->inputs()[i])->bytes);
     std::memcpy(tflite_data, inputs[i].data.data(),
                 inputs[i].data.size() * sizeof(float));
   }