提交 9a819265 编写于 作者: X Xin Pan

fix

test=develop
上级 6de08b5e
...@@ -81,10 +81,10 @@ class CompileTimeInferShapeContext : public InferShapeContext { ...@@ -81,10 +81,10 @@ class CompileTimeInferShapeContext : public InferShapeContext {
"The %s[%d] is @EMPTY@", out, j); "The %s[%d] is @EMPTY@", out, j);
auto *in_var = block_.FindVarRecursive(Inputs(in)[i]); auto *in_var = block_.FindVarRecursive(Inputs(in)[i]);
auto *out_var = block_.FindVarRecursive(Outputs(out)[j]); auto *out_var = block_.FindVarRecursive(Outputs(out)[j]);
PADDLE_ENFORCE_EQ(in_var->GetType(), proto::VarType::LOD_TENSOR, if (in_var->GetType() != proto::VarType::LOD_TENSOR) {
"The %d-th output of Output(%s) must be LoDTensor.", j, VLOG(3) << "input " << in << " is not LodTensor";
out); return;
}
out_var->SetLoDLevel(in_var->GetLoDLevel()); out_var->SetLoDLevel(in_var->GetLoDLevel());
} }
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册