aboutsummaryrefslogtreecommitdiff
path: root/delegate/src/Pack.hpp
diff options
context:
space:
mode:
Diffstat (limited to 'delegate/src/Pack.hpp')
-rw-r--r--delegate/src/Pack.hpp10
1 files changed, 10 insertions, 0 deletions
diff --git a/delegate/src/Pack.hpp b/delegate/src/Pack.hpp
index 59851cd637..5e93ba3f2a 100644
--- a/delegate/src/Pack.hpp
+++ b/delegate/src/Pack.hpp
@@ -99,6 +99,16 @@ TfLiteStatus VisitPackOperator(DelegateData& delegateData,
armnn::IConnectableLayer* layer = delegateData.m_Network->AddStackLayer(desc);
ARMNN_ASSERT(layer != nullptr);
+ // Connect the Constant Inputs
+ auto inputsTensorsProcess = ProcessInputs(layer,
+ delegateData,
+ tfLiteContext,
+ tfLiteNode);
+ if (inputsTensorsProcess == kTfLiteError)
+ {
+ return inputsTensorsProcess;
+ }
+
armnn::IOutputSlot& outputSlot = layer->GetOutputSlot(0);
outputSlot.SetTensorInfo(outputTensorInfo);