Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
2301_76417823
X2Paddle
提交
807b4385
X
X2Paddle
项目概览
2301_76417823
/
X2Paddle
与 Fork 源项目一致
Fork自
PaddlePaddle / X2Paddle
通知
1
Star
0
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
X
X2Paddle
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
前往新版Gitcode,体验更适合开发者的 AI 搜索 >>
提交
807b4385
编写于
8月 02, 2022
作者:
W
wjj19950828
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
add Compare ops
上级
a393c7ee
变更
3
隐藏空白更改
内联
并排
Showing
3 changed file
with
146 addition
and
1 deletion
+146
-1
docs/inference_model_convertor/op_list.md
docs/inference_model_convertor/op_list.md
+1
-1
tests/onnx/test_auto_scan_compare_ops.py
tests/onnx/test_auto_scan_compare_ops.py
+78
-0
tests/onnx/test_auto_scan_equal.py
tests/onnx/test_auto_scan_equal.py
+67
-0
未找到文件。
docs/inference_model_convertor/op_list.md
浏览文件 @
807b4385
...
...
@@ -76,7 +76,7 @@
| 81 | Add | 82 | Concat | 83 | Max | 84 | Min |
| 85 | GreaterOrEqual | 86 | GatherND | 87 | And | 88 | cos |
| 89 | Neg | 90 | SpaceToDepth | 91 | GatherElement | 92 | Sin |
| 93 | CumSum | 94 | Or | 95 | Xor |
|
|
| 93 | CumSum | 94 | Or | 95 | Xor |
96 | Mod
|
## PyTorch
...
...
tests/onnx/test_auto_scan_compare_ops.py
0 → 100644
浏览文件 @
807b4385
# Copyright (c) 2022 PaddlePaddle Authors. All Rights Reserved.
#
# Licensed under the Apache License, Version 2.0 (the "License"
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
from
auto_scan_test
import
OPConvertAutoScanTest
from
hypothesis
import
reproduce_failure
from
onnxbase
import
randtool
import
hypothesis.strategies
as
st
import
numpy
as
np
import
unittest
min_opset_version_map
=
{
"Greater"
:
7
,
"Less"
:
7
,
"GreaterOrEqual"
:
12
,
"LessOrEqual"
:
12
,
}
class
TestCompareopsConvert
(
OPConvertAutoScanTest
):
"""
ONNX op: Compare ops
OPset version: 7~15
"""
def
sample_convert_config
(
self
,
draw
):
input1_shape
=
draw
(
st
.
lists
(
st
.
integers
(
min_value
=
10
,
max_value
=
20
),
min_size
=
2
,
max_size
=
4
))
if
draw
(
st
.
booleans
()):
input2_shape
=
[
input1_shape
[
-
1
]]
else
:
input2_shape
=
input1_shape
if
draw
(
st
.
booleans
()):
input2_shape
=
[
1
]
input_dtype
=
draw
(
st
.
sampled_from
([
"float32"
,
"float64"
]))
config
=
{
"op_names"
:
[
"Greater"
,
"Less"
,
"GreaterOrEqual"
,
"LessOrEqual"
],
"test_data_shapes"
:
[
input1_shape
,
input2_shape
],
"test_data_types"
:
[[
input_dtype
],
[
input_dtype
]],
"inputs_shape"
:
[],
"min_opset_version"
:
7
,
"inputs_name"
:
[
"x"
,
"y"
],
"outputs_name"
:
[
"z"
],
"delta"
:
1e-4
,
"rtol"
:
1e-4
,
"run_dynamic"
:
True
,
}
min_opset_versions
=
list
()
for
op_name
in
config
[
"op_names"
]:
min_opset_versions
.
append
(
min_opset_version_map
[
op_name
])
config
[
"min_opset_version"
]
=
min_opset_versions
attrs
=
{}
return
(
config
,
attrs
)
def
test
(
self
):
self
.
run_and_statis
(
max_examples
=
30
)
if
__name__
==
"__main__"
:
unittest
.
main
()
tests/onnx/test_auto_scan_equal.py
0 → 100644
浏览文件 @
807b4385
# Copyright (c) 2022 PaddlePaddle Authors. All Rights Reserved.
#
# Licensed under the Apache License, Version 2.0 (the "License"
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
from
auto_scan_test
import
OPConvertAutoScanTest
from
hypothesis
import
reproduce_failure
from
onnxbase
import
randtool
import
hypothesis.strategies
as
st
import
numpy
as
np
import
unittest
class
TestEqualConvert
(
OPConvertAutoScanTest
):
"""
ONNX op: Equal
OPset version: 7~15
"""
def
sample_convert_config
(
self
,
draw
):
input1_shape
=
draw
(
st
.
lists
(
st
.
integers
(
min_value
=
10
,
max_value
=
20
),
min_size
=
2
,
max_size
=
4
))
if
draw
(
st
.
booleans
()):
input2_shape
=
[
input1_shape
[
-
1
]]
else
:
input2_shape
=
input1_shape
if
draw
(
st
.
booleans
()):
input2_shape
=
[
1
]
input_dtype
=
draw
(
st
.
sampled_from
([
"int32"
,
"int64"
,
"bool"
]))
config
=
{
"op_names"
:
[
"Equal"
],
"test_data_shapes"
:
[
input1_shape
,
input2_shape
],
"test_data_types"
:
[[
input_dtype
],
[
input_dtype
]],
"inputs_shape"
:
[],
"min_opset_version"
:
7
,
"inputs_name"
:
[
"x"
,
"y"
],
"outputs_name"
:
[
"z"
],
"delta"
:
1e-4
,
"rtol"
:
1e-4
,
"run_dynamic"
:
True
,
}
attrs
=
{}
return
(
config
,
attrs
)
def
test
(
self
):
self
.
run_and_statis
(
max_examples
=
30
)
if
__name__
==
"__main__"
:
unittest
.
main
()
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录