From 98180eff3c1feed19576d82ceac06d476235d973 Mon Sep 17 00:00:00 2001 From: Aron Virginas-Tar Date: Wed, 26 Jun 2019 15:02:47 +0100 Subject: IVGCVSW-3324 Add end-to-end tests for TransposeConvolution2d on CpuRef * Added one end-to-end test for all supported data types and data layout * Implemented RefLayerSupport::IsTransposeConvolution2dSupported() * Fixed formula used in TransposeConvolution2dLayer::InferOutputShapes() Signed-off-by: Aron Virginas-Tar Change-Id: If1ba3c226ecfa17f7fceffae857f39297c6433f2 --- src/armnn/InternalTypes.cpp | 1 + 1 file changed, 1 insertion(+) (limited to 'src/armnn/InternalTypes.cpp') diff --git a/src/armnn/InternalTypes.cpp b/src/armnn/InternalTypes.cpp index 47a6f60534..f62ce92921 100644 --- a/src/armnn/InternalTypes.cpp +++ b/src/armnn/InternalTypes.cpp @@ -58,6 +58,7 @@ char const* GetLayerTypeAsCString(LayerType type) case LayerType::StridedSlice: return "StridedSlice"; case LayerType::Subtraction: return "Subtraction"; case LayerType::Switch: return "Switch"; + case LayerType::TransposeConvolution2d: return "TransposeConvolution2d"; default: BOOST_ASSERT_MSG(false, "Unknown layer type"); return "Unknown"; -- cgit v1.2.1