fill_zeros_like_op.cu 238 字节
Newer Older
F
fix bug  
fengjiayi 已提交
1
#define EIGEN_USE_GPU
F
fengjiayi 已提交
2 3 4 5 6 7
#include "paddle/framework/op_registry.h"
#include "paddle/operators/fill_zeros_like_op.h"

REGISTER_OP_GPU_KERNEL(
    fill_zeros_like,
    paddle::operators::FillZerosLikeKernel<paddle::platform::GPUPlace, float>);