提交 455436e5 编写于 作者: Q qiaolongfei

fix compile problem

上级 e0c8cd8a
...@@ -22,6 +22,7 @@ limitations under the License. */ ...@@ -22,6 +22,7 @@ limitations under the License. */
#include "op_info.h" #include "op_info.h"
#include "paddle/framework/attribute.h" #include "paddle/framework/attribute.h"
#include "paddle/framework/block_desc.h"
#include "paddle/framework/data_type.h" #include "paddle/framework/data_type.h"
#include "paddle/framework/framework.pb.h" #include "paddle/framework/framework.pb.h"
#include "paddle/framework/lod_tensor.h" #include "paddle/framework/lod_tensor.h"
...@@ -349,7 +350,7 @@ class CompileTimeInferShapeContext : public InferShapeContextBase { ...@@ -349,7 +350,7 @@ class CompileTimeInferShapeContext : public InferShapeContextBase {
const std::vector<std::string>& output_names = op_.Output(name); const std::vector<std::string>& output_names = op_.Output(name);
PADDLE_ENFORCE_GT(output_names.size(), 0UL, "Inputs(%s) length is 0", name); PADDLE_ENFORCE_GT(output_names.size(), 0UL, "Inputs(%s) length is 0", name);
for (auto& output : output_names) { for (auto& output : output_names) {
if (!block_.HasVar(name)) return false; if (!block_.HasVar(output)) return false;
} }
return true; return true;
} }
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册