From 028973815bc58221d8a2582da1ecf05fb622de80 Mon Sep 17 00:00:00 2001 From: wanghaoshuang Date: Wed, 27 Sep 2017 14:59:09 +0800 Subject: [PATCH] Fix InferShape of clip op. --- paddle/operators/clip_op.cc | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/paddle/operators/clip_op.cc b/paddle/operators/clip_op.cc index 316d28f174..c208724c43 100644 --- a/paddle/operators/clip_op.cc +++ b/paddle/operators/clip_op.cc @@ -17,6 +17,8 @@ namespace paddle { namespace operators { +using framework::LoDTensor; + class ClipOp : public framework::OperatorWithKernel { public: using framework::OperatorWithKernel::OperatorWithKernel; @@ -28,11 +30,10 @@ class ClipOp : public framework::OperatorWithKernel { PADDLE_ENFORCE(ctx->HasOutput("Out"), "Output(Out) of ClipOp should not be null."); auto x_dims = ctx->GetInputDim("X"); - auto max = ctx->Attrs().Get("max"); - auto min = ctx->Attrs().Get("min"); + auto max = Attr("max"); + auto min = Attr("min"); PADDLE_ENFORCE_LT(min, max, "max should be greater than min."); ctx->SetOutputDim("Out", x_dims); - ctx->ShareLoD("X", /*->*/ "Out"); } }; @@ -43,7 +44,7 @@ class ClipOpMaker : public framework::OpProtoAndCheckerMaker { : OpProtoAndCheckerMaker(proto, op_checker) { AddInput("X", "(Tensor)The input of clip op." - "The input should be a k-D tensor(k > 0 and k < 7)"); + "The dimension size of input tensor must be between [1, 9]."); AddOutput("Out", "(Tensor)The output of clip op with shape as input(X)"); AddAttr( "min", "(float)Minimum value, under which element is replaced by min."); -- GitLab