提交 4acd5aba 编写于 作者: Q qiaolongfei

update comment for input/output length check

上级 8c7ee7a5
...@@ -325,15 +325,21 @@ class CompileTimeInferShapeContext : public InferShapeContextBase { ...@@ -325,15 +325,21 @@ class CompileTimeInferShapeContext : public InferShapeContextBase {
bool HasInput(const std::string& name) const override { bool HasInput(const std::string& name) const override {
const std::vector<std::string>& input_names = op_.Input(name); const std::vector<std::string>& input_names = op_.Input(name);
PADDLE_ENFORCE_EQ(input_names.size(), 1UL, "Inputs(%s) length is not 1", auto length = input_names.size();
name); PADDLE_ENFORCE_EQ(length, 1UL,
"Input(%s) should have only one value, "
"but it have %d now",
name, length);
return block_.HasVar(input_names[0]); return block_.HasVar(input_names[0]);
} }
bool HasOutput(const std::string& name) const override { bool HasOutput(const std::string& name) const override {
const std::vector<std::string>& output_names = op_.Output(name); const std::vector<std::string>& output_names = op_.Output(name);
PADDLE_ENFORCE_EQ(output_names.size(), 1UL, "Outputs(%s) length is not 1", auto length = output_names.size();
name); PADDLE_ENFORCE_EQ(length, 1UL,
"Output(%s) should have only one value, "
"but it have %d now",
name, length);
return block_.HasVar(output_names[0]); return block_.HasVar(output_names[0]);
} }
...@@ -357,7 +363,11 @@ class CompileTimeInferShapeContext : public InferShapeContextBase { ...@@ -357,7 +363,11 @@ class CompileTimeInferShapeContext : public InferShapeContextBase {
DDim GetInputDim(const std::string& name) const override { DDim GetInputDim(const std::string& name) const override {
std::vector<DDim> ddims = GetInputsDim(name); std::vector<DDim> ddims = GetInputsDim(name);
PADDLE_ENFORCE_EQ(ddims.size(), 1UL, "Inputs(%s) length is not 1", name); auto length = ddims.size();
PADDLE_ENFORCE_EQ(length, 1UL,
"Input(%s) should have 1 value, "
"but it has %d now",
name, length);
return ddims[0]; return ddims[0];
} }
...@@ -367,7 +377,11 @@ class CompileTimeInferShapeContext : public InferShapeContextBase { ...@@ -367,7 +377,11 @@ class CompileTimeInferShapeContext : public InferShapeContextBase {
DDim GetOutputDim(const std::string& name) const override { DDim GetOutputDim(const std::string& name) const override {
std::vector<DDim> ddims = GetOutputsDim(name); std::vector<DDim> ddims = GetOutputsDim(name);
PADDLE_ENFORCE_EQ(ddims.size(), 1UL, "Outputs(%s) length is not 1", name); auto length = ddims.size();
PADDLE_ENFORCE_EQ(length, 1UL,
"Output(%s) should have 1 value, "
"but it has %d now",
name, length);
return ddims[0]; return ddims[0];
} }
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册