diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2018-07-25 10:14:24 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-07-25 10:23:11 -0700 |
commit | 0584b943f4eca8a5761480ebb524c930aa808f0d (patch) | |
tree | a3a2a611af1c6089add18f3c7e8e10bb60363ca9 /tensorflow/contrib/lite/interpreter.cc | |
parent | 67b9e8eafd826d3f430eb7f6780e815fcac5859e (diff) |
An ErrorReporter to be used in tests.
PiperOrigin-RevId: 206012444
Diffstat (limited to 'tensorflow/contrib/lite/interpreter.cc')
-rw-r--r-- | tensorflow/contrib/lite/interpreter.cc | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/tensorflow/contrib/lite/interpreter.cc b/tensorflow/contrib/lite/interpreter.cc index 5a5c907b6e..e38597495d 100644 --- a/tensorflow/contrib/lite/interpreter.cc +++ b/tensorflow/contrib/lite/interpreter.cc @@ -279,8 +279,9 @@ TfLiteStatus Interpreter::ReplaceSubgraphsWithDelegateKernels( int node_index; TfLiteDelegateParams* params = CreateDelegateParams(delegate, subgraph); - AddNodeWithParameters(subgraph.input_tensors, subgraph.output_tensors, - nullptr, 0, params, ®istration, &node_index); + TF_LITE_ENSURE_STATUS(AddNodeWithParameters( + subgraph.input_tensors, subgraph.output_tensors, nullptr, 0, params, + ®istration, &node_index)); // Initialize the output tensors's delegate-related fields. for (int tensor_index : subgraph.output_tensors) { |