aboutsummaryrefslogtreecommitdiff
path: root/src/armnn/layers/StandInLayer.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/armnn/layers/StandInLayer.cpp')
-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