Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
PaddlePaddle
X2Paddle
提交
02c3121e
X
X2Paddle
项目概览
PaddlePaddle
/
X2Paddle
大约 1 年 前同步成功
通知
328
Star
698
Fork
167
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
26
列表
看板
标记
里程碑
合并请求
4
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
X
X2Paddle
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
26
Issue
26
列表
看板
标记
里程碑
合并请求
4
合并请求
4
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
未验证
提交
02c3121e
编写于
12月 13, 2022
作者:
W
WJJ1995
提交者:
GitHub
12月 13, 2022
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
Support fmod=1 in Mod OP (#921)
* fixed Gemm bug * re-lint * fixed typo error * support fmod=1
上级
ce96b0f5
变更
2
隐藏空白更改
内联
并排
Showing
2 changed file
with
46 addition
and
9 deletion
+46
-9
tests/onnx/test_auto_scan_mod.py
tests/onnx/test_auto_scan_mod.py
+1
-1
x2paddle/op_mapper/onnx2paddle/opset10.py
x2paddle/op_mapper/onnx2paddle/opset10.py
+45
-8
未找到文件。
tests/onnx/test_auto_scan_mod.py
浏览文件 @
02c3121e
...
...
@@ -42,7 +42,7 @@ class TestModConvert(OPConvertAutoScanTest):
input_data
[
abs
(
input_data
)
<
1.0
]
=
1.0
return
input_data
input_dtype
=
draw
(
st
.
sampled_from
([
"int32"
,
"int64"
]))
input_dtype
=
draw
(
st
.
sampled_from
([
"
float32"
,
"
int32"
,
"int64"
]))
config
=
{
"op_names"
:
[
"Mod"
],
...
...
x2paddle/op_mapper/onnx2paddle/opset10.py
浏览文件 @
02c3121e
...
...
@@ -103,15 +103,52 @@ class OpSet10(OpSet9):
# Support Mod op Since opset version >= 10
val_x
=
self
.
graph
.
get_input_node
(
node
,
idx
=
0
,
copy
=
True
)
val_y
=
self
.
graph
.
get_input_node
(
node
,
idx
=
1
,
copy
=
True
)
input_dtype
=
str
(
val_x
.
dtype
)
assert
"int"
in
input_dtype
,
'Now only support int32 or int64 dtype'
fmod
=
node
.
get_attr
(
'fmod'
,
0
)
assert
fmod
==
0
,
'Now only support fmod == 0'
self
.
paddle_graph
.
add_layer
(
'paddle.mod'
,
inputs
=
{
"x"
:
val_x
.
name
,
"y"
:
val_y
.
name
},
outputs
=
[
node
.
name
])
x_dtype
=
str
(
val_x
.
dtype
)
y_dtype
=
str
(
val_y
.
dtype
)
if
"int"
in
x_dtype
and
"int"
in
y_dtype
and
fmod
==
1
:
fmod
=
0
if
fmod
==
0
:
self
.
paddle_graph
.
add_layer
(
'paddle.mod'
,
inputs
=
{
"x"
:
val_x
.
name
,
"y"
:
val_y
.
name
},
outputs
=
[
node
.
name
])
else
:
# Step1:trunc_div(a, b) = sign(a / b) * floor(abs(a / b))
self
.
paddle_graph
.
add_layer
(
'paddle.divide'
,
inputs
=
{
"x"
:
val_x
.
name
,
"y"
:
val_y
.
name
},
outputs
=
[
node
.
name
+
"_divide"
])
self
.
paddle_graph
.
add_layer
(
'paddle.sign'
,
inputs
=
{
"x"
:
node
.
name
+
"_divide"
},
outputs
=
[
node
.
name
+
"_sign"
])
self
.
paddle_graph
.
add_layer
(
'paddle.abs'
,
inputs
=
{
"x"
:
node
.
name
+
"_divide"
},
outputs
=
[
node
.
name
+
"_abs"
])
self
.
paddle_graph
.
add_layer
(
'paddle.floor'
,
inputs
=
{
"x"
:
node
.
name
+
"_abs"
},
outputs
=
[
node
.
name
+
"_floor"
])
self
.
paddle_graph
.
add_layer
(
'paddle.multiply'
,
inputs
=
{
"x"
:
node
.
name
+
"_sign"
,
"y"
:
node
.
name
+
"_floor"
},
outputs
=
[
node
.
name
+
"_trunc_div"
])
# Step2:result = a - trunc_div(a, b) * b
self
.
paddle_graph
.
add_layer
(
'paddle.multiply'
,
inputs
=
{
"x"
:
node
.
name
+
"_trunc_div"
,
"y"
:
val_y
.
name
},
outputs
=
[
node
.
name
+
"_trunc_div"
])
self
.
paddle_graph
.
add_layer
(
'paddle.subtract'
,
inputs
=
{
"x"
:
val_x
.
name
,
"y"
:
node
.
name
+
"_trunc_div"
},
outputs
=
[
node
.
name
])
@
print_mapping_info
def
IsInf
(
self
,
node
):
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录