From 906f94631aa7ef590b9d8ff45507e818a0d1ac2c Mon Sep 17 00:00:00 2001 From: Jim Flynn Date: Fri, 10 May 2019 13:55:21 +0100 Subject: IVGCVSW-3076 Add ConcatLayer methods to public API !android-nn-driver:1120 Change-Id: I5192fa3deb4ea9766d38ad0bf4dfbfa0b4924c41 Signed-off-by: Jim Flynn --- src/armnnTfParser/TfParser.cpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'src/armnnTfParser/TfParser.cpp') 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)); -- cgit v1.2.1