Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
机器未来
Paddle
提交
2c6bd4ad
P
Paddle
项目概览
机器未来
/
Paddle
与 Fork 源项目一致
Fork自
PaddlePaddle / Paddle
通知
1
Star
1
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
1
列表
看板
标记
里程碑
合并请求
0
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
P
Paddle
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
1
Issue
1
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
未验证
提交
2c6bd4ad
编写于
10月 11, 2022
作者:
S
Sławomir Siwek
提交者:
GitHub
10月 11, 2022
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
hard_swish grad (#46857)
上级
2bcbf8b0
变更
2
隐藏空白更改
内联
并排
Showing
2 changed file
with
7 addition
and
15 deletion
+7
-15
paddle/phi/kernels/onednn/activation_grad_kernel.cc
paddle/phi/kernels/onednn/activation_grad_kernel.cc
+1
-1
python/paddle/fluid/tests/unittests/mkldnn/test_activation_mkldnn_op.py
...fluid/tests/unittests/mkldnn/test_activation_mkldnn_op.py
+6
-14
未找到文件。
paddle/phi/kernels/onednn/activation_grad_kernel.cc
浏览文件 @
2c6bd4ad
...
...
@@ -243,7 +243,7 @@ void HardSwishGradKernel(const Context& dev_ctx,
float
offset
,
DenseTensor
*
dx
)
{
HardSwishOneDNNGradFunctor
<
T
>
functor
;
functor
(
dev_ctx
,
x
,
dout
,
threshold
,
0
,
dx
);
functor
(
dev_ctx
,
x
,
dout
,
0
,
0
,
dx
);
}
template
<
typename
T
,
typename
Context
>
...
...
python/paddle/fluid/tests/unittests/mkldnn/test_activation_mkldnn_op.py
浏览文件 @
2c6bd4ad
...
...
@@ -128,19 +128,11 @@ class TestMKLDNNSwishDim2(TestSwish):
self
.
dtype
=
np
.
float32
@
skip_check_grad_ci
(
reason
=
"not implemented yet"
)
class
TestMKLDNNHardSwishDim2
(
TestHardSwish
):
def
setUp
(
self
):
super
(
TestMKLDNNHardSwishDim2
,
self
).
setUp
()
self
.
attrs
[
"use_mkldnn"
]
=
True
def
init_dtype
(
self
):
self
.
dtype
=
np
.
float32
def
test_check_grad
(
self
):
pass
self
.
attrs
=
{
"use_mkldnn"
:
True
}
class
TestMKLDNNSigmoidDim2
(
TestSigmoid
):
...
...
@@ -317,11 +309,14 @@ class TestMKLDNNSwishDim4(TestSwish):
def
ref_hardswish
(
x
,
threshold
=
6.0
,
scale
=
6.0
,
offset
=
3.0
):
x_dtype
=
x
.
dtype
if
x_dtype
==
'float16'
:
x_dtype
=
'float16'
x
=
x
.
astype
(
'float32'
)
return
(
x
*
np
.
minimum
(
np
.
maximum
(
x
+
offset
,
0.
),
threshold
)
/
scale
).
astype
(
x
.
dtype
)
scale
).
astype
(
x
_
dtype
)
@
skip_check_grad_ci
(
reason
=
"not implemented yet"
)
class
TestMKLDNNHardSwishDim4
(
TestHardSwish
):
def
setUp
(
self
):
...
...
@@ -343,9 +338,6 @@ class TestMKLDNNHardSwishDim4(TestHardSwish):
def
init_dtype
(
self
):
self
.
dtype
=
np
.
float32
def
test_check_grad
(
self
):
pass
class
TestMKLDNNMish
(
TestActivation
):
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录