OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-06 12:20:52 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity

[ONNX] update default opset_version to 13 (#73898)

Browse Source 
Summary:
And add a new tool to update it in the future, which follows the policy
of using "latest as of 18 months ago". This policy is meant to balance:
* recent enough to increase the odds of being able to successfully
  export
* old enough to increase the odds of exported model being runnable by
  different ONNX implementations

Related changes:

* test_models.py: explicitly fix opset_version to 9 rather than relying on default. Caffe2 doesn't support newer versions.
* symbolic_helper.py:
  * Remove a misleading comment
  * Remove unnecessary check in `_set_opset_version`
  * Use a range to define `_onnx_stable_opsets`
* test_pytorch_common.py:
  * Rename a variable from min -> max. I think it was a copy-paste error.
  * Make skip test messages more informative.
  * Remove unused `skipIfONNXShapeInference`. More on that below.
* test_pytorch_onnx_onnxruntime.py:
  * Make all the `TestCase` classes explicitly specify opset version.
  * Make `test_unsupported_pad` respect `opset_version` by using `run_test`
  * Unrelated simplification: make it obvious that all tests run with `onnx_shape_inference=True`. AFAICT this was already the case.
  * There was one test that was entirely disabled (test_tolist) because it was asking to be skipped whenever `onnx_shape_inference=True`, but it was always True. I changed the model being tested so as to preserve the intended test coverage but still have the test actually pass.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/73898

Reviewed By: msaroufim

Differential Revision: D35264615

Pulled By: malfet

fbshipit-source-id: cda8fbdffe4cc8210d8d96e659e3a9adf1b5f1d2
(cherry picked from commit b5e639e88828d34442282d0b50c977e610a2ba3a)
This commit is contained in:
Gary Miguel 2022-04-06 16:56:19 -07:00 committed by PyTorch MergeBot
parent 0dae42f5bd
commit ca374773b4
137 changed files with 1984 additions and 1047 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
scripts/onnx/test.sh
View File

@ -69,7 +69,7 @@ if [[ "$BUILD_ENVIRONMENT" == *ort_test1* || "${SHARD_NUMBER}" == "1" ]]; then
pytest "${args[@]}" \
"$top_dir/test/onnx/test_pytorch_onnx_onnxruntime.py::TestONNXRuntime_opset7" \
"$top_dir/test/onnx/test_pytorch_onnx_onnxruntime.py::TestONNXRuntime_opset8" \
"$top_dir/test/onnx/test_pytorch_onnx_onnxruntime.py::TestONNXRuntime" \
"$top_dir/test/onnx/test_pytorch_onnx_onnxruntime.py::TestONNXRuntime_opset9" \
"$top_dir/test/onnx/test_custom_ops.py" \
"$top_dir/test/onnx/test_models_onnxruntime.py" \
"$top_dir/test/onnx/test_utility_funs.py" \

4
test/jit/test_export_modes.py
View File

@ -82,7 +82,9 @@ class TestExportModes(JitTestCase):
ModelWithAtenNotONNXOp(), (x, y),
add_node_names=False,
do_constant_folding=False,
operator_export_type=OperatorExportTypes.ONNX_ATEN_FALLBACK)
operator_export_type=OperatorExportTypes.ONNX_ATEN_FALLBACK,
# support for linalg.qr was added in later op set versions.
opset_version=9)
# torch.fmod is using to test ONNX_ATEN.
# If you plan to remove fmod from aten, or found this test failed.

4
test/onnx/expect/TestOperators.test_acos.expect
View File

@ -1,4 +1,4 @@
ir_version: 4
ir_version: 7
producer_name: "pytorch"
producer_version: "CURRENT_VERSION"
graph {
@ -43,5 +43,5 @@ graph {
}
}
opset_import {
version: 9
version: 13
}

4
test/onnx/expect/TestOperators.test_add_broadcast.expect
View File

@ -1,4 +1,4 @@
ir_version: 4
ir_version: 7
producer_name: "pytorch"
producer_version: "CURRENT_VERSION"
graph {
@ -57,5 +57,5 @@ graph {
}
}
opset_import {
version: 9
version: 13
}

4
test/onnx/expect/TestOperators.test_add_left_broadcast.expect
View File

@ -1,4 +1,4 @@
ir_version: 4
ir_version: 7
producer_name: "pytorch"
producer_version: "CURRENT_VERSION"
graph {
@ -57,5 +57,5 @@ graph {
}
}
opset_import {
version: 9
version: 13
}

4
test/onnx/expect/TestOperators.test_add_size1_broadcast.expect
View File

@ -1,4 +1,4 @@
ir_version: 4
ir_version: 7
producer_name: "pytorch"
producer_version: "CURRENT_VERSION"
graph {
@ -60,5 +60,5 @@ graph {
}
}
opset_import {
version: 9
version: 13
}

4
test/onnx/expect/TestOperators.test_add_size1_right_broadcast.expect
View File

@ -1,4 +1,4 @@
ir_version: 4
ir_version: 7
producer_name: "pytorch"
producer_version: "CURRENT_VERSION"
graph {
@ -57,5 +57,5 @@ graph {
}
}
opset_import {
version: 9
version: 13
}

4
test/onnx/expect/TestOperators.test_add_size1_singleton_broadcast.expect
View File

@ -1,4 +1,4 @@
ir_version: 4
ir_version: 7
producer_name: "pytorch"
producer_version: "CURRENT_VERSION"
graph {
@ -60,5 +60,5 @@ graph {
}
}
opset_import {
version: 9
version: 13
}

4
test/onnx/expect/TestOperators.test_addconstant.expect
View File

@ -1,4 +1,4 @@
ir_version: 4
ir_version: 7
producer_name: "pytorch"
producer_version: "CURRENT_VERSION"
graph {
@ -57,5 +57,5 @@ graph {
}
}
opset_import {
version: 9
version: 13
}

4
test/onnx/expect/TestOperators.test_addmm.expect
View File

@ -1,4 +1,4 @@
ir_version: 4
ir_version: 7
producer_name: "pytorch"
producer_version: "CURRENT_VERSION"
graph {
@ -102,5 +102,5 @@ graph {
}
}
opset_import {
version: 9
version: 13
}

9
test/onnx/expect/TestOperators.test_argmax.expect
View File

@ -1,4 +1,4 @@
ir_version: 4
ir_version: 7
producer_name: "pytorch"
producer_version: "CURRENT_VERSION"
graph {
@ -17,6 +17,11 @@ graph {
i: 0
type: INT
}
attribute {
name: "select_last_index"
i: 0
type: INT
}
}
name: "torch_jit"
input {
@ -50,5 +55,5 @@ graph {
}
}
opset_import {
version: 9
version: 13
}

4
test/onnx/expect/TestOperators.test_asin.expect
View File

@ -1,4 +1,4 @@
ir_version: 4
ir_version: 7
producer_name: "pytorch"
producer_version: "CURRENT_VERSION"
graph {
@ -43,5 +43,5 @@ graph {
}
}
opset_import {
version: 9
version: 13
}

4
test/onnx/expect/TestOperators.test_at_op.expect
View File

@ -1,4 +1,4 @@
ir_version: 4
ir_version: 7
producer_name: "pytorch"
producer_version: "CURRENT_VERSION"
graph {
@ -54,5 +54,5 @@ graph {
}
}
opset_import {
version: 9
version: 13
}

4
test/onnx/expect/TestOperators.test_atan.expect
View File

@ -1,4 +1,4 @@
ir_version: 4
ir_version: 7
producer_name: "pytorch"
producer_version: "CURRENT_VERSION"
graph {
@ -43,5 +43,5 @@ graph {
}
}
opset_import {
version: 9
version: 13
}

53
test/onnx/expect/TestOperators.test_avg_pool2d.expect
View File

@ -1,40 +1,43 @@
ir_version: 4
ir_version: 7
producer_name: "pytorch"
producer_version: "CURRENT_VERSION"
graph {
node {
output: "onnx::Pad_1"
name: "Constant_0"
op_type: "Constant"
attribute {
name: "value"
t {
dims: 8
data_type: 7
raw_data: "\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000"
}
type: TENSOR
}
}
node {
input: "onnx::Pad_0"
output: "onnx::AveragePool_1"
name: "Pad_0"
input: "onnx::Pad_1"
output: "onnx::AveragePool_2"
name: "Pad_1"
op_type: "Pad"
attribute {
name: "mode"
s: "constant"
type: STRING
}
attribute {
name: "pads"
ints: 0
ints: 0
ints: 0
ints: 0
ints: 0
ints: 0
ints: 0
ints: 0
type: INTS
}
attribute {
name: "value"
f: 0
type: FLOAT
}
}
node {
input: "onnx::AveragePool_1"
output: "2"
name: "AveragePool_1"
input: "onnx::AveragePool_2"
output: "3"
name: "AveragePool_2"
op_type: "AveragePool"
attribute {
name: "ceil_mode"
i: 0
type: INT
}
attribute {
name: "kernel_shape"
ints: 3
@ -80,7 +83,7 @@ graph {
}
}
output {
name: "2"
name: "3"
type {
tensor_type {
elem_type: 1
@ -103,5 +106,5 @@ graph {
}
}
opset_import {
version: 9
version: 13
}

4
test/onnx/expect/TestOperators.test_baddbmm.expect
View File

@ -1,4 +1,4 @@
ir_version: 4
ir_version: 7
producer_name: "pytorch"
producer_version: "CURRENT_VERSION"
graph {
@ -119,5 +119,5 @@ graph {
}
}
opset_import {
version: 9
version: 13
}

4
test/onnx/expect/TestOperators.test_basic.expect
View File

@ -1,4 +1,4 @@
ir_version: 4
ir_version: 7
producer_name: "pytorch"
producer_version: "CURRENT_VERSION"
graph {
@ -76,5 +76,5 @@ graph {
}
}
opset_import {
version: 9
version: 13
}

4
test/onnx/expect/TestOperators.test_batchnorm.expect
View File

@ -1,4 +1,4 @@
ir_version: 4
ir_version: 7
producer_name: "pytorch"
producer_version: "CURRENT_VERSION"
graph {
@ -145,5 +145,5 @@ graph {
}
}
opset_import {
version: 9
version: 13
}

4
test/onnx/expect/TestOperators.test_batchnorm_1d.expect
View File

@ -1,4 +1,4 @@
ir_version: 4
ir_version: 7
producer_name: "pytorch"
producer_version: "CURRENT_VERSION"
graph {
@ -133,5 +133,5 @@ graph {
}
}
opset_import {
version: 9
version: 13
}

4
test/onnx/expect/TestOperators.test_batchnorm_noaffine.expect
View File

@ -1,4 +1,4 @@
ir_version: 4
ir_version: 7
producer_name: "pytorch"
producer_version: "CURRENT_VERSION"
graph {
@ -135,5 +135,5 @@ graph {
}
}
opset_import {
version: 9
version: 13
}

4
test/onnx/expect/TestOperators.test_batchnorm_onnx_irv4.expect
View File

@ -1,4 +1,4 @@
ir_version: 4
ir_version: 7
producer_name: "pytorch"
producer_version: "CURRENT_VERSION"
graph {
@ -93,5 +93,5 @@ graph {
}
}
opset_import {
version: 9
version: 13
}

4
test/onnx/expect/TestOperators.test_batchnorm_training.expect
View File

@ -1,4 +1,4 @@
ir_version: 4
ir_version: 7
producer_name: "pytorch"
producer_version: "CURRENT_VERSION"
graph {
@ -149,5 +149,5 @@ graph {
}
}
opset_import {
version: 9
version: 13
}

158
test/onnx/expect/TestOperators.test_chunk.expect
View File

@ -1,28 +1,158 @@
ir_version: 4
ir_version: 7
producer_name: "pytorch"
producer_version: "CURRENT_VERSION"
graph {
node {
input: "onnx::Split_0"
output: "1"
output: "2"
name: "Split_0"
op_type: "Split"
input: "onnx::Shape_0"
output: "onnx::Gather_1"
name: "Shape_0"
op_type: "Shape"
}
node {
output: "onnx::Gather_2"
name: "Constant_1"
op_type: "Constant"
attribute {
name: "value"
t {
dims: 1
data_type: 7
raw_data: "\000\000\000\000\000\000\000\000"
}
type: TENSOR
}
}
node {
input: "onnx::Gather_1"
input: "onnx::Gather_2"
output: "onnx::Add_3"
name: "Gather_2"
op_type: "Gather"
attribute {
name: "axis"
i: 0
type: INT
}
}
node {
output: "onnx::Slice_4"
name: "Constant_3"
op_type: "Constant"
attribute {
name: "split"
ints: 2
ints: 1
type: INTS
name: "value"
t {
dims: 1
data_type: 7
raw_data: "\000\000\000\000\000\000\000\000"
}
type: TENSOR
}
}
node {
output: "onnx::Add_5"
name: "Constant_4"
op_type: "Constant"
attribute {
name: "value"
t {
dims: 1
data_type: 7
raw_data: "\001\000\000\000\000\000\000\000"
}
type: TENSOR
}
}
node {
input: "onnx::Add_3"
input: "onnx::Add_5"
output: "onnx::Div_6"
name: "Add_5"
op_type: "Add"
}
node {
output: "onnx::Div_7"
name: "Constant_6"
op_type: "Constant"
attribute {
name: "value"
t {
dims: 1
data_type: 7
raw_data: "\002\000\000\000\000\000\000\000"
}
type: TENSOR
}
}
node {
input: "onnx::Div_6"
input: "onnx::Div_7"
output: "onnx::Mul_8"
name: "Div_7"
op_type: "Div"
}
node {
output: "onnx::Mul_9"
name: "Constant_8"
op_type: "Constant"
attribute {
name: "value"
t {
dims: 1
data_type: 7
raw_data: "\001\000\000\000\000\000\000\000"
}
type: TENSOR
}
}
node {
input: "onnx::Mul_8"
input: "onnx::Mul_9"
output: "onnx::Slice_10"
name: "Mul_9"
op_type: "Mul"
}
node {
input: "onnx::Shape_0"
input: "onnx::Slice_4"
input: "onnx::Slice_10"
input: "onnx::Gather_2"
output: "11"
name: "Slice_10"
op_type: "Slice"
}
node {
output: "onnx::Mul_12"
name: "Constant_11"
op_type: "Constant"
attribute {
name: "value"
t {
dims: 1
data_type: 7
raw_data: "\002\000\000\000\000\000\000\000"
}
type: TENSOR
}
}
node {
input: "onnx::Mul_8"
input: "onnx::Mul_12"
output: "onnx::Slice_13"
name: "Mul_12"
op_type: "Mul"
}
node {
input: "onnx::Shape_0"
input: "onnx::Slice_10"
input: "onnx::Slice_13"
input: "onnx::Gather_2"
output: "14"
name: "Slice_13"
op_type: "Slice"
}
name: "torch_jit"
input {
name: "onnx::Split_0"
name: "onnx::Shape_0"
type {
tensor_type {
elem_type: 1
@ -35,7 +165,7 @@ graph {
}
}
output {
name: "1"
name: "11"
type {
tensor_type {
elem_type: 1
@ -48,7 +178,7 @@ graph {
}
}
output {
name: "2"
name: "14"
type {
tensor_type {
elem_type: 1
@ -62,5 +192,5 @@ graph {
}
}
opset_import {
version: 9
version: 13
}

30
test/onnx/expect/TestOperators.test_clip.expect
View File

@ -1,24 +1,26 @@
ir_version: 4
ir_version: 7
producer_name: "pytorch"
producer_version: "CURRENT_VERSION"
graph {
node {
input: "onnx::Clip_0"
output: "1"
input: "onnx::Clip_6"
input: "onnx::Clip_7"
output: "5"
name: "Clip_0"
op_type: "Clip"
attribute {
name: "max"
f: 0.5
type: FLOAT
}
attribute {
name: "min"
f: -0.5
type: FLOAT
}
}
name: "torch_jit"
initializer {
data_type: 1
name: "onnx::Clip_6"
raw_data: "\000\000\000\277"
}
initializer {
data_type: 1
name: "onnx::Clip_7"
raw_data: "\000\000\000?"
}
input {
name: "onnx::Clip_0"
type {
@ -36,7 +38,7 @@ graph {
}
}
output {
name: "1"
name: "5"
type {
tensor_type {
elem_type: 1
@ -53,5 +55,5 @@ graph {
}
}
opset_import {
version: 9
version: 13
}

28
test/onnx/expect/TestOperators.test_clip_max.expect
View File

@ -1,19 +1,21 @@
ir_version: 4
ir_version: 7
producer_name: "pytorch"
producer_version: "CURRENT_VERSION"
graph {
node {
input: "onnx::Clip_0"
output: "1"
input: ""
input: "onnx::Clip_7"
output: "5"
name: "Clip_0"
op_type: "Clip"
attribute {
name: "max"
f: 0.1
type: FLOAT
}
}
name: "torch_jit"
initializer {
data_type: 1
name: "onnx::Clip_7"
raw_data: "\315\314\314="
}
input {
name: "onnx::Clip_0"
type {
@ -37,22 +39,22 @@ graph {
}
}
output {
name: "1"
name: "5"
type {
tensor_type {
elem_type: 1
shape {
dim {
dim_value: 1
dim_param: "Clip5_dim_0"
}
dim {
dim_value: 2
dim_param: "Clip5_dim_1"
}
dim {
dim_value: 3
dim_param: "Clip5_dim_2"
}
dim {
dim_value: 4
dim_param: "Clip5_dim_3"
}
}
}
@ -60,5 +62,5 @@ graph {
}
}
opset_import {
version: 9
version: 13
}

28
test/onnx/expect/TestOperators.test_clip_min.expect
View File

@ -1,19 +1,21 @@
ir_version: 4
ir_version: 7
producer_name: "pytorch"
producer_version: "CURRENT_VERSION"
graph {
node {
input: "onnx::Clip_0"
output: "1"
input: "onnx::Clip_7"
input: ""
output: "5"
name: "Clip_0"
op_type: "Clip"
attribute {
name: "min"
f: -0.1
type: FLOAT
}
}
name: "torch_jit"
initializer {
data_type: 1
name: "onnx::Clip_7"
raw_data: "\315\314\314\275"
}
input {
name: "onnx::Clip_0"
type {
@ -37,22 +39,22 @@ graph {
}
}
output {
name: "1"
name: "5"
type {
tensor_type {
elem_type: 1
shape {
dim {
dim_value: 1
dim_param: "Clip5_dim_0"
}
dim {
dim_value: 2
dim_param: "Clip5_dim_1"
}
dim {
dim_value: 3
dim_param: "Clip5_dim_2"
}
dim {
dim_value: 4
dim_param: "Clip5_dim_3"
}
}
}
@ -60,5 +62,5 @@ graph {
}
}
opset_import {
version: 9
version: 13
}

4
test/onnx/expect/TestOperators.test_concat2.expect
View File

@ -1,4 +1,4 @@
ir_version: 4
ir_version: 7
producer_name: "pytorch"
producer_version: "CURRENT_VERSION"
graph {
@ -65,5 +65,5 @@ graph {
}
}
opset_import {
version: 9
version: 13
}

4
test/onnx/expect/TestOperators.test_conv.expect
View File

@ -1,4 +1,4 @@
ir_version: 4
ir_version: 7
producer_name: "pytorch"
producer_version: "CURRENT_VERSION"
graph {
@ -118,5 +118,5 @@ graph {
}
}
opset_import {
version: 9
version: 13
}

4
test/onnx/expect/TestOperators.test_conv_onnx_irv4.expect
View File

@ -1,4 +1,4 @@
ir_version: 4
ir_version: 7
producer_name: "pytorch"
producer_version: "CURRENT_VERSION"
graph {
@ -96,5 +96,5 @@ graph {
}
}
opset_import {
version: 9
version: 13
}

4
test/onnx/expect/TestOperators.test_convtranspose.expect
View File

@ -1,4 +1,4 @@
ir_version: 4
ir_version: 7
producer_name: "pytorch"
producer_version: "CURRENT_VERSION"
graph {
@ -124,5 +124,5 @@ graph {
}
}
opset_import {
version: 9
version: 13
}

4
test/onnx/expect/TestOperators.test_cos.expect
View File

@ -1,4 +1,4 @@
ir_version: 4
ir_version: 7
producer_name: "pytorch"
producer_version: "CURRENT_VERSION"
graph {
@ -43,5 +43,5 @@ graph {
}
}
opset_import {
version: 9
version: 13
}

4
test/onnx/expect/TestOperators.test_dict.expect
View File

@ -1,4 +1,4 @@
ir_version: 4
ir_version: 7
producer_name: "pytorch"
producer_version: "CURRENT_VERSION"
graph {
@ -60,5 +60,5 @@ graph {
}
}
opset_import {
version: 9
version: 13
}

4
test/onnx/expect/TestOperators.test_dict_str.expect
View File

@ -1,4 +1,4 @@
ir_version: 4
ir_version: 7
producer_name: "pytorch"
producer_version: "CURRENT_VERSION"
graph {
@ -63,5 +63,5 @@ graph {
}
}
opset_import {
version: 9
version: 13
}

4
test/onnx/expect/TestOperators.test_dim.expect
View File

@ -1,4 +1,4 @@
ir_version: 4
ir_version: 7
producer_name: "pytorch"
producer_version: "CURRENT_VERSION"
graph {
@ -28,5 +28,5 @@ graph {
}
}
opset_import {
version: 9
version: 13
}

4
test/onnx/expect/TestOperators.test_dropout.expect
View File

@ -1,4 +1,4 @@
ir_version: 4
ir_version: 7
producer_name: "pytorch"
producer_version: "CURRENT_VERSION"
graph {
@ -42,5 +42,5 @@ graph {
}
}
opset_import {
version: 9
version: 13
}

51
test/onnx/expect/TestOperators.test_dropout_default.expect
View File

@ -1,23 +1,46 @@
ir_version: 4
ir_version: 7
producer_name: "pytorch"
producer_version: "CURRENT_VERSION"
graph {
node {
input: "x"
output: "onnx::ReduceMax_1"
output: "2"
name: "Dropout_0"
op_type: "Dropout"
output: "onnx::Dropout_1"
name: "Constant_0"
op_type: "Constant"
attribute {
name: "ratio"
f: 0.5
type: FLOAT
name: "value"
t {
data_type: 1
raw_data: "\000\000\000?"
}
type: TENSOR
}
}
node {
input: "onnx::ReduceMax_1"
output: "3"
name: "ReduceMax_1"
output: "onnx::Dropout_2"
name: "Constant_1"
op_type: "Constant"
attribute {
name: "value"
t {
data_type: 9
raw_data: "\001"
}
type: TENSOR
}
}
node {
input: "x"
input: "onnx::Dropout_1"
input: "onnx::Dropout_2"
output: "onnx::ReduceMax_3"
output: "4"
name: "Dropout_2"
op_type: "Dropout"
}
node {
input: "onnx::ReduceMax_3"
output: "5"
name: "ReduceMax_3"
op_type: "ReduceMax"
attribute {
name: "keepdims"
@ -43,7 +66,7 @@ graph {
}
}
output {
name: "3"
name: "5"
type {
tensor_type {
elem_type: 1
@ -54,5 +77,5 @@ graph {
}
}
opset_import {
version: 9
version: 13
}

51
test/onnx/expect/TestOperators.test_dropout_training.expect
View File

@ -1,23 +1,46 @@
ir_version: 4
ir_version: 7
producer_name: "pytorch"
producer_version: "CURRENT_VERSION"
graph {
node {
input: "x"
output: "onnx::ReduceMax_1"
output: "2"
name: "Dropout_0"
op_type: "Dropout"
output: "onnx::Dropout_1"
name: "Constant_0"
op_type: "Constant"
attribute {
name: "ratio"
f: 0.5
type: FLOAT
name: "value"
t {
data_type: 1
raw_data: "\000\000\000?"
}
type: TENSOR
}
}
node {
input: "onnx::ReduceMax_1"
output: "3"
name: "ReduceMax_1"
output: "onnx::Dropout_2"
name: "Constant_1"
op_type: "Constant"
attribute {
name: "value"
t {
data_type: 9
raw_data: "\001"
}
type: TENSOR
}
}
node {
input: "x"
input: "onnx::Dropout_1"
input: "onnx::Dropout_2"
output: "onnx::ReduceMax_3"
output: "4"
name: "Dropout_2"
op_type: "Dropout"
}
node {
input: "onnx::ReduceMax_3"
output: "5"
name: "ReduceMax_3"
op_type: "ReduceMax"
attribute {
name: "keepdims"
@ -43,7 +66,7 @@ graph {
}
}
output {
name: "3"
name: "5"
type {
tensor_type {
elem_type: 1
@ -54,5 +77,5 @@ graph {
}
}
opset_import {
version: 9
version: 13
}

4
test/onnx/expect/TestOperators.test_elu.expect
View File

@ -1,4 +1,4 @@
ir_version: 4
ir_version: 7
producer_name: "pytorch"
producer_version: "CURRENT_VERSION"
graph {
@ -60,5 +60,5 @@ graph {
}
}
opset_import {
version: 9
version: 13
}

368
test/onnx/expect/TestOperators.test_embedding_bags.expect
View File

@ -1,47 +1,359 @@
ir_version: 4
ir_version: 7
producer_name: "pytorch"
producer_version: "CURRENT_VERSION"
graph {
node {
input: "weight"
input: "input"
input: "offsets"
output: "3"
output: "4"
output: "onnx::Cast_3"
op_type: "Constant"
attribute {
name: "value"
t {
data_type: 7
raw_data: "\001\000\000\000\000\000\000\000"
}
type: TENSOR
}
}
node {
input: "onnx::Cast_3"
output: "onnx::Loop_4"
op_type: "Cast"
attribute {
name: "to"
i: 9
type: INT
}
}
node {
output: "5"
output: "6"
op_type: "ATen"
op_type: "Constant"
attribute {
name: "include_last_offset"
name: "value"
t {
dims: 1
data_type: 7
raw_data: "\000\000\000\000\000\000\000\000"
}
type: TENSOR
}
}
node {
input: "input"
output: "onnx::Gather_6"
op_type: "Shape"
}
node {
output: "onnx::Gather_7"
op_type: "Constant"
attribute {
name: "value"
t {
data_type: 7
raw_data: "\000\000\000\000\000\000\000\000"
}
type: TENSOR
}
}
node {
input: "onnx::Gather_6"
input: "onnx::Gather_7"
output: "onnx::Unsqueeze_8"
op_type: "Gather"
attribute {
name: "axis"
i: 0
type: INT
}
}
node {
output: "onnx::Unsqueeze_9"
op_type: "Constant"
attribute {
name: "mode"
i: 1
type: INT
name: "value"
t {
dims: 1
data_type: 7
raw_data: "\000\000\000\000\000\000\000\000"
}
type: TENSOR
}
}
node {
input: "onnx::Unsqueeze_8"
input: "onnx::Unsqueeze_9"
output: "onnx::Concat_10"
op_type: "Unsqueeze"
}
node {
input: "offsets"
input: "onnx::Concat_10"
output: "onnx::Slice_11"
op_type: "Concat"
attribute {
name: "operator"
s: "embedding_bag"
type: STRING
}
attribute {
name: "overload_name"
s: ""
type: STRING
}
attribute {
name: "scale_grad_by_freq"
name: "axis"
i: 0
type: INT
}
}
node {
output: "onnx::Slice_12"
op_type: "Constant"
attribute {
name: "sparse"
name: "value"
t {
dims: 1
data_type: 7
raw_data: "\000\000\000\000\000\000\000\000"
}
type: TENSOR
}
}
node {
output: "onnx::Slice_13"
op_type: "Constant"
attribute {
name: "value"
t {
dims: 1
data_type: 7
raw_data: "\001\000\000\000\000\000\000\000"
}
type: TENSOR
}
}
node {
output: "onnx::Slice_14"
op_type: "Constant"
attribute {
name: "value"
t {
dims: 1
data_type: 7
raw_data: "\377\377\377\377\377\377\377\177"
}
type: TENSOR
}
}
node {
output: "onnx::Slice_15"
op_type: "Constant"
attribute {
name: "value"
t {
dims: 1
data_type: 7
raw_data: "\001\000\000\000\000\000\000\000"
}
type: TENSOR
}
}
node {
input: "onnx::Slice_11"
input: "onnx::Slice_13"
input: "onnx::Slice_14"
input: "onnx::Slice_12"
input: "onnx::Slice_15"
output: "onnx::Shape_16"
op_type: "Slice"
}
node {
input: "onnx::Shape_16"
output: "onnx::Gather_17"
op_type: "Shape"
}
node {
output: "onnx::Gather_18"
op_type: "Constant"
attribute {
name: "value"
t {
data_type: 7
raw_data: "\000\000\000\000\000\000\000\000"
}
type: TENSOR
}
}
node {
input: "onnx::Gather_17"
input: "onnx::Gather_18"
output: "onnx::Loop_19"
op_type: "Gather"
attribute {
name: "axis"
i: 0
type: INT
}
}
node {
input: "onnx::Loop_19"
input: "onnx::Loop_4"
output: "20"
op_type: "Loop"
attribute {
name: "body"
g {
node {
input: "onnx::Slice_11"
input: "21"
output: "23"
name: "Gather_0"
op_type: "Gather"
attribute {
name: "axis"
i: 0
type: INT
}
}
node {
input: "onnx::Shape_16"
input: "21"
output: "24"
name: "Gather_1"
op_type: "Gather"
attribute {
name: "axis"
i: 0
type: INT
}
}
node {
output: "25"
name: "Constant_2"
op_type: "Constant"
attribute {
name: "value"
t {
dims: 1
data_type: 7
raw_data: "\000\000\000\000\000\000\000\000"
}
type: TENSOR
}
}
node {
input: "23"
input: "25"
output: "26"
name: "Unsqueeze_3"
op_type: "Unsqueeze"
}
node {
output: "27"
name: "Constant_4"
op_type: "Constant"
attribute {
name: "value"
t {
dims: 1
data_type: 7
raw_data: "\000\000\000\000\000\000\000\000"
}
type: TENSOR
}
}
node {
input: "24"
input: "27"
output: "28"
name: "Unsqueeze_5"
op_type: "Unsqueeze"
}
node {
input: "input"
input: "26"
input: "28"
input: "5"
output: "29"
name: "Slice_6"
op_type: "Slice"
}
node {
input: "weight"
input: "29"
output: "30"
name: "Gather_7"
op_type: "Gather"
attribute {
name: "axis"
i: 0
type: INT
}
}
node {
input: "30"
output: "31"
name: "ReduceMean_8"
op_type: "ReduceMean"
attribute {
name: "axes"
ints: 0
type: INTS
}
attribute {
name: "keepdims"
i: 0
type: INT
}
}
node {
input: "onnx::Loop_4"
output: "32"
name: "Cast_9"
op_type: "Cast"
attribute {
name: "to"
i: 9
type: INT
}
}
name: "torch_jit1"
input {
name: "21"
type {
tensor_type {
elem_type: 7
shape {
}
}
}
}
input {
name: "22"
type {
tensor_type {
elem_type: 9
shape {
}
}
}
}
output {
name: "32"
type {
tensor_type {
elem_type: 9
shape {
}
}
}
}
output {
name: "31"
type {
tensor_type {
elem_type: 1
shape {
dim {
dim_param: "Loop20_dim_1"
}
}
}
}
}
}
type: GRAPH
}
}
name: "torch_jit"
initializer {
dims: 10
@ -93,16 +405,16 @@ graph {
}
}
output {
name: "3"
name: "20"
type {
tensor_type {
elem_type: 1
shape {
dim {
dim_value: 1
dim_param: "Loop20_dim_0"
}
dim {
dim_value: 8
dim_param: "Loop20_dim_1"
}
}
}
@ -110,5 +422,5 @@ graph {
}
}
opset_import {
version: 9
version: 13
}

4
test/onnx/expect/TestOperators.test_empty_like.expect
View File

@ -1,4 +1,4 @@
ir_version: 4
ir_version: 7
producer_name: "pytorch"
producer_version: "CURRENT_VERSION"
graph {
@ -36,5 +36,5 @@ graph {
}
}
opset_import {
version: 9
version: 13
}

4
test/onnx/expect/TestOperators.test_equal.expect
View File

@ -1,4 +1,4 @@
ir_version: 4
ir_version: 7
producer_name: "pytorch"
producer_version: "CURRENT_VERSION"
graph {
@ -72,5 +72,5 @@ graph {
}
}
opset_import {
version: 9
version: 13
}

4
test/onnx/expect/TestOperators.test_erf.expect
View File

@ -1,4 +1,4 @@
ir_version: 4
ir_version: 7
producer_name: "pytorch"
producer_version: "CURRENT_VERSION"
graph {
@ -55,5 +55,5 @@ graph {
}
}
opset_import {
version: 9
version: 13
}

4
test/onnx/expect/TestOperators.test_exp.expect
View File

@ -1,4 +1,4 @@
ir_version: 4
ir_version: 7
producer_name: "pytorch"
producer_version: "CURRENT_VERSION"
graph {
@ -43,5 +43,5 @@ graph {
}
}
opset_import {
version: 9
version: 13
}

4
test/onnx/expect/TestOperators.test_expand.expect
View File

@ -1,4 +1,4 @@
ir_version: 4
ir_version: 7
producer_name: "pytorch"
producer_version: "CURRENT_VERSION"
graph {
@ -131,5 +131,5 @@ graph {
}
}
opset_import {
version: 9
version: 13
}

86
test/onnx/expect/TestOperators.test_flatten.expect
View File

@ -1,4 +1,4 @@
ir_version: 4
ir_version: 7
producer_name: "pytorch"
producer_version: "CURRENT_VERSION"
graph {
@ -9,30 +9,60 @@ graph {
op_type: "Shape"
}
node {
input: "onnx::Slice_1"
output: "onnx::Concat_2"
name: "Slice_1"
op_type: "Slice"
output: "onnx::Slice_2"
name: "Constant_1"
op_type: "Constant"
attribute {
name: "axes"
ints: 0
type: INTS
}
attribute {
name: "ends"
ints: 0
type: INTS
}
attribute {
name: "starts"
ints: 0
type: INTS
name: "value"
t {
dims: 1
data_type: 7
raw_data: "\000\000\000\000\000\000\000\000"
}
type: TENSOR
}
}
node {
output: "onnx::Concat_3"
output: "onnx::Slice_3"
name: "Constant_2"
op_type: "Constant"
attribute {
name: "value"
t {
dims: 1
data_type: 7
raw_data: "\000\000\000\000\000\000\000\000"
}
type: TENSOR
}
}
node {
output: "onnx::Slice_4"
name: "Constant_3"
op_type: "Constant"
attribute {
name: "value"
t {
dims: 1
data_type: 7
raw_data: "\000\000\000\000\000\000\000\000"
}
type: TENSOR
}
}
node {
input: "onnx::Slice_1"
input: "onnx::Slice_3"
input: "onnx::Slice_4"
input: "onnx::Slice_2"
output: "onnx::Concat_5"
name: "Slice_4"
op_type: "Slice"
}
node {
output: "onnx::Concat_6"
name: "Constant_5"
op_type: "Constant"
attribute {
name: "value"
t {
@ -44,10 +74,10 @@ graph {
}
}
node {
input: "onnx::Concat_2"
input: "onnx::Concat_3"
output: "onnx::Reshape_4"
name: "Concat_3"
input: "onnx::Concat_5"
input: "onnx::Concat_6"
output: "onnx::Reshape_7"
name: "Concat_6"
op_type: "Concat"
attribute {
name: "axis"
@ -57,9 +87,9 @@ graph {
}
node {
input: "onnx::Shape_0"
input: "onnx::Reshape_4"
output: "5"
name: "Reshape_4"
input: "onnx::Reshape_7"
output: "8"
name: "Reshape_7"
op_type: "Reshape"
}
name: "torch_jit"
@ -86,7 +116,7 @@ graph {
}
}
output {
name: "5"
name: "8"
type {
tensor_type {
elem_type: 1
@ -100,5 +130,5 @@ graph {
}
}
opset_import {
version: 9
version: 13
}

4
test/onnx/expect/TestOperators.test_flatten2D.expect
View File

@ -1,4 +1,4 @@
ir_version: 4
ir_version: 7
producer_name: "pytorch"
producer_version: "CURRENT_VERSION"
graph {
@ -54,5 +54,5 @@ graph {
}
}
opset_import {
version: 9
version: 13
}

48
test/onnx/expect/TestOperators.test_frobenius_norm.expect
View File

@ -1,35 +1,49 @@
ir_version: 4
ir_version: 7
producer_name: "pytorch"
producer_version: "CURRENT_VERSION"
graph {
node {
input: "x"
input: "x"
output: "onnx::ReduceSum_1"
name: "Mul_0"
name: "Constant_0"
op_type: "Constant"
attribute {
name: "value"
t {
dims: 2
data_type: 7
raw_data: "\000\000\000\000\000\000\000\000\001\000\000\000\000\000\000\000"
}
type: TENSOR
}
}
node {
input: "x"
input: "x"
output: "onnx::ReduceSum_2"
name: "Mul_1"
op_type: "Mul"
}
node {
input: "onnx::ReduceSum_2"
input: "onnx::ReduceSum_1"
output: "onnx::Sqrt_2"
name: "ReduceSum_1"
output: "onnx::Sqrt_3"
name: "ReduceSum_2"
op_type: "ReduceSum"
attribute {
name: "axes"
ints: 0
ints: 1
type: INTS
}
attribute {
name: "keepdims"
i: 1
type: INT
}
attribute {
name: "noop_with_empty_axes"
i: 0
type: INT
}
}
node {
input: "onnx::Sqrt_2"
output: "3"
name: "Sqrt_2"
input: "onnx::Sqrt_3"
output: "4"
name: "Sqrt_3"
op_type: "Sqrt"
}
name: "torch_jit"
@ -53,7 +67,7 @@ graph {
}
}
output {
name: "3"
name: "4"
type {
tensor_type {
elem_type: 1
@ -73,5 +87,5 @@ graph {
}
}
opset_import {
version: 9
version: 13
}

4
test/onnx/expect/TestOperators.test_full.expect
View File

@ -1,4 +1,4 @@
ir_version: 4
ir_version: 7
producer_name: "pytorch"
producer_version: "CURRENT_VERSION"
graph {
@ -36,5 +36,5 @@ graph {
}
}
opset_import {
version: 9
version: 13
}

4
test/onnx/expect/TestOperators.test_full_like.expect
View File

@ -1,4 +1,4 @@
ir_version: 4
ir_version: 7
producer_name: "pytorch"
producer_version: "CURRENT_VERSION"
graph {
@ -36,5 +36,5 @@ graph {
}
}
opset_import {
version: 9
version: 13
}

112
test/onnx/expect/TestOperators.test_gather.expect
View File

@ -1,114 +1,22 @@
ir_version: 4
ir_version: 7
producer_name: "pytorch"
producer_version: "CURRENT_VERSION"
graph {
node {
output: "onnx::OneHot_2"
name: "Constant_0"
op_type: "Constant"
attribute {
name: "value"
t {
dims: 2
data_type: 7
raw_data: "\000\000\000\000\000\000\000\000\001\000\000\000\000\000\000\000"
}
type: TENSOR
}
}
node {
output: "onnx::Gather_3"
name: "Constant_1"
op_type: "Constant"
attribute {
name: "value"
t {
dims: 1
data_type: 7
raw_data: "\001\000\000\000\000\000\000\000"
}
type: TENSOR
}
}
node {
input: "onnx::Shape_0"
output: "onnx::Gather_4"
name: "Shape_2"
op_type: "Shape"
}
node {
input: "onnx::Gather_4"
input: "onnx::Gather_3"
output: "onnx::OneHot_5"
name: "Gather_3"
op_type: "Gather"
attribute {
name: "axis"
i: 0
type: INT
}
}
node {
input: "onnx::OneHot_1"
input: "onnx::OneHot_5"
input: "onnx::OneHot_2"
output: "onnx::Cast_6"
name: "OneHot_4"
op_type: "OneHot"
input: "onnx::GatherElements_0"
input: "onnx::GatherElements_1"
output: "2"
name: "GatherElements_0"
op_type: "GatherElements"
attribute {
name: "axis"
i: 1
type: INT
}
}
node {
input: "onnx::Cast_6"
output: "onnx::Mul_7"
name: "Cast_5"
op_type: "Cast"
attribute {
name: "to"
i: 1
type: INT
}
}
node {
input: "onnx::Shape_0"
output: "onnx::Mul_8"
name: "Unsqueeze_6"
op_type: "Unsqueeze"
attribute {
name: "axes"
ints: 2
type: INTS
}
}
node {
input: "onnx::Mul_8"
input: "onnx::Mul_7"
output: "onnx::ReduceSum_9"
name: "Mul_7"
op_type: "Mul"
}
node {
input: "onnx::ReduceSum_9"
output: "10"
name: "ReduceSum_8"
op_type: "ReduceSum"
attribute {
name: "axes"
ints: 1
type: INTS
}
attribute {
name: "keepdims"
i: 0
type: INT
}
}
name: "torch_jit"
input {
name: "onnx::Shape_0"
name: "onnx::GatherElements_0"
type {
tensor_type {
elem_type: 1
@ -127,7 +35,7 @@ graph {
}
}
input {
name: "onnx::OneHot_1"
name: "onnx::GatherElements_1"
type {
tensor_type {
elem_type: 7
@ -146,7 +54,7 @@ graph {
}
}
output {
name: "10"
name: "2"
type {
tensor_type {
elem_type: 1
@ -166,5 +74,5 @@ graph {
}
}
opset_import {
version: 9
version: 13
}

26
test/onnx/expect/TestOperators.test_ge.expect
View File

@ -1,23 +1,17 @@
ir_version: 4
ir_version: 7
producer_name: "pytorch"
producer_version: "CURRENT_VERSION"
graph {
node {
input: "onnx::Less_0"
input: "onnx::Less_1"
output: "onnx::Not_2"
name: "Less_0"
op_type: "Less"
}
node {
input: "onnx::Not_2"
output: "3"
name: "Not_1"
op_type: "Not"
input: "onnx::GreaterOrEqual_0"
input: "onnx::GreaterOrEqual_1"
output: "2"
name: "GreaterOrEqual_0"
op_type: "GreaterOrEqual"
}
name: "torch_jit"
input {
name: "onnx::Less_0"
name: "onnx::GreaterOrEqual_0"
type {
tensor_type {
elem_type: 6
@ -33,7 +27,7 @@ graph {
}
}
input {
name: "onnx::Less_1"
name: "onnx::GreaterOrEqual_1"
type {
tensor_type {
elem_type: 6
@ -49,7 +43,7 @@ graph {
}
}
output {
name: "3"
name: "2"
type {
tensor_type {
elem_type: 9
@ -66,5 +60,5 @@ graph {
}
}
opset_import {
version: 9
version: 13
}

4
test/onnx/expect/TestOperators.test_gelu.expect
View File

@ -1,4 +1,4 @@
ir_version: 4
ir_version: 7
producer_name: "pytorch"
producer_version: "CURRENT_VERSION"
graph {
@ -122,5 +122,5 @@ graph {
}
}
opset_import {
version: 9
version: 13
}

4
test/onnx/expect/TestOperators.test_gt.expect
View File

@ -1,4 +1,4 @@
ir_version: 4
ir_version: 7
producer_name: "pytorch"
producer_version: "CURRENT_VERSION"
graph {
@ -72,5 +72,5 @@ graph {
}
}
opset_import {
version: 9
version: 13
}

48
test/onnx/expect/TestOperators.test_hardtanh.expect
View File

@ -1,22 +1,40 @@
ir_version: 4
ir_version: 7
producer_name: "pytorch"
producer_version: "CURRENT_VERSION"
graph {
node {
output: "onnx::Clip_1"
name: "Constant_0"
op_type: "Constant"
attribute {
name: "value"
t {
data_type: 1
raw_data: "\000\000\000\277"
}
type: TENSOR
}
}
node {
output: "onnx::Clip_2"
name: "Constant_1"
op_type: "Constant"
attribute {
name: "value"
t {
data_type: 1
raw_data: "\000\000\000?"
}
type: TENSOR
}
}
node {
input: "input"
output: "1"
name: "Clip_0"
input: "onnx::Clip_1"
input: "onnx::Clip_2"
output: "3"
name: "Clip_2"
op_type: "Clip"
attribute {
name: "max"
f: 0.5
type: FLOAT
}
attribute {
name: "min"
f: -0.5
type: FLOAT
}
}
name: "torch_jit"
input {
@ -36,7 +54,7 @@ graph {
}
}
output {
name: "1"
name: "3"
type {
tensor_type {
elem_type: 1
@ -53,5 +71,5 @@ graph {
}
}
opset_import {
version: 9
version: 13
}

4
test/onnx/expect/TestOperators.test_implicit_expand.expect
View File

@ -1,4 +1,4 @@
ir_version: 4
ir_version: 7
producer_name: "pytorch"
producer_version: "CURRENT_VERSION"
graph {
@ -57,5 +57,5 @@ graph {
}
}
opset_import {
version: 9
version: 13
}

4
test/onnx/expect/TestOperators.test_index.expect
View File

@ -1,4 +1,4 @@
ir_version: 4
ir_version: 7
producer_name: "pytorch"
producer_version: "CURRENT_VERSION"
graph {
@ -59,5 +59,5 @@ graph {
}
}
opset_import {
version: 9
version: 13
}

4
test/onnx/expect/TestOperators.test_isnan.expect
View File

@ -1,4 +1,4 @@
ir_version: 4
ir_version: 7
producer_name: "pytorch"
producer_version: "CURRENT_VERSION"
graph {
@ -37,5 +37,5 @@ graph {
}
}
opset_import {
version: 9
version: 13
}

4
test/onnx/expect/TestOperators.test_layer_norm_aten.expect
View File

@ -1,4 +1,4 @@
ir_version: 4
ir_version: 7
producer_name: "pytorch"
producer_version: "CURRENT_VERSION"
graph {
@ -128,5 +128,5 @@ graph {
}
}
opset_import {
version: 9
version: 13
}

26
test/onnx/expect/TestOperators.test_le.expect
View File

@ -1,23 +1,17 @@
ir_version: 4
ir_version: 7
producer_name: "pytorch"
producer_version: "CURRENT_VERSION"
graph {
node {
input: "onnx::Greater_0"
input: "onnx::Greater_1"
output: "onnx::Not_2"
name: "Greater_0"
op_type: "Greater"
}
node {
input: "onnx::Not_2"
output: "3"
name: "Not_1"
op_type: "Not"
input: "onnx::LessOrEqual_0"
input: "onnx::LessOrEqual_1"
output: "2"
name: "LessOrEqual_0"
op_type: "LessOrEqual"
}
name: "torch_jit"
input {
name: "onnx::Greater_0"
name: "onnx::LessOrEqual_0"
type {
tensor_type {
elem_type: 6
@ -33,7 +27,7 @@ graph {
}
}
input {
name: "onnx::Greater_1"
name: "onnx::LessOrEqual_1"
type {
tensor_type {
elem_type: 6
@ -49,7 +43,7 @@ graph {
}
}
output {
name: "3"
name: "2"
type {
tensor_type {
elem_type: 9
@ -66,5 +60,5 @@ graph {
}
}
opset_import {
version: 9
version: 13
}

4
test/onnx/expect/TestOperators.test_linear.expect
View File

@ -1,4 +1,4 @@
ir_version: 4
ir_version: 7
producer_name: "pytorch"
producer_version: "CURRENT_VERSION"
graph {
@ -102,5 +102,5 @@ graph {
}
}
opset_import {
version: 9
version: 13
}

4
test/onnx/expect/TestOperators.test_log_sigmoid.expect
View File

@ -1,4 +1,4 @@
ir_version: 4
ir_version: 7
producer_name: "pytorch"
producer_version: "CURRENT_VERSION"
graph {
@ -61,5 +61,5 @@ graph {
}
}
opset_import {
version: 9
version: 13
}

4
test/onnx/expect/TestOperators.test_logsoftmax.expect
View File

@ -1,4 +1,4 @@
ir_version: 4
ir_version: 7
producer_name: "pytorch"
producer_version: "CURRENT_VERSION"
graph {
@ -60,5 +60,5 @@ graph {
}
}
opset_import {
version: 9
version: 13
}

4
test/onnx/expect/TestOperators.test_lt.expect
View File

@ -1,4 +1,4 @@
ir_version: 4
ir_version: 7
producer_name: "pytorch"
producer_version: "CURRENT_VERSION"
graph {
@ -72,5 +72,5 @@ graph {
}
}
opset_import {
version: 9
version: 13
}

4
test/onnx/expect/TestOperators.test_max.expect
View File

@ -1,4 +1,4 @@
ir_version: 4
ir_version: 7
producer_name: "pytorch"
producer_version: "CURRENT_VERSION"
graph {
@ -60,5 +60,5 @@ graph {
}
}
opset_import {
version: 9
version: 13
}

9
test/onnx/expect/TestOperators.test_maxpool.expect
View File

@ -1,4 +1,4 @@
ir_version: 4
ir_version: 7
producer_name: "pytorch"
producer_version: "CURRENT_VERSION"
graph {
@ -7,6 +7,11 @@ graph {
output: "1"
name: "MaxPool_0"
op_type: "MaxPool"
attribute {
name: "ceil_mode"
i: 0
type: INT
}
attribute {
name: "kernel_shape"
ints: 3
@ -65,5 +70,5 @@ graph {
}
}
opset_import {
version: 9
version: 13
}

81
test/onnx/expect/TestOperators.test_maxpool_indices.expect
View File

@ -1,4 +1,4 @@
ir_version: 4
ir_version: 7
producer_name: "pytorch"
producer_version: "CURRENT_VERSION"
graph {
@ -8,6 +8,11 @@ graph {
output: "onnx::Sub_2"
name: "MaxPool_0"
op_type: "MaxPool"
attribute {
name: "ceil_mode"
i: 0
type: INT
}
attribute {
name: "kernel_shape"
ints: 3
@ -43,31 +48,61 @@ graph {
}
}
node {
input: "onnx::Slice_4"
output: "onnx::Sub_5"
name: "Slice_2"
op_type: "Slice"
output: "onnx::Slice_5"
name: "Constant_2"
op_type: "Constant"
attribute {
name: "axes"
ints: 2
type: INTS
}
attribute {
name: "ends"
ints: 1
type: INTS
}
attribute {
name: "starts"
ints: 0
type: INTS
name: "value"
t {
dims: 1
data_type: 7
raw_data: "\002\000\000\000\000\000\000\000"
}
type: TENSOR
}
}
node {
output: "onnx::Slice_6"
name: "Constant_3"
op_type: "Constant"
attribute {
name: "value"
t {
dims: 1
data_type: 7
raw_data: "\000\000\000\000\000\000\000\000"
}
type: TENSOR
}
}
node {
output: "onnx::Slice_7"
name: "Constant_4"
op_type: "Constant"
attribute {
name: "value"
t {
dims: 1
data_type: 7
raw_data: "\001\000\000\000\000\000\000\000"
}
type: TENSOR
}
}
node {
input: "onnx::Slice_4"
input: "onnx::Slice_6"
input: "onnx::Slice_7"
input: "onnx::Slice_5"
output: "onnx::Sub_8"
name: "Slice_5"
op_type: "Slice"
}
node {
input: "onnx::Sub_2"
input: "onnx::Sub_5"
output: "6"
name: "Sub_3"
input: "onnx::Sub_8"
output: "9"
name: "Sub_6"
op_type: "Sub"
}
name: "torch_jit"
@ -110,7 +145,7 @@ graph {
}
}
output {
name: "6"
name: "9"
type {
tensor_type {
elem_type: 7
@ -130,5 +165,5 @@ graph {
}
}
opset_import {
version: 9
version: 13
}

4
test/onnx/expect/TestOperators.test_mean.expect
View File

@ -1,4 +1,4 @@
ir_version: 4
ir_version: 7
producer_name: "pytorch"
producer_version: "CURRENT_VERSION"
graph {
@ -48,5 +48,5 @@ graph {
}
}
opset_import {
version: 9
version: 13
}

4
test/onnx/expect/TestOperators.test_mean_dtype.expect
View File

@ -1,4 +1,4 @@
ir_version: 4
ir_version: 7
producer_name: "pytorch"
producer_version: "CURRENT_VERSION"
graph {
@ -59,5 +59,5 @@ graph {
}
}
opset_import {
version: 9
version: 13
}

4
test/onnx/expect/TestOperators.test_meshgrid.expect
View File

@ -1,4 +1,4 @@
ir_version: 4
ir_version: 7
producer_name: "pytorch"
producer_version: "CURRENT_VERSION"
graph {
@ -318,5 +318,5 @@ graph {
}
}
opset_import {
version: 9
version: 13
}

4
test/onnx/expect/TestOperators.test_min.expect
View File

@ -1,4 +1,4 @@
ir_version: 4
ir_version: 7
producer_name: "pytorch"
producer_version: "CURRENT_VERSION"
graph {
@ -60,5 +60,5 @@ graph {
}
}
opset_import {
version: 9
version: 13
}

25
test/onnx/expect/TestOperators.test_mm.expect
View File

@ -1,27 +1,12 @@
ir_version: 4
ir_version: 7
producer_name: "pytorch"
producer_version: "CURRENT_VERSION"
graph {
node {
output: "onnx::Gemm_2"
name: "Constant_0"
op_type: "Constant"
attribute {
name: "value"
t {
dims: 1
data_type: 1
raw_data: "\000\000\200?"
}
type: TENSOR
}
}
node {
input: "onnx::Gemm_0"
input: "onnx::Gemm_1"
input: "onnx::Gemm_2"
output: "3"
name: "Gemm_1"
output: "2"
name: "Gemm_0"
op_type: "Gemm"
attribute {
name: "alpha"
@ -68,7 +53,7 @@ graph {
}
}
output {
name: "3"
name: "2"
type {
tensor_type {
elem_type: 1
@ -85,5 +70,5 @@ graph {
}
}
opset_import {
version: 9
version: 13
}

44
test/onnx/expect/TestOperators.test_narrow.expect
View File

@ -1,29 +1,35 @@
ir_version: 4
ir_version: 7
producer_name: "pytorch"
producer_version: "CURRENT_VERSION"
graph {
node {
input: "onnx::Slice_0"
output: "1"
input: "onnx::Slice_14"
input: "onnx::Slice_15"
input: "onnx::Slice_16"
output: "12"
name: "Slice_0"
op_type: "Slice"
attribute {
name: "axes"
ints: 0
type: INTS
}
attribute {
name: "ends"
ints: 2
type: INTS
}
attribute {
name: "starts"
ints: 0
type: INTS
}
}
name: "torch_jit"
initializer {
dims: 1
data_type: 7
name: "onnx::Slice_14"
raw_data: "\000\000\000\000\000\000\000\000"
}
initializer {
dims: 1
data_type: 7
name: "onnx::Slice_15"
raw_data: "\002\000\000\000\000\000\000\000"
}
initializer {
dims: 1
data_type: 7
name: "onnx::Slice_16"
raw_data: "\000\000\000\000\000\000\000\000"
}
input {
name: "onnx::Slice_0"
type {
@ -41,7 +47,7 @@ graph {
}
}
output {
name: "1"
name: "12"
type {
tensor_type {
elem_type: 1
@ -58,5 +64,5 @@ graph {
}
}
opset_import {
version: 9
version: 13
}

4
test/onnx/expect/TestOperators.test_ne.expect
View File

@ -1,4 +1,4 @@
ir_version: 4
ir_version: 7
producer_name: "pytorch"
producer_version: "CURRENT_VERSION"
graph {
@ -78,5 +78,5 @@ graph {
}
}
opset_import {
version: 9
version: 13
}

4
test/onnx/expect/TestOperators.test_nonzero.expect
View File

@ -1,4 +1,4 @@
ir_version: 4
ir_version: 7
producer_name: "pytorch"
producer_version: "CURRENT_VERSION"
graph {
@ -58,5 +58,5 @@ graph {
}
}
opset_import {
version: 9
version: 13
}

4
test/onnx/expect/TestOperators.test_norm_p1.expect
View File

@ -1,4 +1,4 @@
ir_version: 4
ir_version: 7
producer_name: "pytorch"
producer_version: "CURRENT_VERSION"
graph {
@ -62,5 +62,5 @@ graph {
}
}
opset_import {
version: 9
version: 13
}

4
test/onnx/expect/TestOperators.test_norm_p2.expect
View File

@ -1,4 +1,4 @@
ir_version: 4
ir_version: 7
producer_name: "pytorch"
producer_version: "CURRENT_VERSION"
graph {
@ -62,5 +62,5 @@ graph {
}
}
opset_import {
version: 9
version: 13
}

4
test/onnx/expect/TestOperators.test_ones_like.expect
View File

@ -1,4 +1,4 @@
ir_version: 4
ir_version: 7
producer_name: "pytorch"
producer_version: "CURRENT_VERSION"
graph {
@ -36,5 +36,5 @@ graph {
}
}
opset_import {
version: 9
version: 13
}

201
test/onnx/expect/TestOperators.test_pad.expect
View File

@ -1,31 +1,190 @@
ir_version: 4
ir_version: 7
producer_name: "pytorch"
producer_version: "CURRENT_VERSION"
graph {
node {
input: "onnx::ConstantOfShape_27"
output: "onnx::Concat_10"
name: "ConstantOfShape_0"
op_type: "ConstantOfShape"
attribute {
name: "value"
t {
dims: 1
data_type: 7
raw_data: "\000\000\000\000\000\000\000\000"
}
type: TENSOR
}
}
node {
input: "onnx::Concat_28"
input: "onnx::Concat_10"
output: "onnx::Reshape_11"
name: "Concat_1"
op_type: "Concat"
attribute {
name: "axis"
i: 0
type: INT
}
}
node {
output: "onnx::Reshape_12"
name: "Constant_2"
op_type: "Constant"
attribute {
name: "value"
t {
dims: 2
data_type: 7
raw_data: "\377\377\377\377\377\377\377\377\002\000\000\000\000\000\000\000"
}
type: TENSOR
}
}
node {
input: "onnx::Reshape_11"
input: "onnx::Reshape_12"
output: "onnx::Slice_13"
name: "Reshape_3"
op_type: "Reshape"
}
node {
output: "onnx::Slice_14"
name: "Constant_4"
op_type: "Constant"
attribute {
name: "value"
t {
dims: 1
data_type: 7
raw_data: "\000\000\000\000\000\000\000\000"
}
type: TENSOR
}
}
node {
output: "onnx::Slice_15"
name: "Constant_5"
op_type: "Constant"
attribute {
name: "value"
t {
dims: 1
data_type: 7
raw_data: "\377\377\377\377\377\377\377\377"
}
type: TENSOR
}
}
node {
output: "onnx::Slice_16"
name: "Constant_6"
op_type: "Constant"
attribute {
name: "value"
t {
dims: 1
data_type: 7
raw_data: "\001\000\000\000\000\000\000\200"
}
type: TENSOR
}
}
node {
output: "onnx::Slice_17"
name: "Constant_7"
op_type: "Constant"
attribute {
name: "value"
t {
dims: 1
data_type: 7
raw_data: "\377\377\377\377\377\377\377\377"
}
type: TENSOR
}
}
node {
input: "onnx::Slice_13"
input: "onnx::Slice_15"
input: "onnx::Slice_16"
input: "onnx::Slice_14"
input: "onnx::Slice_17"
output: "onnx::Transpose_18"
name: "Slice_8"
op_type: "Slice"
}
node {
input: "onnx::Transpose_18"
output: "onnx::Reshape_19"
name: "Transpose_9"
op_type: "Transpose"
attribute {
name: "perm"
ints: 1
ints: 0
type: INTS
}
}
node {
output: "onnx::Reshape_20"
name: "Constant_10"
op_type: "Constant"
attribute {
name: "value"
t {
dims: 1
data_type: 7
raw_data: "\377\377\377\377\377\377\377\377"
}
type: TENSOR
}
}
node {
input: "onnx::Reshape_19"
input: "onnx::Reshape_20"
output: "onnx::Cast_21"
name: "Reshape_11"
op_type: "Reshape"
}
node {
input: "onnx::Cast_21"
output: "onnx::Pad_22"
name: "Cast_12"
op_type: "Cast"
attribute {
name: "to"
i: 7
type: INT
}
}
node {
input: "input"
output: "1"
name: "Pad_0"
input: "onnx::Pad_22"
output: "23"
name: "Pad_13"
op_type: "Pad"
attribute {
name: "mode"
s: "reflect"
type: STRING
}
attribute {
name: "pads"
ints: 0
ints: 0
ints: 0
ints: 2
ints: 0
ints: 0
ints: 1
ints: 3
type: INTS
}
}
name: "torch_jit"
initializer {
dims: 1
data_type: 7
name: "onnx::ConstantOfShape_27"
raw_data: "\004\000\000\000\000\000\000\000"
}
initializer {
dims: 4
data_type: 7
name: "onnx::Concat_28"
raw_data: "\002\000\000\000\000\000\000\000\003\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\001\000\000\000\000\000\000\000"
}
input {
name: "input"
type {
@ -49,22 +208,22 @@ graph {
}
}
output {
name: "1"
name: "23"
type {
tensor_type {
elem_type: 1
shape {
dim {
dim_value: 1
dim_param: "Pad23_dim_0"
}
dim {
dim_value: 1
dim_param: "Pad23_dim_1"
}
dim {
dim_value: 3
dim_param: "Pad23_dim_2"
}
dim {
dim_value: 9
dim_param: "Pad23_dim_3"
}
}
}
@ -72,5 +231,5 @@ graph {
}
}
opset_import {
version: 9
version: 13
}

4
test/onnx/expect/TestOperators.test_params.expect
View File

@ -1,4 +1,4 @@
ir_version: 4
ir_version: 7
producer_name: "pytorch"
producer_version: "CURRENT_VERSION"
graph {
@ -92,5 +92,5 @@ graph {
}
}
opset_import {
version: 9
version: 13
}

4
test/onnx/expect/TestOperators.test_params_onnx_irv4.expect
View File

@ -1,4 +1,4 @@
ir_version: 4
ir_version: 7
producer_name: "pytorch"
producer_version: "CURRENT_VERSION"
graph {
@ -76,5 +76,5 @@ graph {
}
}
opset_import {
version: 9
version: 13
}

4
test/onnx/expect/TestOperators.test_permute2.expect
View File

@ -1,4 +1,4 @@
ir_version: 4
ir_version: 7
producer_name: "pytorch"
producer_version: "CURRENT_VERSION"
graph {
@ -77,5 +77,5 @@ graph {
}
}
opset_import {
version: 9
version: 13
}

4
test/onnx/expect/TestOperators.test_pow.expect
View File

@ -1,4 +1,4 @@
ir_version: 4
ir_version: 7
producer_name: "pytorch"
producer_version: "CURRENT_VERSION"
graph {
@ -78,5 +78,5 @@ graph {
}
}
opset_import {
version: 9
version: 13
}

14
test/onnx/expect/TestOperators.test_prelu.expect
View File

@ -1,11 +1,11 @@
ir_version: 4
ir_version: 7
producer_name: "pytorch"
producer_version: "CURRENT_VERSION"
graph {
node {
input: "onnx::PRelu_0"
input: "onnx::PRelu_4"
output: "3"
input: "onnx::PRelu_5"
output: "4"
name: "PRelu_0"
op_type: "PRelu"
}
@ -15,7 +15,7 @@ graph {
dims: 1
dims: 1
data_type: 1
name: "onnx::PRelu_4"
name: "onnx::PRelu_5"
raw_data: "\000\000\200>\000\000\200>"
}
input {
@ -41,7 +41,7 @@ graph {
}
}
input {
name: "onnx::PRelu_4"
name: "onnx::PRelu_5"
type {
tensor_type {
elem_type: 1
@ -60,7 +60,7 @@ graph {
}
}
output {
name: "3"
name: "4"
type {
tensor_type {
elem_type: 1
@ -83,5 +83,5 @@ graph {
}
}
opset_import {
version: 9
version: 13
}

4
test/onnx/expect/TestOperators.test_prod.expect
View File

@ -1,4 +1,4 @@
ir_version: 4
ir_version: 7
producer_name: "pytorch"
producer_version: "CURRENT_VERSION"
graph {
@ -48,5 +48,5 @@ graph {
}
}
opset_import {
version: 9
version: 13
}

4
test/onnx/expect/TestOperators.test_prod_dtype.expect
View File

@ -1,4 +1,4 @@
ir_version: 4
ir_version: 7
producer_name: "pytorch"
producer_version: "CURRENT_VERSION"
graph {
@ -59,5 +59,5 @@ graph {
}
}
opset_import {
version: 9
version: 13
}

4
test/onnx/expect/TestOperators.test_rand.expect
View File

@ -1,4 +1,4 @@
ir_version: 4
ir_version: 7
producer_name: "pytorch"
producer_version: "CURRENT_VERSION"
graph {
@ -69,5 +69,5 @@ graph {
}
}
opset_import {
version: 9
version: 13
}

4
test/onnx/expect/TestOperators.test_randn.expect
View File

@ -1,4 +1,4 @@
ir_version: 4
ir_version: 7
producer_name: "pytorch"
producer_version: "CURRENT_VERSION"
graph {
@ -69,5 +69,5 @@ graph {
}
}
opset_import {
version: 9
version: 13
}

30
test/onnx/expect/TestOperators.test_reduce_sum_negative_indices.expect
View File

@ -1,17 +1,27 @@
ir_version: 4
ir_version: 7
producer_name: "pytorch"
producer_version: "CURRENT_VERSION"
graph {
node {
input: "onnx::ReduceSum_0"
output: "1"
name: "ReduceSum_0"
op_type: "ReduceSum"
output: "onnx::ReduceSum_1"
name: "Constant_0"
op_type: "Constant"
attribute {
name: "axes"
ints: -1
type: INTS
name: "value"
t {
dims: 1
data_type: 7
raw_data: "\377\377\377\377\377\377\377\377"
}
type: TENSOR
}
}
node {
input: "onnx::ReduceSum_0"
input: "onnx::ReduceSum_1"
output: "2"
name: "ReduceSum_1"
op_type: "ReduceSum"
attribute {
name: "keepdims"
i: 0
@ -36,7 +46,7 @@ graph {
}
}
output {
name: "1"
name: "2"
type {
tensor_type {
elem_type: 1
@ -50,5 +60,5 @@ graph {
}
}
opset_import {
version: 9
version: 13
}

4
test/onnx/expect/TestOperators.test_reduced_mean.expect
View File

@ -1,4 +1,4 @@
ir_version: 4
ir_version: 7
producer_name: "pytorch"
producer_version: "CURRENT_VERSION"
graph {
@ -62,5 +62,5 @@ graph {
}
}
opset_import {
version: 9
version: 13
}

4
test/onnx/expect/TestOperators.test_reduced_mean_dtype.expect
View File

@ -1,4 +1,4 @@
ir_version: 4
ir_version: 7
producer_name: "pytorch"
producer_version: "CURRENT_VERSION"
graph {
@ -73,5 +73,5 @@ graph {
}
}
opset_import {
version: 9
version: 13
}

4
test/onnx/expect/TestOperators.test_reduced_mean_keepdim.expect
View File

@ -1,4 +1,4 @@
ir_version: 4
ir_version: 7
producer_name: "pytorch"
producer_version: "CURRENT_VERSION"
graph {
@ -66,5 +66,5 @@ graph {
}
}
opset_import {
version: 9
version: 13
}

4
test/onnx/expect/TestOperators.test_reduced_prod.expect
View File

@ -1,4 +1,4 @@
ir_version: 4
ir_version: 7
producer_name: "pytorch"
producer_version: "CURRENT_VERSION"
graph {
@ -62,5 +62,5 @@ graph {
}
}
opset_import {
version: 9
version: 13
}

4
test/onnx/expect/TestOperators.test_reduced_prod_dtype.expect
View File

@ -1,4 +1,4 @@
ir_version: 4
ir_version: 7
producer_name: "pytorch"
producer_version: "CURRENT_VERSION"
graph {
@ -73,5 +73,5 @@ graph {
}
}
opset_import {
version: 9
version: 13
}

4
test/onnx/expect/TestOperators.test_reduced_prod_keepdim.expect
View File

@ -1,4 +1,4 @@
ir_version: 4
ir_version: 7
producer_name: "pytorch"
producer_version: "CURRENT_VERSION"
graph {
@ -65,5 +65,5 @@ graph {
}
}
opset_import {
version: 9
version: 13
}

31
test/onnx/expect/TestOperators.test_reduced_sum.expect
View File

@ -1,18 +1,27 @@
ir_version: 4
ir_version: 7
producer_name: "pytorch"
producer_version: "CURRENT_VERSION"
graph {
node {
input: "onnx::ReduceSum_0"
output: "1"
name: "ReduceSum_0"
op_type: "ReduceSum"
output: "onnx::ReduceSum_1"
name: "Constant_0"
op_type: "Constant"
attribute {
name: "axes"
ints: 1
ints: 2
type: INTS
name: "value"
t {
dims: 2
data_type: 7
raw_data: "\001\000\000\000\000\000\000\000\002\000\000\000\000\000\000\000"
}
type: TENSOR
}
}
node {
input: "onnx::ReduceSum_0"
input: "onnx::ReduceSum_1"
output: "2"
name: "ReduceSum_1"
op_type: "ReduceSum"
attribute {
name: "keepdims"
i: 0
@ -43,7 +52,7 @@ graph {
}
}
output {
name: "1"
name: "2"
type {
tensor_type {
elem_type: 1
@ -60,5 +69,5 @@ graph {
}
}
opset_import {
version: 9
version: 13
}

34
test/onnx/expect/TestOperators.test_reduced_sum_dtype.expect
View File

@ -1,11 +1,25 @@
ir_version: 4
ir_version: 7
producer_name: "pytorch"
producer_version: "CURRENT_VERSION"
graph {
node {
input: "onnx::Cast_0"
output: "onnx::ReduceSum_1"
name: "Cast_0"
name: "Constant_0"
op_type: "Constant"
attribute {
name: "value"
t {
dims: 1
data_type: 7
raw_data: "\000\000\000\000\000\000\000\000"
}
type: TENSOR
}
}
node {
input: "onnx::Cast_0"
output: "onnx::ReduceSum_2"
name: "Cast_1"
op_type: "Cast"
attribute {
name: "to"
@ -14,15 +28,11 @@ graph {
}
}
node {
input: "onnx::ReduceSum_2"
input: "onnx::ReduceSum_1"
output: "2"
name: "ReduceSum_1"
output: "3"
name: "ReduceSum_2"
op_type: "ReduceSum"
attribute {
name: "axes"
ints: 0
type: INTS
}
attribute {
name: "keepdims"
i: 0
@ -53,7 +63,7 @@ graph {
}
}
output {
name: "2"
name: "3"
type {
tensor_type {
elem_type: 11
@ -73,5 +83,5 @@ graph {
}
}
opset_import {
version: 9
version: 13
}

30
test/onnx/expect/TestOperators.test_reduced_sum_keepdim.expect
View File

@ -1,17 +1,27 @@
ir_version: 4
ir_version: 7
producer_name: "pytorch"
producer_version: "CURRENT_VERSION"
graph {
node {
input: "onnx::ReduceSum_0"
output: "1"
name: "ReduceSum_0"
op_type: "ReduceSum"
output: "onnx::ReduceSum_1"
name: "Constant_0"
op_type: "Constant"
attribute {
name: "axes"
ints: 2
type: INTS
name: "value"
t {
dims: 1
data_type: 7
raw_data: "\002\000\000\000\000\000\000\000"
}
type: TENSOR
}
}
node {
input: "onnx::ReduceSum_0"
input: "onnx::ReduceSum_1"
output: "2"
name: "ReduceSum_1"
op_type: "ReduceSum"
attribute {
name: "keepdims"
i: 1
@ -42,7 +52,7 @@ graph {
}
}
output {
name: "1"
name: "2"
type {
tensor_type {
elem_type: 1
@ -65,5 +75,5 @@ graph {
}
}
opset_import {
version: 9
version: 13
}

4
test/onnx/expect/TestOperators.test_reducemax.expect
View File

@ -1,4 +1,4 @@
ir_version: 4
ir_version: 7
producer_name: "pytorch"
producer_version: "CURRENT_VERSION"
graph {
@ -48,5 +48,5 @@ graph {
}
}
opset_import {
version: 9
version: 13
}

4
test/onnx/expect/TestOperators.test_reducemin.expect
View File

@ -1,4 +1,4 @@
ir_version: 4
ir_version: 7
producer_name: "pytorch"
producer_version: "CURRENT_VERSION"
graph {
@ -48,5 +48,5 @@ graph {
}
}
opset_import {
version: 9
version: 13
}

4
test/onnx/expect/TestOperators.test_remainder.expect
View File

@ -1,4 +1,4 @@
ir_version: 4
ir_version: 7
producer_name: "pytorch"
producer_version: "CURRENT_VERSION"
graph {
@ -89,5 +89,5 @@ graph {
}
}
opset_import {
version: 9
version: 13
}

Some files were not shown because too many files have changed in this diff Show More