diff options
| author | Sadik Armagan <sadik.armagan@arm.com> | 2022-01-14 12:56:35 +0000 |
|---|---|---|
| committer | TeresaARM <teresa.charlinreyes@arm.com> | 2022-01-18 15:15:58 +0000 |
| commit | 529195f3ae752192d5f205345235bf537f472cbc (patch) | |
| tree | 2ee9dba45e0c181138725607133c6f4bf1a6828a /delegate/src/Pack.hpp | |
| parent | 611c7fb97412230d5cefee047081455fb60db06c (diff) | |
| download | armnn-529195f3ae752192d5f205345235bf537f472cbc.tar.gz | |
MLCE-736 'Adding support for TfLite Models'
* Added constant input supports for Pack/Stack, Concatenation operators
* Added Int32 support to Pack/Stack operator on CpuRef
* Removed unsupported operator from TfLite Delegate
Signed-off-by: Sadik Armagan <sadik.armagan@arm.com>
Change-Id: I64203e174300d23eedeb22bddefe07e931c4eff3
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); |