diff --git a/paddle/fluid/operators/arg_max_op.cc b/paddle/fluid/operators/arg_max_op.cc index 5603607357ab860bbf11928a1faa4a3ed94f2580..859cccd1b2dfcc8b2278e11157468095fd6a396d 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 8f57c63beb432e01992ea5325844b8b2bd7f6387..c9c102bdccf32b02b055cdfd800957c6c3723183 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