diff options
Diffstat (limited to 'src/backends/reference/workloads/TransposeConvolution2d.cpp')
-rw-r--r-- | src/backends/reference/workloads/TransposeConvolution2d.cpp | 8 |
1 files changed, 2 insertions, 6 deletions
diff --git a/src/backends/reference/workloads/TransposeConvolution2d.cpp b/src/backends/reference/workloads/TransposeConvolution2d.cpp index c34a309806..7408e92982 100644 --- a/src/backends/reference/workloads/TransposeConvolution2d.cpp +++ b/src/backends/reference/workloads/TransposeConvolution2d.cpp @@ -52,12 +52,8 @@ void TransposeConvolution2dImpl(const TransposeConvolution2dDescriptor& descript std::vector<float> outputBuffer(outputShape.GetNumElements(), 0); - const std::vector<float> inputVec = inputDecoder.DecodeTensor(inputShape.GetNumElements()); - - const unsigned channelStep = weightsWidth * weightsHeight * weightsDepth; - - const std::vector<float> filterVec = - weightsDecoder.DecodeTensor(weightsShape.GetNumElements(), channelStep); + const std::vector<float> inputVec = inputDecoder.DecodeTensor(inputShape); + const std::vector<float> filterVec = weightsDecoder.DecodeTensor(weightsShape); for (unsigned int batch = 0u; batch < numBatches; ++batch) { |