Skip to content

P
Paddle

Crayon鑫 / Paddle
与 Fork 源项目一致

Fork自 PaddlePaddle / Paddle

通知 1
Star 1
Fork 0
P
Paddle
提交 3206970b 编写于 作者: T tangwei12
浏览文件
操作

sampling op rename

上级 e0ab2f71
隐藏空白更改
内联 并排
Showing with 4 addition and 3 deletion
paddle/fluid/operators/sampling_id_op.cc
浏览文件 @ 3206970b
...@@ -58,7 +58,8 @@ SamplingId Operator. ...@@ -58,7 +58,8 @@ SamplingId Operator.
namespace ops = paddle::operators; namespace ops = paddle::operators;
REGISTER_OP_CUDA_KERNEL( 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, double>,
ops::SamplingIdKernel<paddle::platform::CUDADeviceContext, int>, ops::SamplingIdKernel<paddle::platform::CUDADeviceContext, int>,
ops::SamplingIdKernel<paddle::platform::CUDADeviceContext, int64_t>); ops::SamplingIdKernel<paddle::platform::CUDADeviceContext, int64_t>);
paddle/fluid/operators/sampling_id_op.cu
浏览文件 @ 3206970b
...@@ -30,11 +30,11 @@ class SamplingIdOp : public framework::OperatorWithKernel { ...@@ -30,11 +30,11 @@ class SamplingIdOp : public framework::OperatorWithKernel {
} // namespace paddle } // namespace paddle
namespace ops = paddle::operators; namespace ops = paddle::operators;
REGISTER_OPERATOR(samplingid, ops::SamplingIdOp, ops::SamplingIdOpMaker, REGISTER_OPERATOR(sampling_id, ops::SamplingIdOp, ops::SamplingIdOpMaker,
paddle::framework::EmptyGradOpMaker); paddle::framework::EmptyGradOpMaker);
REGISTER_OP_CPU_KERNEL( 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, int64_t>,
ops::SamplingIdKernel<paddle::platform::CPUDeviceContext, float>, ops::SamplingIdKernel<paddle::platform::CPUDeviceContext, float>,
ops::SamplingIdKernel<paddle::platform::CPUDeviceContext, double>); ops::SamplingIdKernel<paddle::platform::CPUDeviceContext, double>);
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或