diff options
author | Andrew Selle <aselle@google.com> | 2018-02-13 12:24:02 -0800 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-02-13 12:27:40 -0800 |
commit | a3496e14f7ca0f77b804b5be87cd43f919a7c09f (patch) | |
tree | 8f7fe5756ceba80e15eb02e19f7fc038316d8327 /tensorflow/contrib/lite/interpreter_test.cc | |
parent | ca07b694fc5307b9eec6e3c449382d823b78a162 (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.cc | 2 |
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, ®); |