diff --git a/src/relay/op/nn/convolution.h b/src/relay/op/nn/convolution.h index 0c089a41c5f20..36eaed672550d 100644 --- a/src/relay/op/nn/convolution.h +++ b/src/relay/op/nn/convolution.h @@ -1103,7 +1103,7 @@ bool Conv2DTransposeRel(const Array& types, int num_inputs, const Attrs& a << " kernel_size=" << param->kernel_size << " wshape=" << Array(wshape); } if (param->channels.defined()) { - ICHECK(reporter->AssertEQ(param->channels, wshape[1])) + ICHECK(reporter->AssertEQ(param->channels, wshape[0])) << "Conv2DTransposed: shape of weight is inconsistent with channels, " << " channels=" << param->channels << " wshape=" << Array(wshape); }