commit 98180eff3c1feed19576d82ceac06d476235d973
author Aron Virginas-Tar <Aron.Virginas-Tar@arm.com> Wed Jun 26 15:02:47 2019 +0100
committer Áron Virginás-Tar <aron.virginas-tar@arm.com> Thu Jun 27 14:34:50 2019 +0000
tree 3fb83f43e02a10de12c7b2d2af2984bf4785b25d
parent 735a450d3b53a2d745b9a7a6d85747e25ec37ede

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 <Aron.Virginas-Tar@arm.com>
Change-Id: If1ba3c226ecfa17f7fceffae857f39297c6433f2

src/backends/reference/RefLayerSupport.hpp

diff --git a/src/backends/reference/RefLayerSupport.hpp b/src/backends/reference/RefLayerSupport.hpp
index 5d24149..9c397fe 100644
--- a/src/backends/reference/RefLayerSupport.hpp
+++ b/src/backends/reference/RefLayerSupport.hpp
@@ -266,6 +266,14 @@
                           const TensorInfo& alpha,
                           const TensorInfo& output,
                           Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
+
+    bool IsTransposeConvolution2dSupported(
+        const TensorInfo& input,
+        const TensorInfo& output,
+        const TransposeConvolution2dDescriptor& descriptor,
+        const TensorInfo& weights,
+        const Optional<TensorInfo>& biases,
+        Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
 };
 
 } // namespace armnn