fill_zeros_like_op.cu 216 字节
Newer Older
F
fengjiayi 已提交
1 2 3 4 5 6
#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>);