提交 990741aa 编写于 作者: N nhzlx

add weight's dim assert

上级 fc41eb40
...@@ -37,6 +37,7 @@ class Conv2dOpConverter : public OpConverter { ...@@ -37,6 +37,7 @@ class Conv2dOpConverter : public OpConverter {
auto* Y_t = Y_v->GetMutable<framework::LoDTensor>(); auto* Y_t = Y_v->GetMutable<framework::LoDTensor>();
auto* weight_data = Y_t->mutable_data<float>(platform::CPUPlace()); 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 n_output = Y_t->dims()[0];
const int filter_h = Y_t->dims()[2]; const int filter_h = Y_t->dims()[2];
const int filter_w = Y_t->dims()[3]; const int filter_w = Y_t->dims()[3];
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册