From b2c4f023e82b5c8f4755f191fe22f860e60b4f60 Mon Sep 17 00:00:00 2001
From: Sergey Slashchinin <sergei.slashchinin@xperience.ai>
Date: Wed, 9 Sep 2020 01:04:18 +0300
Subject: [PATCH] Add testdata for Gather with multiple outputs

---
 .../onnx/data/input_gather_multi_output.npy    | Bin 0 -> 144 bytes
 .../onnx/data/output_gather_multi_output.npy   | Bin 0 -> 136 bytes
 testdata/dnn/onnx/generate_onnx_models.py      |  17 +++++++++++++++++
 .../dnn/onnx/models/gather_multi_output.onnx   | Bin 0 -> 432 bytes
 4 files changed, 17 insertions(+)
 create mode 100644 testdata/dnn/onnx/data/input_gather_multi_output.npy
 create mode 100644 testdata/dnn/onnx/data/output_gather_multi_output.npy
 create mode 100644 testdata/dnn/onnx/models/gather_multi_output.onnx

diff --git a/testdata/dnn/onnx/data/input_gather_multi_output.npy b/testdata/dnn/onnx/data/input_gather_multi_output.npy
new file mode 100644
index 0000000000000000000000000000000000000000..b88e81dd366f6535afacf799b5777147fa646ff4
GIT binary patch
literal 144
zcmbR27wQ`j$;eQ~P_3SlTAW;@Zl$1ZlV+l>qoAIaUsO_*m=~X4l#&V(cT3DEP6dh=
bXCxM+0{I$-ItoTWq^YA&t3V#$Vn71`6m%Pi

literal 0
HcmV?d00001

diff --git a/testdata/dnn/onnx/data/output_gather_multi_output.npy b/testdata/dnn/onnx/data/output_gather_multi_output.npy
new file mode 100644
index 0000000000000000000000000000000000000000..b92aa732b984c7ed4006655e52f6ed3f456ace35
GIT binary patch
literal 136
zcmbR27wQ`j$;eQ~P_3SlTAW;@Zl$1ZlV+l>qoAIaUsO_*m=~X4l#&V(cT3DEP6dh=
aXCxM+0{I$-ItoUbItsN4WCJb+C;$L4O&e+e

literal 0
HcmV?d00001

diff --git a/testdata/dnn/onnx/generate_onnx_models.py b/testdata/dnn/onnx/generate_onnx_models.py
index 4458f3ea6..9b6f45604 100644
--- a/testdata/dnn/onnx/generate_onnx_models.py
+++ b/testdata/dnn/onnx/generate_onnx_models.py
@@ -1023,3 +1023,20 @@ def forward(self, x, kernel, bias):
 bias = Variable(torch.randn(4))
 model = Conv1dBias()
 save_data_and_model_multy_inputs("conv1d_variable_wb", model, x, kernel, bias)
+
+class GatherMultiOutput(nn.Module):
+    def __init__(self, in_dim = 2):
+        super(GatherMultiOutput, self).__init__()
+        self.in_dim = in_dim
+        self.lin_inp = nn.Linear(in_dim, 2, bias=False)
+    def forward(self, x):
+        x_projected = self.lin_inp(x).long()
+        x_gather = x_projected[:,0,:]
+        x_float1 = x_gather.float()
+        x_float2 = x_gather.float()
+        x_float3 = x_gather.float()
+        return x_float1+x_float2+x_float3
+
+x = Variable(torch.zeros([1, 2, 2]))
+model = GatherMultiOutput()
+save_data_and_model("gather_multi_output", x, model)
diff --git a/testdata/dnn/onnx/models/gather_multi_output.onnx b/testdata/dnn/onnx/models/gather_multi_output.onnx
new file mode 100644
index 0000000000000000000000000000000000000000..1b9195a43b36d22b9160e807de6f82ba06c6d3c4
GIT binary patch
literal 432
zcmZ{g!AiqG5Qd#ilWfNzVJRA-qU0ojO14dt)PrDq5DI+(FCp4MBc&<rwzQ8Tc<2jw
z@a&=Bb9r#nrl5!e!~F35%nv`+m_AvFeEKxB6@DGwyeN2yYBrwC1rpw1(6nhk5&iin
z_8e`e@B*R9z_+z-G7~X(D#dlHNJMVwYal3T%f@a#o{3~E;)X-_+O<$!B%^ui8m8W(
z2j#4xf>#JWLW@CQ)B8jW(`npvw4HJfv63vanFYmrnIA9+%Y6S={trV`8e0Du+NB}*
zYdA&-5h6yoXRCLEK^!`YJa{5o@etY)@EwS@fSKD8>(7~}r^`t`748`&l$g+wHTFzr
u_4e`pRolD<pSS0cx_LM)TV@PR3d}%4K`trwu_jYEJ3>Pex+fiCR=)vpOIZ5=

literal 0
HcmV?d00001