diff options
| author | Kevin May <kevin.may@arm.com> | 2022-12-15 10:15:21 +0000 |
|---|---|---|
| committer | Teresa Charlin <teresa.charlinreyes@arm.com> | 2022-12-15 17:16:12 +0000 |
| commit | 5b58e31109f661f70331048e22f8a66934e74d15 (patch) | |
| tree | 7ad8f88bf7427a1dd2622b53fcd0b4b8632ee34e /src/backends/tosaReference/TosaRefLayerSupport.cpp | |
| parent | 3fbad94012ee6b1c6ec8e93d760619b15d98c464 (diff) | |
| download | armnn-5b58e31109f661f70331048e22f8a66934e74d15.tar.gz | |
IVGCVSW-7170 Add Concat support to TOSA Reference Backend
* Change comment for the unique tensor names in all tosa common operators
Signed-off-by: Kevin May <kevin.may@arm.com>
Change-Id: I247b4b2365d5f0173218c5dfd11fba12d2399959
Diffstat (limited to 'src/backends/tosaReference/TosaRefLayerSupport.cpp')
| -rw-r--r-- | src/backends/tosaReference/TosaRefLayerSupport.cpp | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/src/backends/tosaReference/TosaRefLayerSupport.cpp b/src/backends/tosaReference/TosaRefLayerSupport.cpp index e5427ebc93..0d0d07a783 100644 --- a/src/backends/tosaReference/TosaRefLayerSupport.cpp +++ b/src/backends/tosaReference/TosaRefLayerSupport.cpp @@ -43,6 +43,13 @@ bool TosaRefLayerSupport::IsLayerSupported(const LayerType& type, inputInfos.push_back(&infos[1]); outputInfos.push_back(&infos[2]); break; + case LayerType::Concat: + for (unsigned int i = 0; i < infos.size() - 1; ++i) + { + inputInfos.push_back(&infos[i]); + } + outputInfos.push_back(&infos.back()); + break; case LayerType::Constant: outputInfos.push_back(&infos[0]); break; |