|
|
@ -32,8 +32,7 @@ class LoDResetOp : public framework::OperatorWithKernel {
|
|
|
|
PADDLE_ENFORCE_GT(level0.size(), 1,
|
|
|
|
PADDLE_ENFORCE_GT(level0.size(), 1,
|
|
|
|
"If Input(Y) not provided, the target lod should be "
|
|
|
|
"If Input(Y) not provided, the target lod should be "
|
|
|
|
"specified by attribute `target_lod`.");
|
|
|
|
"specified by attribute `target_lod`.");
|
|
|
|
}
|
|
|
|
} else {
|
|
|
|
else {
|
|
|
|
|
|
|
|
ctx->ShareLoD( "Y", "Out");
|
|
|
|
ctx->ShareLoD( "Y", "Out");
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|