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

fix compile problem

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