diff options
| author | James Conroy <james.conroy@arm.com> | 2019-10-08 15:41:34 +0100 |
|---|---|---|
| committer | Jim Flynn Arm <jim.flynn@arm.com> | 2019-10-10 08:48:39 +0000 |
| commit | c8724c7b9ff663538bd32ad789dbcc3e1aa88637 (patch) | |
| tree | 629a55e0f949fd9bede7e3b34cdc7ff337f9762c /src/armnn/test/InferOutputTests.cpp | |
| parent | 92bbcaed655d1dfe696f12f264599589b8ada602 (diff) | |
| download | armnn-c8724c7b9ff663538bd32ad789dbcc3e1aa88637.tar.gz | |
IVGCVSW-3944 Add ArgMinMax output shape validation
Signed-off-by: James Conroy <james.conroy@arm.com>
Change-Id: I469895da158b062cd19248832525fa21527f7d41
Diffstat (limited to 'src/armnn/test/InferOutputTests.cpp')
| -rw-r--r-- | src/armnn/test/InferOutputTests.cpp | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/src/armnn/test/InferOutputTests.cpp b/src/armnn/test/InferOutputTests.cpp index 8606745623..3293cef0f7 100644 --- a/src/armnn/test/InferOutputTests.cpp +++ b/src/armnn/test/InferOutputTests.cpp @@ -11,6 +11,12 @@ BOOST_AUTO_TEST_SUITE(LayerValidateOutput) +// ArgMinMax +ARMNN_SIMPLE_TEST_CASE(ArgMinMaxInferOutputShape4d, ArgMinMaxInferOutputShape4dTest) +ARMNN_SIMPLE_TEST_CASE(ArgMinMaxInferOutputShape3d, ArgMinMaxInferOutputShape3dTest) +ARMNN_SIMPLE_TEST_CASE(ArgMinMaxInferOutputShape2d, ArgMinMaxInferOutputShape2dTest) +ARMNN_SIMPLE_TEST_CASE(ArgMinMaxInferOutputShape1d, ArgMinMaxInferOutputShape1dTest) + // BatchToSpace ARMNN_SIMPLE_TEST_CASE(BatchToSpaceInferOutputShape, BatchToSpaceInferOutputShapeTest) |