add weight's dim assert

990741aa · nhzlx · fc41eb40 · 990741aa
隐藏空白更改
内联并排

Showing with 1 addition and 0 deletion

paddle/fluid/inference/tensorrt/convert/conv2d_op.cc paddle/fluid/inference/tensorrt/convert/conv2d_op.cc +1 -0

未找到文件。
--- a/paddle/fluid/inference/tensorrt/convert/conv2d_op.cc
+++ b/paddle/fluid/inference/tensorrt/convert/conv2d_op.cc
@@ -37,6 +37,7 @@ class Conv2dOpConverter : public OpConverter {
    auto* Y_t = Y_v->GetMutable<framework::LoDTensor>();
    auto* weight_data = Y_t->mutable_data<float>(platform::CPUPlace());

+    PADDLE_ENFORCE(Y_t->dims().size(), 4UL);
    const int n_output = Y_t->dims()[0];
    const int filter_h = Y_t->dims()[2];
    const int filter_w = Y_t->dims()[3];