Skip to content

P
Paddle

wmsofts / Paddle
与 Fork 源项目一致

Fork自 PaddlePaddle / Paddle

通知 1
Star 0
Fork 0
P
Paddle

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

切换分支/标签
查找文件 普通视图历史永久链接Permalink
add_op.cu 151 字节
Newer Older
Q
add_op kernel implementation  
qijun 已提交
1 
#include "paddle/framework/op_registry.h"
Y
Add type_alias to import framework into ops  
Yu Yang 已提交
2 
#include "paddle/operators/add_op.h"
Y
Add a sample op, `add_op`  
Yu Yang 已提交
3
Y
Add type_alias to import framework into ops  
Yu Yang 已提交
4 
REGISTER_OP_GPU_KERNEL(add_two, ops::AddKernel<ops::GPUPlace, float>);