|
|
@ -43,4 +43,4 @@ REGISTER_OP_GPU_KERNEL(
|
|
|
|
ops::ReduceKernel<paddle::platform::GPUPlace, float, ops::MinFunctor>);
|
|
|
|
ops::ReduceKernel<paddle::platform::GPUPlace, float, ops::MinFunctor>);
|
|
|
|
REGISTER_OP_GPU_KERNEL(reduce_min_grad,
|
|
|
|
REGISTER_OP_GPU_KERNEL(reduce_min_grad,
|
|
|
|
ops::ReduceGradKernel<paddle::platform::GPUPlace, float,
|
|
|
|
ops::ReduceGradKernel<paddle::platform::GPUPlace, float,
|
|
|
|
ops::MaxOrMinGradFunctor>);
|
|
|
|
ops::MaxOrMinGradFunctor>);
|
|
|
|