aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/contrib/lite/interpreter_test.cc
diff options
context:
space:
mode:
authorGravatar Andrew Selle <aselle@google.com>2018-02-13 12:24:02 -0800
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2018-02-13 12:27:40 -0800
commita3496e14f7ca0f77b804b5be87cd43f919a7c09f (patch)
tree8f7fe5756ceba80e15eb02e19f7fc038316d8327 /tensorflow/contrib/lite/interpreter_test.cc
parentca07b694fc5307b9eec6e3c449382d823b78a162 (diff)
Fix bug in populating the execution plan sent to the delegate.
- memcpy was missing the array size. - modified the unit test to verify that the execution plan is trivial on first delegate invocation. PiperOrigin-RevId: 185569606
Diffstat (limited to 'tensorflow/contrib/lite/interpreter_test.cc')
-rw-r--r--tensorflow/contrib/lite/interpreter_test.cc2
1 files changed, 2 insertions, 0 deletions
diff --git a/tensorflow/contrib/lite/interpreter_test.cc b/tensorflow/contrib/lite/interpreter_test.cc
index 4b309748f7..28c96e5dde 100644
--- a/tensorflow/contrib/lite/interpreter_test.cc
+++ b/tensorflow/contrib/lite/interpreter_test.cc
@@ -773,6 +773,8 @@ class TestDelegate : public ::testing::Test {
for (int exec_index = 0; exec_index < execution_plan->size;
exec_index++) {
int node_index = execution_plan->data[exec_index];
+ // Check that we are an identity map to start.
+ TFLITE_CHECK_EQ(exec_index, node_index);
TfLiteNode* node;
TfLiteRegistration* reg;
context->GetNodeAndRegistration(context, node_index, &node, &reg);