diff options
Diffstat (limited to 'delegate/src/Pack.hpp')
-rw-r--r-- | delegate/src/Pack.hpp | 10 |
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); |