diff --git a/paddle/fluid/operators/arg_max_op.cc b/paddle/fluid/operators/arg_max_op.cc index 5603607357..859cccd1b2 100644 --- a/paddle/fluid/operators/arg_max_op.cc +++ b/paddle/fluid/operators/arg_max_op.cc @@ -13,11 +13,6 @@ See the License for the specific language governing permissions and limitations under the License. */ #include "paddle/fluid/operators/arg_max_op.h" -/* -REGISTER_ARG_MINMAX_OP_WITHOUT_GRADIENT(arg_max, ArgMax); - -REGISTER_ARG_MINMAX_KERNEL(arg_max, ArgMax, CPU); -*/ REGISTER_OPERATOR(arg_max, paddle::operators::ArgMaxOp, paddle::operators::ArgMaxOpMaker, diff --git a/paddle/fluid/operators/arg_max_op.cu b/paddle/fluid/operators/arg_max_op.cu index 8f57c63beb..c9c102bdcc 100644 --- a/paddle/fluid/operators/arg_max_op.cu +++ b/paddle/fluid/operators/arg_max_op.cu @@ -14,8 +14,6 @@ limitations under the License. */ #include "paddle/fluid/operators/arg_max_op.h" -// REGISTER_ARG_MINMAX_KERNEL(arg_max, ArgMax, CUDA); - REGISTER_OP_CUDA_KERNEL( arg_max, paddle::operators::ArgMaxKernel