Skip to content

P
Paddle

机器未来 / Paddle
与 Fork 源项目一致

Fork自 PaddlePaddle / Paddle

通知 1
Star 1
Fork 0
P
Paddle

体验新版 GitCode，发现更多精彩内容 >>

切换分支/标签
查找文件 普通视图历史永久链接Permalink
softmax_op.cu 184 字节
Newer Older
Y
Add skeletons of `mul`, `rowwise_add`, `sigmoid`, `softmax` ops  
Yu Yang 已提交
1 
#include <paddle/framework/op_registry.h>
Y
Follow comments, correct implement `DDim::size()`  
Yu Yang 已提交
2 
#include <paddle/operators/softmax_op.h>
Y
Add skeletons of `mul`, `rowwise_add`, `sigmoid`, `softmax` ops  
Yu Yang 已提交
3 4 5 

REGISTER_OP_GPU_KERNEL(
    softmax, paddle::operators::SoftmaxKernel<paddle::platform::GPUPlace>);