From 679d8fc6fe0349c7438cbe45f4d4a1deba52837e Mon Sep 17 00:00:00 2001 From: chengduozh Date: Fri, 30 Nov 2018 18:52:50 +0800 Subject: [PATCH] rename op name test=develop --- paddle/fluid/operators/lstm_cudnn_op.cc | 8 ++++---- paddle/fluid/operators/lstm_cudnn_op.cu.cc | 4 ++-- 2 files changed, 6 insertions(+), 6 deletions(-) diff --git a/paddle/fluid/operators/lstm_cudnn_op.cc b/paddle/fluid/operators/lstm_cudnn_op.cc index ca60fb4b0b..c9a4a31738 100644 --- a/paddle/fluid/operators/lstm_cudnn_op.cc +++ b/paddle/fluid/operators/lstm_cudnn_op.cc @@ -205,14 +205,14 @@ class CudnnLSTMGradOp : public framework::OperatorWithKernel { } // namespace paddle namespace ops = paddle::operators; -REGISTER_OPERATOR(cudnn_lstm, ops::CudnnLSTMOp, ops::CudnnLSTMOpMaker, +REGISTER_OPERATOR(lstm_cudnn, ops::CudnnLSTMOp, ops::CudnnLSTMOpMaker, paddle::framework::DefaultGradOpDescMaker); -REGISTER_OPERATOR(cudnn_lstm_grad, ops::CudnnLSTMGradOp); +REGISTER_OPERATOR(lstm_cudnn_grad, ops::CudnnLSTMGradOp); REGISTER_OP_CPU_KERNEL( - cudnn_lstm, + lstm_cudnn, ops::CudnnLSTMKernel); REGISTER_OP_CPU_KERNEL( - cudnn_lstm_grad, + lstm_cudnn_grad, ops::CudnnLSTMGradKernel); diff --git a/paddle/fluid/operators/lstm_cudnn_op.cu.cc b/paddle/fluid/operators/lstm_cudnn_op.cu.cc index 7a67bbe539..353ab17597 100644 --- a/paddle/fluid/operators/lstm_cudnn_op.cu.cc +++ b/paddle/fluid/operators/lstm_cudnn_op.cu.cc @@ -487,8 +487,8 @@ class CudnnLSTMGPUGradKernel : public framework::OpKernel { namespace ops = paddle::operators; REGISTER_OP_CUDA_KERNEL( - cudnn_lstm, + lstm_cudnn, ops::CudnnLSTMGPUKernel); REGISTER_OP_CUDA_KERNEL( - cudnn_lstm_grad, + lstm_cudnn_grad, ops::CudnnLSTMGPUGradKernel); -- GitLab