We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent fc41eb4 commit 990741aCopy full SHA for 990741a
paddle/fluid/inference/tensorrt/convert/conv2d_op.cc
@@ -37,6 +37,7 @@ class Conv2dOpConverter : public OpConverter {
37
auto* Y_t = Y_v->GetMutable<framework::LoDTensor>();
38
auto* weight_data = Y_t->mutable_data<float>(platform::CPUPlace());
39
40
+ PADDLE_ENFORCE(Y_t->dims().size(), 4UL);
41
const int n_output = Y_t->dims()[0];
42
const int filter_h = Y_t->dims()[2];
43
const int filter_w = Y_t->dims()[3];
0 commit comments