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

diff --git a/3rdparty/dmlc-core b/3rdparty/dmlc-core
index fbe142b..3943914 160000 (submodule)
--- a/3rdparty/dmlc-core
+++ b/3rdparty/dmlc-core
@@ -1 +1 @@
-Subproject commit fbe142b267a8edd1f1188fa2140d88f7ae308661
+Subproject commit 3943914eed66470bd010df581e29e4dca4f7df6f

diff --git a/src/relay/op/nn/pooling.cc b/src/relay/op/nn/pooling.cc
index 4dd763b..44c9f89 100644 (file)
--- a/src/relay/op/nn/pooling.cc
+++ b/src/relay/op/nn/pooling.cc
@@ -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";