rowwise_add_op.cu 151 字节
Newer Older
1
#include "paddle/operators/rowwise_add_op.h"
2

3 4
REGISTER_OP_GPU_KERNEL(rowwise_add,
                       ops::RowWiseAddKernel<ops::GPUPlace, float>);