diff --git a/paddle/framework/operator.cc b/paddle/framework/operator.cc index a1f1be5f34..fe8096835d 100644 --- a/paddle/framework/operator.cc +++ b/paddle/framework/operator.cc @@ -74,7 +74,9 @@ void UseALL() { static DDim GetDims(const Scope& scope, const std::string& name) { Variable* var = scope.FindVar(name); - if (var->IsType()) { + if (var == nullptr) { + return DDim({-1}); + } else if (var->IsType()) { return var->Get().dims(); } else if (var->IsType()) { return var->Get().GetCompleteDims();