diff --git a/paddle/framework/operator.cc b/paddle/framework/operator.cc index a1f1be5f34264c11e8125f78650d63d9996aea84..fe8096835d6f4e01e4e8a190722d4eb4fdb4095f 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();