From aec942c3f258db46e0fb8054d39c0e7c3c411728 Mon Sep 17 00:00:00 2001 From: Aron Virginas-Tar Date: Wed, 14 Aug 2019 14:37:42 +0100 Subject: IVGCVSW-3640 Add multi-channel TransposeConvolution2d unit tests to CL backend * Fixed bug in multi-channel test and reference workload implementation * Enabled multi-channel tests on CL backend Signed-off-by: Aron Virginas-Tar Change-Id: I08bb523bc68d9c93a1012b4f487a5bce36a085b1 --- src/backends/reference/workloads/TransposeConvolution2d.cpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'src/backends/reference/workloads/TransposeConvolution2d.cpp') diff --git a/src/backends/reference/workloads/TransposeConvolution2d.cpp b/src/backends/reference/workloads/TransposeConvolution2d.cpp index acbfe0cc90..52cc18c17a 100644 --- a/src/backends/reference/workloads/TransposeConvolution2d.cpp +++ b/src/backends/reference/workloads/TransposeConvolution2d.cpp @@ -83,7 +83,7 @@ void TransposeConvolution2dImpl(const TransposeConvolution2dDescriptor& descript inputDecoder[inputIndex]; const unsigned int weightsIndex = - dataLayoutIndexed.GetIndex(weightsShape, batch, dOutput, yWeights, xWeights); + dataLayoutIndexed.GetIndex(weightsShape, dOutput, dInput, yWeights, xWeights); weightsDecoder[weightsIndex]; const unsigned int outputIndex = -- cgit v1.2.1