diff --git a/paddle/fluid/operators/reshape_op_npu.cc b/paddle/fluid/operators/reshape_op_npu.cc index cd0f0bb2558974ea8d3024be88a452c927912b8b..79a4cd116f3b939cdbf11992a6386b196b0d77ff 100644 --- a/paddle/fluid/operators/reshape_op_npu.cc +++ b/paddle/fluid/operators/reshape_op_npu.cc @@ -67,10 +67,20 @@ namespace ops = paddle::operators; REGISTER_OP_NPU_KERNEL( reshape2, ops::Reshape2NPUKernel, + ops::Reshape2NPUKernel, + ops::Reshape2NPUKernel, + ops::Reshape2NPUKernel, + ops::Reshape2NPUKernel, + ops::Reshape2NPUKernel, ops::Reshape2NPUKernel); REGISTER_OP_NPU_KERNEL( reshape2_grad, ops::Reshape2GradNPUKernel, + ops::Reshape2GradNPUKernel, + ops::Reshape2GradNPUKernel, + ops::Reshape2GradNPUKernel, + ops::Reshape2GradNPUKernel, + ops::Reshape2GradNPUKernel, ops::Reshape2GradNPUKernel);