diff options
Diffstat (limited to 'delegate/src/Control.hpp')
-rw-r--r-- | delegate/src/Control.hpp | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/delegate/src/Control.hpp b/delegate/src/Control.hpp index 2f83d2a37e..17f23d81ad 100644 --- a/delegate/src/Control.hpp +++ b/delegate/src/Control.hpp @@ -303,6 +303,13 @@ TfLiteStatus VisitMeanOperator(DelegateData& delegateData, armnn::IOutputSlot& outputSlot = meanLayer->GetOutputSlot(0); outputSlot.SetTensorInfo(outputTensorInfo); + + // try to connect the Constant Inputs if there are any + if(ProcessInputs(meanLayer,delegateData, tfLiteContext, tfLiteNode) != kTfLiteOk ) + { + return kTfLiteError; + } + return Connect(meanLayer, tfLiteNode, delegateData); } |