aboutsummaryrefslogtreecommitdiff
path: root/src/armnn
diff options
context:
space:
mode:
authorAron Virginas-Tar <Aron.Virginas-Tar@arm.com>2019-10-23 10:41:35 +0100
committerMatteo Martincigh <matteo.martincigh@arm.com>2019-10-23 13:19:45 +0000
commit85121a23c516428d8677f9988df3f40c1ceb2ea9 (patch)
tree5027f05d62e56b1960ed49490a5b12c7344751d0 /src/armnn
parentc95b8381ac73a975ad82b930f9ed3049901c6753 (diff)
downloadarmnn-85121a23c516428d8677f9988df3f40c1ceb2ea9.tar.gz
IVGCVSW-4010 Add serialization support for StandInLayer
Signed-off-by: Aron Virginas-Tar <Aron.Virginas-Tar@arm.com> Change-Id: I2017c15b485b82437c6ffdac4d3112bcc2cbe9e9
Diffstat (limited to 'src/armnn')
-rw-r--r--src/armnn/layers/StandInLayer.cpp6
1 files changed, 2 insertions, 4 deletions
diff --git a/src/armnn/layers/StandInLayer.cpp b/src/armnn/layers/StandInLayer.cpp
index fdc905fea2..7d693bfffb 100644
--- a/src/armnn/layers/StandInLayer.cpp
+++ b/src/armnn/layers/StandInLayer.cpp
@@ -10,7 +10,7 @@ namespace armnn
{
StandInLayer::StandInLayer(const StandInDescriptor& param, const char* name)
- : LayerWithParameters(param.m_NumInputs, 1, LayerType::StandIn, param, name)
+ : LayerWithParameters(param.m_NumInputs, param.m_NumOutputs, LayerType::StandIn, param, name)
{
}
@@ -42,6 +42,4 @@ void StandInLayer::Accept(ILayerVisitor& visitor) const
{
visitor.VisitStandInLayer(this, GetParameters(), GetName());
}
-} //namespace armnn
-
-
+} // namespace armnn