diff options
Diffstat (limited to 'src/backends/reference/workloads/TransposeConvolution2d.cpp')
| -rw-r--r-- | src/backends/reference/workloads/TransposeConvolution2d.cpp | 2 |
1 files changed, 1 insertions, 1 deletions
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 = |