Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
PaddlePaddle
X2Paddle
提交
f0d02885
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看板
提交
f0d02885
编写于
8月 12, 2022
作者:
W
wjj19950828
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
fixed mod op
上级
ebf440ac
变更
2
隐藏空白更改
内联
并排
Showing
2 changed file
with
16 addition
and
4 deletion
+16
-4
tests/onnx/test_auto_scan_mod.py
tests/onnx/test_auto_scan_mod.py
+1
-2
x2paddle/op_mapper/onnx2paddle/opset10.py
x2paddle/op_mapper/onnx2paddle/opset10.py
+15
-2
未找到文件。
tests/onnx/test_auto_scan_mod.py
浏览文件 @
f0d02885
...
...
@@ -42,8 +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"
,
"float32"
,
"float64"
]))
input_dtype
=
draw
(
st
.
sampled_from
([
"int32"
,
"int64"
]))
config
=
{
"op_names"
:
[
"Mod"
],
...
...
x2paddle/op_mapper/onnx2paddle/opset10.py
浏览文件 @
f0d02885
...
...
@@ -46,8 +46,6 @@ def _get_same_padding(in_size, kernel_size, stride, autopad):
class
OpSet10
(
OpSet9
):
def
__init__
(
self
,
decoder
,
paddle_graph
):
super
(
OpSet10
,
self
).
__init__
(
decoder
,
paddle_graph
)
# Support Mod op Since opset version >= 10
self
.
elementwise_ops
.
update
({
"Mod"
:
"paddle.mod"
})
@
print_mapping_info
def
AveragePool
(
self
,
node
):
...
...
@@ -99,3 +97,18 @@ class OpSet10(OpSet9):
inputs
=
{
'x'
:
val_x
if
isinstance
(
val_x
,
str
)
else
val_x
.
name
},
outputs
=
layer_outputs
,
**
layer_attrs
)
@
print_mapping_info
def
Mod
(
self
,
node
):
# 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
])
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录