diff --git a/paddle/function/GemmConvOp.cpp b/paddle/function/GemmConvOp.cpp index bb7bc6477927ef4984b5d52d8f4ebc46ebc1fab1..9dab831d32cef4c72a885bc806295fd5eb220c06 100644 --- a/paddle/function/GemmConvOp.cpp +++ b/paddle/function/GemmConvOp.cpp @@ -239,7 +239,7 @@ public: Col2ImFunctor col2im; GemmFunctor gemm; size_t inputOffset = (inputChannels / groups_) * inputHeight * inputWidth; - size_t outputOffset = + size_t outputOffset = (outputChannels / groups_) * outputHeight * outputWidth; size_t filterOffset = filter.getElements() / groups_;