diff options
author | Aron Virginas-Tar <Aron.Virginas-Tar@arm.com> | 2019-09-19 14:39:37 +0100 |
---|---|---|
committer | Áron Virginás-Tar <aron.virginas-tar@arm.com> | 2019-09-20 14:49:41 +0000 |
commit | 8a1b2184f7261eba82ad8346a3ec8bb7d5800f57 (patch) | |
tree | 86b47d84fdf56c308c8aee451538315d63334067 /src/armnn/layers | |
parent | dd6247f52dcb33bd03391b7ee573d7082e18cca7 (diff) | |
download | armnn-8a1b2184f7261eba82ad8346a3ec8bb7d5800f57.tar.gz |
Fix some minor issues around SpaceToDepth
* Removed unnecessary code from SpaceToDepthLayer::InferOutputShapes()
* Refactored SpaceToDepthQueueDescriptor::Validate() and added extra
checks for block size and output depth
Signed-off-by: Aron Virginas-Tar <Aron.Virginas-Tar@arm.com>
Change-Id: Ieeed3144e2589b2e8695ef65ce17752bc595332f
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 |