Skip to content

P
Paddle

urhero / Paddle
与 Fork 源项目一致

Fork自 PaddlePaddle / Paddle

通知 1
Star 0
Fork 0
P
Paddle

体验新版 GitCode，发现更多精彩内容 >>

未验证 提交 a5bdf485 编写于 作者: W Wilber 提交者： GitHub
浏览文件
操作

fill op error message enhancement. test=develop (#23560) 

fill op error message enhancement
上级 d37cd740
显示空白变更内容
内联 并排
Showing with 1 addition and 2 deletion
paddle/fluid/operators/fill_op.cc
浏览文件 @ a5bdf485
...@@ -44,8 +44,7 @@ class FillOp : public framework::OperatorWithKernel { ...@@ -44,8 +44,7 @@ class FillOp : public framework::OperatorWithKernel {
using framework::OperatorWithKernel::OperatorWithKernel; using framework::OperatorWithKernel::OperatorWithKernel;
void InferShape(framework::InferShapeContext* context) const override { void InferShape(framework::InferShapeContext* context) const override {
PADDLE_ENFORCE_EQ(context->HasOutput("Out"), true, OP_INOUT_CHECK(context->HasOutput("Out"), "Output", "Out", "Fill");
"Output(Out) of FillOp should not be null.");
auto& shape = context->Attrs().Get<std::vector<int>>("shape"); auto& shape = context->Attrs().Get<std::vector<int>>("shape");
context->SetOutputDim("Out", framework::make_ddim(shape)); context->SetOutputDim("Out", framework::make_ddim(shape));
} }
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或