From 8ff3590eda2a6488f4b06f5ce6ffe553ae42d0a6 Mon Sep 17 00:00:00 2001 From: dongzhihong Date: Fri, 4 Aug 2017 01:15:56 +0800 Subject: [PATCH] fix op name --- paddle/operators/rowwise_add_op.cc | 20 ++++++++++---------- paddle/operators/rowwise_add_op.cu | 2 +- paddle/operators/rowwise_add_op.h | 6 +++--- 3 files changed, 14 insertions(+), 14 deletions(-) diff --git a/paddle/operators/rowwise_add_op.cc b/paddle/operators/rowwise_add_op.cc index cc763a8cf45..178ea3c6145 100644 --- a/paddle/operators/rowwise_add_op.cc +++ b/paddle/operators/rowwise_add_op.cc @@ -16,7 +16,7 @@ namespace paddle { namespace operators { -class RowWiseAddOp : public OperatorWithKernel { +class RowwiseAddOp : public OperatorWithKernel { protected: void InferShape(const InferShapeContext &ctx) const override { PADDLE_ENFORCE(ctx.InputSize() == 2UL, @@ -32,9 +32,9 @@ protected: } }; -class RowWiseAddOpMaker : public OpProtoAndCheckerMaker { +class RowwiseAddOpMaker : public OpProtoAndCheckerMaker { public: - RowWiseAddOpMaker(OpProto *proto, OpAttrChecker *op_checker) + RowwiseAddOpMaker(OpProto *proto, OpAttrChecker *op_checker) : OpProtoAndCheckerMaker(proto, op_checker) { AddInput("X", "The left input of row-wise add op, must be matrix"); AddInput("b", "The right input of row-wise add op, must be vector"); @@ -46,13 +46,13 @@ for i in xrange(X.shape[0]): )DOC"); } }; -class RowWiseAddGradOp : public OperatorWithKernel { +class RowwiseAddGradOp : public OperatorWithKernel { protected: void InferShape(const InferShapeContext &ctx) const override { PADDLE_ENFORCE(ctx.InputSize() == 4UL, - "RowWiseAddGrad inputs is I, O, OG, size must be 4"); + "RowwiseAddGrad inputs is I, O, OG, size must be 4"); PADDLE_ENFORCE(ctx.OutputSize() == 2, - "RowWiseAddGrad output is IG, size must be 2"); + "RowwiseAddGrad output is IG, size must be 2"); ctx.Output(0)->Resize(ctx.Input(0)->dims()); ctx.Output(1)->Resize(ctx.Input(1)->dims()); } @@ -61,10 +61,10 @@ protected: } // namespace operators } // namespace paddle -REGISTER_OP(rowwise_add, ops::RowWiseAddOp, ops::RowWiseAddOpMaker); +REGISTER_OP(rowwise_add, ops::RowwiseAddOp, ops::RowwiseAddOpMaker); REGISTER_OP_CPU_KERNEL(rowwise_add, - ops::RowWiseAddKernel); + ops::RowwiseAddKernel); -REGISTER_GRADIENT_OP(rowwise_add, rowwise_add_grad, ops::RowWiseAddGradOp); +REGISTER_GRADIENT_OP(rowwise_add, rowwise_add_grad, ops::RowwiseAddGradOp); REGISTER_OP_CPU_KERNEL(rowwise_add_grad, - ops::RowWiseAddGradKernel); + ops::RowwiseAddGradKernel); diff --git a/paddle/operators/rowwise_add_op.cu b/paddle/operators/rowwise_add_op.cu index 4b33e38ebab..f48dfeb6f2c 100644 --- a/paddle/operators/rowwise_add_op.cu +++ b/paddle/operators/rowwise_add_op.cu @@ -1,4 +1,4 @@ #include "paddle/operators/rowwise_add_op.h" REGISTER_OP_GPU_KERNEL(rowwise_add, - ops::RowWiseAddKernel); + ops::RowwiseAddKernel); diff --git a/paddle/operators/rowwise_add_op.h b/paddle/operators/rowwise_add_op.h index 940459e0f17..321f51e61d4 100644 --- a/paddle/operators/rowwise_add_op.h +++ b/paddle/operators/rowwise_add_op.h @@ -19,7 +19,7 @@ namespace paddle { namespace operators { template -class RowWiseAddKernel : public OpKernel { +class RowwiseAddKernel : public OpKernel { public: void Compute(const ExecutionContext& context) const override { auto out = context.Output(0); @@ -39,7 +39,7 @@ public: }; template -class RowWiseAddGradKernel : public OpKernel { +class RowwiseAddGradKernel : public OpKernel { public: void Compute(const ExecutionContext& context) const override { auto XGrad = context.Output(0); @@ -51,7 +51,7 @@ public: auto OutGrad = EigenMatrix::From(*context.Input(3)); EigenMatrix::From(*XGrad).device(*(context.GetEigenDevice())) = OutGrad; - // const int dimension = bGrad.dimension(0); + // https://eigen.tuxfamily.org/dox/unsupported/TensorBase_8h_source.html EigenVector::Flatten(*bGrad).device(*(context.GetEigenDevice())) = OutGrad.cumsum(1); // colwise add -- GitLab