File tree Expand file tree Collapse file tree 1 file changed +4
-4
lines changed Expand file tree Collapse file tree 1 file changed +4
-4
lines changed Original file line number Diff line number Diff line change @@ -71,8 +71,7 @@ class MaxOutOp : public framework::OperatorWithKernel {
71
71
using framework::OperatorWithKernel::OperatorWithKernel;
72
72
void InferShape (framework::InferShapeContext* ctx) const override {
73
73
PADDLE_ENFORCE (ctx->HasInput (" X" ),
74
- " Input(X) of MaxoutOp"
75
- " should not be null." );
74
+ " Input(X) of MaxoutOpshould not be null." );
76
75
PADDLE_ENFORCE (ctx->HasOutput (" Out" ),
77
76
" Output(Out) of MaxoutOp should not be null." );
78
77
auto in_x_dims = ctx->GetInputDim (" X" );
@@ -90,9 +89,10 @@ class MaxOutOpGrad : public framework::OperatorWithKernel {
90
89
public:
91
90
using framework::OperatorWithKernel::OperatorWithKernel;
92
91
void InferShape (framework::InferShapeContext* ctx) const override {
93
- PADDLE_ENFORCE (ctx->HasInput (" X" ), " Input(X) must not be null." );
92
+ PADDLE_ENFORCE (ctx->HasInput (" X" ),
93
+ " Input(X) of MaxOutOpGrad must not be null." );
94
94
PADDLE_ENFORCE (ctx->HasOutput (framework::GradVarName (" X" )),
95
- " Input(X@GRAD) should not be null." );
95
+ " Output(Grad@X) of MaxOutOpGrad should not be null." );
96
96
ctx->SetOutputDim (framework::GradVarName (" X" ), ctx->GetInputDim (" X" ));
97
97
}
98
98
};
You can’t perform that action at this time.
0 commit comments