diff options
| author | Jim Flynn <jim.flynn@arm.com> | 2019-05-10 13:55:21 +0100 |
|---|---|---|
| committer | Matteo Martincigh <matteo.martincigh@arm.com> | 2019-05-14 10:35:28 +0000 |
| commit | 906f94631aa7ef590b9d8ff45507e818a0d1ac2c (patch) | |
| tree | bbb9d805b9424f92fe4f6a7030878e101873b03f /src/armnnTfParser/TfParser.cpp | |
| parent | 305620201159eb47e005a46d60f80c6a2c763d51 (diff) | |
| download | armnn-906f94631aa7ef590b9d8ff45507e818a0d1ac2c.tar.gz | |
IVGCVSW-3076 Add ConcatLayer methods to public API
!android-nn-driver:1120
Change-Id: I5192fa3deb4ea9766d38ad0bf4dfbfa0b4924c41
Signed-off-by: Jim Flynn <jim.flynn@arm.com>
Diffstat (limited to 'src/armnnTfParser/TfParser.cpp')
| -rwxr-xr-x | src/armnnTfParser/TfParser.cpp | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/src/armnnTfParser/TfParser.cpp b/src/armnnTfParser/TfParser.cpp index d7cfba8dfe..e5948d55f4 100755 --- a/src/armnnTfParser/TfParser.cpp +++ b/src/armnnTfParser/TfParser.cpp @@ -2120,7 +2120,7 @@ ParsedTfOperationPtr TfParser::ParseConcat(const tensorflow::NodeDef& nodeDef, // Update the output shape mergeDims[concatDim] = mergeDim; - armnn::IConnectableLayer *layer = m_Network->AddMergerLayer(concatDescriptor, nodeDef.name().c_str()); + armnn::IConnectableLayer *layer = m_Network->AddConcatLayer(concatDescriptor, nodeDef.name().c_str()); layer->GetOutputSlot(0).SetTensorInfo(armnn::TensorInfo(mergeDims, DataType::Float32)); |