Skip to content

P
PaddleDetection

PaddlePaddle / PaddleDetection
大约 1 年 前同步成功

通知 692
Star 11112
Fork 2696
P
PaddleDetection

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

切换分支/标签
查找文件 普通视图历史永久链接Permalink
sigmoid_op.cu 117 字节
Newer Older
Q
replace Tensor::tensor to EigenTensor::From  
qijun 已提交
1 
#include "paddle/operators/sigmoid_op.h"
Y
Add skeletons of `mul`, `rowwise_add`, `sigmoid`, `softmax` ops  
Yu Yang 已提交
2
Y
Add type_alias to import framework into ops  
Yu Yang 已提交
3 
REGISTER_OP_GPU_KERNEL(sigmoid, ops::SigmoidKernel<ops::GPUPlace, float>);