diff --git a/paddle/operators/clip_op.cc b/paddle/operators/clip_op.cc index c208724c436002244a784ea3915a0c8a32763b78..6b9f53206404614018e960cf2ec7b0bd4e4704ae 100644 --- a/paddle/operators/clip_op.cc +++ b/paddle/operators/clip_op.cc @@ -34,6 +34,7 @@ class ClipOp : public framework::OperatorWithKernel { auto min = Attr("min"); PADDLE_ENFORCE_LT(min, max, "max should be greater than min."); ctx->SetOutputDim("Out", x_dims); + ctx->ShareLoD("X", /*->*/ "Out"); } };