clip_op.cc 3.1 KB
Newer Older
W
wanghaoshuang 已提交
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23
/* Copyright (c) 2016 PaddlePaddle Authors. All Rights Reserve.

   Licensed under the Apache License, Version 2.0 (the "License");
   you may not use this file except in compliance with the License.
   You may obtain a copy of the License at

   http://www.apache.org/licenses/LICENSE-2.0

   Unless required by applicable law or agreed to in writing, software
   distributed under the License is distributed on an "AS IS" BASIS,
   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
   See the License for the specific language governing permissions and
   limitations under the License. */

#include "paddle/operators/clip_op.h"

namespace paddle {
namespace operators {

class ClipOp : public framework::OperatorWithKernel {
 public:
  using framework::OperatorWithKernel::OperatorWithKernel;

24
  void InferShape(framework::InferShapeContext* ctx) const override {
Q
Qiao Longfei 已提交
25 26 27 28 29
    PADDLE_ENFORCE(ctx->HasInput("X"),
                   "Input(X) of ClipOp should not be null.");
    PADDLE_ENFORCE(ctx->HasOutput("Out"),
                   "Output(Out) of ClipOp should not be null.");
    auto x_dims = ctx->GetInputDim("X");
30 31
    auto max = ctx->Attrs().Get<float>("max");
    auto min = ctx->Attrs().Get<float>("min");
W
wanghaoshuang 已提交
32
    PADDLE_ENFORCE_LT(min, max, "max should be greater than min.");
Q
Qiao Longfei 已提交
33
    ctx->SetOutputDim("Out", x_dims);
W
wanghaoshuang 已提交
34
    ctx->ShareLoD("X", /*->*/ "Out");
W
wanghaoshuang 已提交
35 36 37
  }
};

38
template <typename AttrType>
W
wanghaoshuang 已提交
39 40
class ClipOpMaker : public framework::OpProtoAndCheckerMaker {
 public:
Q
Qiao Longfei 已提交
41
  ClipOpMaker(framework::OpProto* proto, framework::OpAttrChecker* op_checker)
W
wanghaoshuang 已提交
42
      : OpProtoAndCheckerMaker(proto, op_checker) {
43 44
    AddInput("X",
             "(Tensor)The input of clip op."
W
wanghaoshuang 已提交
45
             "The number of dimensions must be between [1, 9].");
46 47 48 49 50
    AddOutput("Out", "(Tensor)The output of clip op with shape as input(X)");
    AddAttr<AttrType>(
        "min", "(float)Minimum value, under which element is replaced by min.");
    AddAttr<AttrType>(
        "max", "(float)Maximum value, above which element is replaced by max");
W
wanghaoshuang 已提交
51
    AddComment(R"DOC(
52 53 54
Clip Operator.

The clip operator limits the value of given input within an interval. The interval is
K
kexinzhao 已提交
55 56 57 58 59
specified with arguments 'min' and 'max':

$$
Out = \min(\max(X, min), max)
$$
60

W
wanghaoshuang 已提交
61 62 63 64 65 66 67 68
)DOC");
  }
};

class ClipOpGrad : public framework::OperatorWithKernel {
 public:
  using framework::OperatorWithKernel::OperatorWithKernel;

69
  void InferShape(framework::InferShapeContext* ctx) const override {
Q
Qiao Longfei 已提交
70 71 72 73 74 75
    PADDLE_ENFORCE(ctx->HasInput("X"), "Input(X) should not be null");
    PADDLE_ENFORCE(ctx->HasInput(framework::GradVarName("Out")),
                   "Input(Out@GRAD) should not be null");
    auto x_dims = ctx->GetInputDim("X");
    if (ctx->HasOutput(framework::GradVarName("X"))) {
      ctx->SetOutputDim(framework::GradVarName("X"), x_dims);
W
wanghaoshuang 已提交
76
    }
W
wanghaoshuang 已提交
77 78 79 80 81 82 83
  }
};

}  // namespace operators
}  // namespace paddle

namespace ops = paddle::operators;
84 85
REGISTER_OP(clip, ops::ClipOp, ops::ClipOpMaker<float>, clip_grad,
            ops::ClipOpGrad);
Q
QI JUN 已提交
86 87 88 89
REGISTER_OP_CPU_KERNEL(
    clip, ops::ClipKernel<paddle::platform::CPUDeviceContext, float>);
REGISTER_OP_CPU_KERNEL(
    clip_grad, ops::ClipGradKernel<paddle::platform::CPUDeviceContext, float>);