sampling op rename

revert-12469-sum_op_dim_fix
tangwei12 7 years ago
parent e0ab2f7158
commit 3206970b77

@ -58,7 +58,8 @@ SamplingId Operator.
namespace ops = paddle::operators;
REGISTER_OP_CUDA_KERNEL(
slice, ops::SamplingIdKernel<paddle::platform::CUDADeviceContext, float>,
sampling_id,
ops::SamplingIdKernel<paddle::platform::CUDADeviceContext, float>,
ops::SamplingIdKernel<paddle::platform::CUDADeviceContext, double>,
ops::SamplingIdKernel<paddle::platform::CUDADeviceContext, int>,
ops::SamplingIdKernel<paddle::platform::CUDADeviceContext, int64_t>);

@ -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<paddle::platform::CPUDeviceContext, int>,
sampling_id, ops::SamplingIdKernel<paddle::platform::CPUDeviceContext, int>,
ops::SamplingIdKernel<paddle::platform::CPUDeviceContext, int64_t>,
ops::SamplingIdKernel<paddle::platform::CPUDeviceContext, float>,
ops::SamplingIdKernel<paddle::platform::CPUDeviceContext, double>);

Loading…
Cancel
Save