diff options
Diffstat (limited to 'src/armnn/layers')
| -rw-r--r-- | src/armnn/layers/SpaceToDepthLayer.cpp | 4 |
1 files changed, 1 insertions, 3 deletions
diff --git a/src/armnn/layers/SpaceToDepthLayer.cpp b/src/armnn/layers/SpaceToDepthLayer.cpp index b24490f82f..8a9f1c296c 100644 --- a/src/armnn/layers/SpaceToDepthLayer.cpp +++ b/src/armnn/layers/SpaceToDepthLayer.cpp @@ -47,8 +47,6 @@ std::vector<TensorShape> SpaceToDepthLayer::InferOutputShapes(const std::vector< TensorShape inputShape = inputShapes[0]; TensorShape outputShape(inputShape); - outputShape[0] = inputShape[0]; - DataLayoutIndexed dimensionIndices{m_Param.m_DataLayout}; unsigned int hIndex = dimensionIndices.GetHeightIndex(); unsigned int wIndex = dimensionIndices.GetWidthIndex(); @@ -82,4 +80,4 @@ void SpaceToDepthLayer::Accept(ILayerVisitor& visitor) const visitor.VisitSpaceToDepthLayer(this, GetParameters(), GetName()); } -} // namespace armnn
\ No newline at end of file +} // namespace armnn |