File tree Expand file tree Collapse file tree 1 file changed +6
-3
lines changed Expand file tree Collapse file tree 1 file changed +6
-3
lines changed Original file line number Diff line number Diff line change @@ -26,10 +26,13 @@ class PReluOp : public framework::OperatorWithKernel {
26
26
std::string mode = ctx->Attrs ().Get <std::string>(" mode" );
27
27
28
28
auto x_dim = ctx->GetInputDim (" X" );
29
- PADDLE_ENFORCE (ctx->HasInput (" X" ), " Input(X) should not be null" );
30
- PADDLE_ENFORCE (ctx->HasInput (" Alpha" ), " Input(Alpha) should not be null" );
29
+ PADDLE_ENFORCE (ctx->HasInput (" X" ),
30
+ " Input(X) of PreluOp should not be null" );
31
+ PADDLE_ENFORCE (ctx->HasInput (" Alpha" ),
32
+ " Input(Alpha) of PreluOp should not be null" );
31
33
32
- PADDLE_ENFORCE (ctx->HasOutput (" Out" ), " Output(Out) should not be null" );
34
+ PADDLE_ENFORCE (ctx->HasOutput (" Out" ),
35
+ " Output(Out) of PreluOp should not be null" );
33
36
if (mode == " all" ) {
34
37
PADDLE_ENFORCE (product (ctx->GetInputDim (" Alpha" )) == 1 ,
35
38
" For mode 'all', size of weight Alpha must be one." );
You can’t perform that action at this time.
0 commit comments