IVGCVSW-3944 Add ArgMinMax output shape validation
authorJames Conroy <james.conroy@arm.com>
Tue, 8 Oct 2019 14:41:34 +0000 (15:41 +0100)
committerJim Flynn Arm <jim.flynn@arm.com>
Thu, 10 Oct 2019 08:48:39 +0000 (08:48 +0000)
commitc8724c7b9ff663538bd32ad789dbcc3e1aa88637
tree629a55e0f949fd9bede7e3b34cdc7ff337f9762c
parent92bbcaed655d1dfe696f12f264599589b8ada602
IVGCVSW-3944 Add ArgMinMax output shape validation

Signed-off-by: James Conroy <james.conroy@arm.com>
Change-Id: I469895da158b062cd19248832525fa21527f7d41
src/armnn/layers/ArgMinMaxLayer.cpp
src/armnn/layers/ArgMinMaxLayer.hpp
src/armnn/test/InferOutputTests.cpp
src/armnn/test/InferOutputTests.hpp
src/backends/backendsCommon/WorkloadData.cpp
src/backends/backendsCommon/test/layerTests/ArgMinMaxTestImpl.cpp
src/backends/neon/test/NeonLayerTests.cpp