diff --git a/src/operator/ConvImpl.cpp b/src/operator/ConvImpl.cpp index 0be31befe2019d70b628db878443f14b1d622f1c..9eb3bdabe5bbaa1aefa97604b4ec0159004daa60 100644 --- a/src/operator/ConvImpl.cpp +++ b/src/operator/ConvImpl.cpp @@ -67,7 +67,7 @@ AIDGE_ASSERT(op_.getInput(0), "missing input #0 in Conv Operator."); op_.dilationDims(), op_.kernelDims(), op_.getInput(0)->template dims<3>(), // input dimensions - dynamic_cast<const Conv_Op<2>&>(mOp).outChannels(), // outChannels + dynamic_cast<const Conv_Op<1>&>(mOp).outChannels(), // outChannels input0.getImpl()->rawPtr(), // input input1.getImpl()->rawPtr(), // weight op_.getInput(2) ? input2.getImpl()->rawPtr() : nullptr, // bias