m53297601
/
Paddle
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Optimize the error message of OP. (#27478)

Browse Source 
* iCafe 9009: Optimize the error message of OP.

* Optimize the error message of GatherTreeOP.
my_2.0rc
xiemoyuan 5 years ago committed by GitHub
parent aed0080181
commit 99e3337368
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 7 additions and 9 deletions
Show Stats Download Patch File Download Diff File

16
paddle/fluid/operators/gather_tree_op.cc
Unescape View File

@ -22,18 +22,16 @@ class GatherTreeOp : public framework::OperatorWithKernel {
using framework::OperatorWithKernel::OperatorWithKernel; using framework::OperatorWithKernel::OperatorWithKernel;
void InferShape(framework::InferShapeContext* ctx) const override { void InferShape(framework::InferShapeContext* ctx) const override {
PADDLE_ENFORCE(ctx->HasInput("Ids"), OP_INOUT_CHECK(ctx->HasInput("Ids"), "Input", "Ids", "GatherTree");
"Input(Ids) of GatherTreeOp should not be null."); OP_INOUT_CHECK(ctx->HasInput("Parents"), "Input", "Parents", "GatherTree");
PADDLE_ENFORCE(ctx->HasInput("Parents"), OP_INOUT_CHECK(ctx->HasOutput("Out"), "Output", "Out", "GatherTree");
"Input(Parents) of GatherTreeOp should not be null.");
PADDLE_ENFORCE(ctx->HasOutput("Out"),
"Output(Out) of GatherTreeOp should not be null.");
auto ids_dims = ctx->GetInputDim("Ids"); auto ids_dims = ctx->GetInputDim("Ids");
auto parents_dims = ctx->GetInputDim("Parents"); auto parents_dims = ctx->GetInputDim("Parents");
PADDLE_ENFORCE(ids_dims == parents_dims, PADDLE_ENFORCE_EQ(ids_dims == parents_dims, true,
"The shape of Input(Parents) must be same with the shape of " platform::errors::InvalidArgument(
"Input(Ids)."); "The shape of Input(Parents) must be same with the "
"shape of Input(Ids)."));
ctx->SetOutputDim("Out", ids_dims); ctx->SetOutputDim("Out", ids_dims);
} }

Write Preview
Loading…
Cancel
Save