From be05aa02e3d4d5bdeb909592f454b6f7795718e0 Mon Sep 17 00:00:00 2001 From: Abduragim Shtanchaev Date: Wed, 7 Jun 2023 09:02:58 +0300 Subject: [PATCH] chage test case to opset = 11 --- testdata/dnn/onnx/data/input_split_0.npy | Bin 0 -> 176 bytes testdata/dnn/onnx/data/output_split_0.npy | Bin 0 -> 176 bytes testdata/dnn/onnx/generate_onnx_models.py | 11 +++++++++++ testdata/dnn/onnx/models/split_0.onnx | Bin 0 -> 286 bytes 4 files changed, 11 insertions(+) create mode 100644 testdata/dnn/onnx/data/input_split_0.npy create mode 100644 testdata/dnn/onnx/data/output_split_0.npy create mode 100644 testdata/dnn/onnx/models/split_0.onnx diff --git a/testdata/dnn/onnx/data/input_split_0.npy b/testdata/dnn/onnx/data/input_split_0.npy new file mode 100644 index 0000000000000000000000000000000000000000..5576f4a92d3ffa3912452962f53d0d6abae28017 GIT binary patch literal 176 zcmbR27wQ`j$;eQ~P_3SlTAW;@Zl$1ZlV+l>qoAIaUsO_*m=~X4l#&V(cT3DEP6dh= iXCxM+0{I$-Its=*3PwPrsiRP1p@$OIwYF_ literal 0 HcmV?d00001 diff --git a/testdata/dnn/onnx/data/output_split_0.npy b/testdata/dnn/onnx/data/output_split_0.npy new file mode 100644 index 0000000000000000000000000000000000000000..e014b04cab2dfde36bf28337fab7d062efa7b9b6 GIT binary patch literal 176 zcmbR27wQ`j$;eQ~P_3SlTAW;@Zl$1ZlV+l>qoAIaUsO_*m=~X4l#&V(cT3DEP6dh= fXCxM+0{I%oItoTWq^YA&t3V#$Vqj>nCj|olT7o2g literal 0 HcmV?d00001 diff --git a/testdata/dnn/onnx/generate_onnx_models.py b/testdata/dnn/onnx/generate_onnx_models.py index bec86026f..10704ee46 100644 --- a/testdata/dnn/onnx/generate_onnx_models.py +++ b/testdata/dnn/onnx/generate_onnx_models.py @@ -868,6 +868,15 @@ def forward(self, x): tup = torch.split(x, self.split_size_sections, self.dim) return torch.cat(tup) +class SimpleSplit(nn.Module): + def forward(self, image): + return torch.cat([img for img in image]) + + +model = SimpleSplit() +input = torch.ones((1, 3, 2, 2)) +save_data_and_model("split_0", input, model, version=11) + model = Split() input = Variable(torch.tensor([1., 2.], dtype=torch.float32)) save_data_and_model("split_1", input, model) @@ -888,6 +897,8 @@ def forward(self, x): model = Split(dim=-1, split_size_sections=[1, 2]) save_data_and_model("split_6", input2, model, version=13) + + class SplitSizes(nn.Module): def __init__(self, *args, **kwargs): super(SplitSizes, self).__init__() diff --git a/testdata/dnn/onnx/models/split_0.onnx b/testdata/dnn/onnx/models/split_0.onnx new file mode 100644 index 0000000000000000000000000000000000000000..1be2c819c113ed0162f9fd974455c3887119ae2e GIT binary patch literal 286 zcmd;J6Jjr@EXglQ&X8g?(lgLAuxe%Ea^hmm%uP&B72?+qF38C&iO(-BDJU(8H;`h3 z2q>|F#kIJ(SQ0BTizOHqFfwWJaIqGHR68&(U}WcV;X>CeB&Z);Selwzm5MNr116-z z4wZwLnOf`sG?SgngbP!X5TmgayS{UNUUFiI5*vg@afuKo*hBGInI%#3Ft3PliEuCq i32-rSFaj|%5Ho>rk~kNmv5*j#08o+#S;C1)fExg-U`9Rw literal 0 HcmV?d00001