From 3206970b770fb3a45d7a7c85566cab6b16db28d7 Mon Sep 17 00:00:00 2001 From: tangwei12 Date: Mon, 6 Aug 2018 10:24:59 +0800 Subject: [PATCH] sampling op rename --- paddle/fluid/operators/sampling_id_op.cc | 3 ++- paddle/fluid/operators/sampling_id_op.cu | 4 ++-- 2 files changed, 4 insertions(+), 3 deletions(-) diff --git a/paddle/fluid/operators/sampling_id_op.cc b/paddle/fluid/operators/sampling_id_op.cc index 20e3d4321..b9e3b0372 100644 --- a/paddle/fluid/operators/sampling_id_op.cc +++ b/paddle/fluid/operators/sampling_id_op.cc @@ -58,7 +58,8 @@ SamplingId Operator. namespace ops = paddle::operators; REGISTER_OP_CUDA_KERNEL( - slice, ops::SamplingIdKernel, + sampling_id, + ops::SamplingIdKernel, ops::SamplingIdKernel, ops::SamplingIdKernel, ops::SamplingIdKernel); diff --git a/paddle/fluid/operators/sampling_id_op.cu b/paddle/fluid/operators/sampling_id_op.cu index 4fa10de2c..f82ba68ce 100644 --- a/paddle/fluid/operators/sampling_id_op.cu +++ b/paddle/fluid/operators/sampling_id_op.cu @@ -30,11 +30,11 @@ class SamplingIdOp : public framework::OperatorWithKernel { } // namespace paddle namespace ops = paddle::operators; -REGISTER_OPERATOR(samplingid, ops::SamplingIdOp, ops::SamplingIdOpMaker, +REGISTER_OPERATOR(sampling_id, ops::SamplingIdOp, ops::SamplingIdOpMaker, paddle::framework::EmptyGradOpMaker); REGISTER_OP_CPU_KERNEL( - slice, ops::SamplingIdKernel, + sampling_id, ops::SamplingIdKernel, ops::SamplingIdKernel, ops::SamplingIdKernel, ops::SamplingIdKernel); -- GitLab