aboutsummaryrefslogtreecommitdiff
path: root/src/armnn/test/InferOutputTests.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/armnn/test/InferOutputTests.cpp')
-rw-r--r--src/armnn/test/InferOutputTests.cpp9
1 files changed, 9 insertions, 0 deletions
diff --git a/src/armnn/test/InferOutputTests.cpp b/src/armnn/test/InferOutputTests.cpp
index 24ae8b263b..4581d87a5b 100644
--- a/src/armnn/test/InferOutputTests.cpp
+++ b/src/armnn/test/InferOutputTests.cpp
@@ -31,4 +31,13 @@ ARMNN_SIMPLE_TEST_CASE(StackInferOutputShapeFromInputsNoMatch, StackInferOut
ARMNN_SIMPLE_TEST_CASE(StackValidateTensorShapesFromInputsMatch, StackValidateTensorShapesFromInputsMatchTest)
ARMNN_SIMPLE_TEST_CASE(StackValidateTensorShapesFromInputsNoMatch, StackValidateTensorShapesFromInputsNoMatchTest)
+// Convolution2D
+ARMNN_SIMPLE_TEST_CASE(Convolution2dInferOutputShape, Convolution2dInferOutputShapeTest)
+
+// DepthwiseConvolution2D
+ARMNN_SIMPLE_TEST_CASE(DepthwiseConvolution2dInferOutputShape, DepthwiseConvolution2dInferOutputShapeTest)
+
+// TransposeConvolution2D
+ARMNN_SIMPLE_TEST_CASE(TransposeConvolution2dInferOutputShape, TransposeConvolution2dInferOutputShapeTest)
+
BOOST_AUTO_TEST_SUITE_END()