Skip to content

Commit 3206970

Browse files
committed
sampling op rename
1 parent e0ab2f7 commit 3206970

File tree

2 files changed

+4
-3
lines changed

2 files changed

+4
-3
lines changed

paddle/fluid/operators/sampling_id_op.cc

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -58,7 +58,8 @@ SamplingId Operator.
5858

5959
namespace ops = paddle::operators;
6060
REGISTER_OP_CUDA_KERNEL(
61-
slice, ops::SamplingIdKernel<paddle::platform::CUDADeviceContext, float>,
61+
sampling_id,
62+
ops::SamplingIdKernel<paddle::platform::CUDADeviceContext, float>,
6263
ops::SamplingIdKernel<paddle::platform::CUDADeviceContext, double>,
6364
ops::SamplingIdKernel<paddle::platform::CUDADeviceContext, int>,
6465
ops::SamplingIdKernel<paddle::platform::CUDADeviceContext, int64_t>);

paddle/fluid/operators/sampling_id_op.cu

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -30,11 +30,11 @@ class SamplingIdOp : public framework::OperatorWithKernel {
3030
} // namespace paddle
3131

3232
namespace ops = paddle::operators;
33-
REGISTER_OPERATOR(samplingid, ops::SamplingIdOp, ops::SamplingIdOpMaker,
33+
REGISTER_OPERATOR(sampling_id, ops::SamplingIdOp, ops::SamplingIdOpMaker,
3434
paddle::framework::EmptyGradOpMaker);
3535

3636
REGISTER_OP_CPU_KERNEL(
37-
slice, ops::SamplingIdKernel<paddle::platform::CPUDeviceContext, int>,
37+
sampling_id, ops::SamplingIdKernel<paddle::platform::CPUDeviceContext, int>,
3838
ops::SamplingIdKernel<paddle::platform::CPUDeviceContext, int64_t>,
3939
ops::SamplingIdKernel<paddle::platform::CPUDeviceContext, float>,
4040
ops::SamplingIdKernel<paddle::platform::CPUDeviceContext, double>);

0 commit comments

Comments
 (0)