提交 679d8fc6 编写于 作者: C chengduozh

rename op name

test=develop
上级 1013d6d0
...@@ -205,14 +205,14 @@ class CudnnLSTMGradOp : public framework::OperatorWithKernel { ...@@ -205,14 +205,14 @@ class CudnnLSTMGradOp : public framework::OperatorWithKernel {
} // namespace paddle } // namespace paddle
namespace ops = paddle::operators; namespace ops = paddle::operators;
REGISTER_OPERATOR(cudnn_lstm, ops::CudnnLSTMOp, ops::CudnnLSTMOpMaker, REGISTER_OPERATOR(lstm_cudnn, ops::CudnnLSTMOp, ops::CudnnLSTMOpMaker,
paddle::framework::DefaultGradOpDescMaker<true>); paddle::framework::DefaultGradOpDescMaker<true>);
REGISTER_OPERATOR(cudnn_lstm_grad, ops::CudnnLSTMGradOp); REGISTER_OPERATOR(lstm_cudnn_grad, ops::CudnnLSTMGradOp);
REGISTER_OP_CPU_KERNEL( REGISTER_OP_CPU_KERNEL(
cudnn_lstm, lstm_cudnn,
ops::CudnnLSTMKernel<paddle::platform::CPUDeviceContext, float>); ops::CudnnLSTMKernel<paddle::platform::CPUDeviceContext, float>);
REGISTER_OP_CPU_KERNEL( REGISTER_OP_CPU_KERNEL(
cudnn_lstm_grad, lstm_cudnn_grad,
ops::CudnnLSTMGradKernel<paddle::platform::CPUDeviceContext, float>); ops::CudnnLSTMGradKernel<paddle::platform::CPUDeviceContext, float>);
...@@ -487,8 +487,8 @@ class CudnnLSTMGPUGradKernel : public framework::OpKernel<T> { ...@@ -487,8 +487,8 @@ class CudnnLSTMGPUGradKernel : public framework::OpKernel<T> {
namespace ops = paddle::operators; namespace ops = paddle::operators;
REGISTER_OP_CUDA_KERNEL( REGISTER_OP_CUDA_KERNEL(
cudnn_lstm, lstm_cudnn,
ops::CudnnLSTMGPUKernel<paddle::platform::CUDADeviceContext, float>); ops::CudnnLSTMGPUKernel<paddle::platform::CUDADeviceContext, float>);
REGISTER_OP_CUDA_KERNEL( REGISTER_OP_CUDA_KERNEL(
cudnn_lstm_grad, lstm_cudnn_grad,
ops::CudnnLSTMGPUGradKernel<paddle::platform::CUDADeviceContext, float>); ops::CudnnLSTMGPUGradKernel<paddle::platform::CUDADeviceContext, float>);
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册