rowwise_add_op.cu 188 字节
Newer Older
1 2 3 4 5
#include <paddle/framework/op_registry.h>
#include <paddle/operators/rowwise_add_op.h>

REGISTER_OP_GPU_KERNEL(
    mul, paddle::operators::RowWiseAddKernel<paddle::platform ::GPUPlace>);