Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
wmsofts
Paddle
提交
a2103008
P
Paddle
项目概览
wmsofts
/
Paddle
与 Fork 源项目一致
Fork自
PaddlePaddle / Paddle
通知
1
Star
0
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
P
Paddle
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
体验新版 GitCode,发现更多精彩内容 >>
未验证
提交
a2103008
编写于
6月 04, 2018
作者:
T
Tao Luo
提交者:
GitHub
6月 04, 2018
浏览文件
操作
浏览文件
下载
差异文件
Merge pull request #11147 from tensor-tang/mkldnn_name
rename Mkldnn to MKLDNN
上级
32ec75ac
6ac47a3d
变更
1
隐藏空白更改
内联
并排
Showing
1 changed file
with
12 addition
and
12 deletion
+12
-12
paddle/fluid/operators/activation_mkldnn_op.cc
paddle/fluid/operators/activation_mkldnn_op.cc
+12
-12
未找到文件。
paddle/fluid/operators/activation_mkldnn_op.cc
浏览文件 @
a2103008
...
...
@@ -222,35 +222,35 @@ struct MKLDNNActivationGradFunc : public BaseActivationFunctor<T> {
};
template
<
typename
T
>
using
ReluM
kldnn
Functor
=
using
ReluM
KLDNN
Functor
=
MKLDNNActivationFunc
<
T
,
mkldnn
::
algorithm
::
eltwise_relu
>
;
template
<
typename
T
>
using
TanhM
kldnn
Functor
=
using
TanhM
KLDNN
Functor
=
MKLDNNActivationFunc
<
T
,
mkldnn
::
algorithm
::
eltwise_tanh
>
;
template
<
typename
T
>
using
SqrtM
kldnn
Functor
=
using
SqrtM
KLDNN
Functor
=
MKLDNNActivationFunc
<
T
,
mkldnn
::
algorithm
::
eltwise_sqrt
>
;
template
<
typename
T
>
using
AbsM
kldnn
Functor
=
using
AbsM
KLDNN
Functor
=
MKLDNNActivationFunc
<
T
,
mkldnn
::
algorithm
::
eltwise_abs
>
;
template
<
typename
T
>
using
ReluM
kldnn
GradFunctor
=
using
ReluM
KLDNN
GradFunctor
=
MKLDNNActivationGradFunc
<
T
,
mkldnn
::
algorithm
::
eltwise_relu
>
;
template
<
typename
T
>
using
TanhM
kldnn
GradFunctor
=
using
TanhM
KLDNN
GradFunctor
=
MKLDNNActivationGradFunc
<
T
,
mkldnn
::
algorithm
::
eltwise_tanh
>
;
template
<
typename
T
>
using
SqrtM
kldnn
GradFunctor
=
using
SqrtM
KLDNN
GradFunctor
=
MKLDNNActivationGradFunc
<
T
,
mkldnn
::
algorithm
::
eltwise_sqrt
>
;
template
<
typename
T
>
using
AbsM
kldnn
GradFunctor
=
using
AbsM
KLDNN
GradFunctor
=
MKLDNNActivationGradFunc
<
T
,
mkldnn
::
algorithm
::
eltwise_abs
>
;
}
// namespace operators
}
// namespace paddle
...
...
@@ -265,9 +265,9 @@ namespace ops = paddle::operators;
ops::MKLDNNActivationGradKernel<ops::grad_functor<float>>);
#define FOR_EACH_MKLDNN_KERNEL_FUNCTOR(__macro) \
__macro(relu, ReluM
kldnnFunctor, ReluMkldnn
GradFunctor); \
__macro(tanh, TanhM
kldnnFunctor, TanhMkldnn
GradFunctor); \
__macro(sqrt, SqrtM
kldnnFunctor, SqrtMkldnn
GradFunctor); \
__macro(abs, AbsM
kldnnFunctor, AbsMkldnn
GradFunctor);
__macro(relu, ReluM
KLDNNFunctor, ReluMKLDNN
GradFunctor); \
__macro(tanh, TanhM
KLDNNFunctor, TanhMKLDNN
GradFunctor); \
__macro(sqrt, SqrtM
KLDNNFunctor, SqrtMKLDNN
GradFunctor); \
__macro(abs, AbsM
KLDNNFunctor, AbsMKLDNN
GradFunctor);
FOR_EACH_MKLDNN_KERNEL_FUNCTOR
(
REGISTER_ACTIVATION_MKLDNN_KERNEL
);
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录