Make the behavior of data nullptr check of pooling layer same as others. (#3322)

This commit is contained in:
Ligeng Zhu 2019-06-08 12:17:29 -04:00 коммит произвёл Tianqi Chen
Родитель 4cee98ba6c
Коммит bd244ebd8d
2 изменённых файлов: 3 добавлений и 2 удалений

2
3rdparty/dmlc-core поставляемый

@ -1 +1 @@
Subproject commit fbe142b267a8edd1f1188fa2140d88f7ae308661
Subproject commit 3943914eed66470bd010df581e29e4dca4f7df6f

Просмотреть файл

@ -70,7 +70,8 @@ bool Pool2DRel(const Array<Type>& types,
CHECK_EQ(types.size(), 2);
const auto* data = types[0].as<TensorTypeNode>();
CHECK(data != nullptr);
if (data == nullptr) return false;
const auto dshape = data->shape;
CHECK_GE(dshape.size(), 2U)
<< "Pool2D only support input >= 2-D: input must have height and width";