projects / platform / upstream / armnn.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: dd6247f) | patch
Fix some minor issues around SpaceToDepth
authorAron Virginas-Tar <Aron.Virginas-Tar@arm.com>
Thu, 19 Sep 2019 13:39:37 +0000 (14:39 +0100)
committerÁron Virginás-Tar <aron.virginas-tar@arm.com>
Fri, 20 Sep 2019 14:49:41 +0000 (14:49 +0000)
commit8a1b2184f7261eba82ad8346a3ec8bb7d5800f57
tree86b47d84fdf56c308c8aee451538315d63334067tree | snapshot
parentdd6247f52dcb33bd03391b7ee573d7082e18cca7commit | diff
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
src/armnn/layers/SpaceToDepthLayer.cpp diff | blob | history
src/backends/backendsCommon/WorkloadData.cpp diff | blob | history
Domain: Machine Learning / ML Framework;