diff options
Diffstat (limited to 'delegate/src/Control.hpp')
-rw-r--r-- | delegate/src/Control.hpp | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/delegate/src/Control.hpp b/delegate/src/Control.hpp index 17f23d81ad..a3ea6e92a7 100644 --- a/delegate/src/Control.hpp +++ b/delegate/src/Control.hpp @@ -172,7 +172,10 @@ TfLiteStatus VisitConcatenationOperator(DelegateData& delegateData, armnn::IOutputSlot& outputSlot = concatenationLayer->GetOutputSlot(0); outputSlot.SetTensorInfo(outputTensorInfo); - Connect(concatenationLayer, tfLiteNode, delegateData); + if(Connect(concatenationLayer, tfLiteNode, delegateData) != kTfLiteOk) + { + return kTfLiteError; + } if (activationType == kTfLiteActNone) { |