From dc29cb0ace346b5af02ab66a1bf0c9c1f2218516 Mon Sep 17 00:00:00 2001
From: Smirnov Egor <s.e.a.98@yandex.ru>
Date: Wed, 18 Aug 2021 18:42:27 +0300
Subject: [PATCH] add tests for dynamic batch_size and axis selection in Scale
 layer

---
 .../dnn/onnx/data/input_dynamic_batch.npy     | Bin 0 -> 416 bytes
 .../dnn/onnx/data/input_scale_broadcast_0.npy | Bin 0 -> 288 bytes
 .../dnn/onnx/data/input_scale_broadcast_1.npy | Bin 0 -> 144 bytes
 .../dnn/onnx/data/input_scale_broadcast_2.npy | Bin 0 -> 160 bytes
 .../dnn/onnx/data/output_dynamic_batch.npy    | Bin 0 -> 224 bytes
 .../dnn/onnx/data/output_scale_broadcast.npy  | Bin 0 -> 288 bytes
 testdata/dnn/onnx/generate_onnx_models.py     |  26 +++++++++++++++
 testdata/dnn/onnx/models/dynamic_batch.onnx   | Bin 0 -> 340 bytes
 testdata/dnn/onnx/models/scale_broadcast.onnx |  30 ++++++++++++++++++
 9 files changed, 56 insertions(+)
 create mode 100644 testdata/dnn/onnx/data/input_dynamic_batch.npy
 create mode 100644 testdata/dnn/onnx/data/input_scale_broadcast_0.npy
 create mode 100644 testdata/dnn/onnx/data/input_scale_broadcast_1.npy
 create mode 100644 testdata/dnn/onnx/data/input_scale_broadcast_2.npy
 create mode 100644 testdata/dnn/onnx/data/output_dynamic_batch.npy
 create mode 100644 testdata/dnn/onnx/data/output_scale_broadcast.npy
 create mode 100644 testdata/dnn/onnx/models/dynamic_batch.onnx
 create mode 100644 testdata/dnn/onnx/models/scale_broadcast.onnx

diff --git a/testdata/dnn/onnx/data/input_dynamic_batch.npy b/testdata/dnn/onnx/data/input_dynamic_batch.npy
new file mode 100644
index 0000000000000000000000000000000000000000..6146df46f902c0dc531be3057d99bdb878644afc
GIT binary patch
literal 416
zcmbR27wQ`j$;eQ~P_3SlTAW;@Zl$1ZlV+l>qoAIaUsO_*m=~X4l#&V(cT3DEP6dh=
iXCxM+0{I$7Its>MWTL5~P^&-&;9_8Cuph;^!vFxRQ&D38

literal 0
HcmV?d00001

diff --git a/testdata/dnn/onnx/data/input_scale_broadcast_0.npy b/testdata/dnn/onnx/data/input_scale_broadcast_0.npy
new file mode 100644
index 0000000000000000000000000000000000000000..84e55f5f6ea30d85a179fe0a05638ff9f27980d7
GIT binary patch
literal 288
zcmbR27wQ`j$;eQ~P_3SlTAW;@Zl$1ZlV+l>qoAIaUsO_*m=~X4l#&V(cT3DEP6dh=
jXCxM+0{I$7Itqq53MM)VrkXkmwF+baE(V4M`@xI>_?<a}

literal 0
HcmV?d00001

diff --git a/testdata/dnn/onnx/data/input_scale_broadcast_1.npy b/testdata/dnn/onnx/data/input_scale_broadcast_1.npy
new file mode 100644
index 0000000000000000000000000000000000000000..b081291560f65a4c96c6647ebce17643bb4e62e2
GIT binary patch
literal 144
zcmbR27wQ`j$;eQ~P_3SlTAW;@Zl$1ZlV+l>qoAIaUsO_*m=~X4l#&V(cT3DEP6dh=
gXCxM+0{I$-ItnH_3Wl0G3bhL40WJoH276=-04u8=g#Z8m

literal 0
HcmV?d00001

diff --git a/testdata/dnn/onnx/data/input_scale_broadcast_2.npy b/testdata/dnn/onnx/data/input_scale_broadcast_2.npy
new file mode 100644
index 0000000000000000000000000000000000000000..95938facbf34bfcee44974dd4d44d6a0f43ea22b
GIT binary patch
literal 160
zcmbR27wQ`j$;eQ~P_3SlTAW;@Zl$1ZlV+l>qoAIaUsO_*m=~X4l#&V(cT3DEP6dh=
iXCxM+0{I$7Itqq53MN2OQ%9jzfegUKz|dfi4+8)+OCg^C

literal 0
HcmV?d00001

diff --git a/testdata/dnn/onnx/data/output_dynamic_batch.npy b/testdata/dnn/onnx/data/output_dynamic_batch.npy
new file mode 100644
index 0000000000000000000000000000000000000000..4a69fa17765a6306ab2dfe0f1cb86c6047931592
GIT binary patch
literal 224
zcmbR27wQ`j$;eQ~P_3SlTAW;@Zl$1ZlV+l>qoAIaUsO_*m=~X4l#&V(cT3DEP6dh=
jXCxM+0{I#yIts=*3WhogMw&VbwF+baE(V4Mds<=u7y2!c

literal 0
HcmV?d00001

diff --git a/testdata/dnn/onnx/data/output_scale_broadcast.npy b/testdata/dnn/onnx/data/output_scale_broadcast.npy
new file mode 100644
index 0000000000000000000000000000000000000000..84e55f5f6ea30d85a179fe0a05638ff9f27980d7
GIT binary patch
literal 288
zcmbR27wQ`j$;eQ~P_3SlTAW;@Zl$1ZlV+l>qoAIaUsO_*m=~X4l#&V(cT3DEP6dh=
jXCxM+0{I$7Itqq53MM)VrkXkmwF+baE(V4M`@xI>_?<a}

literal 0
HcmV?d00001

diff --git a/testdata/dnn/onnx/generate_onnx_models.py b/testdata/dnn/onnx/generate_onnx_models.py
index 6187bbe54..538df2739 100644
--- a/testdata/dnn/onnx/generate_onnx_models.py
+++ b/testdata/dnn/onnx/generate_onnx_models.py
@@ -1117,6 +1117,19 @@ def forward(self, x):
 model = Scale()
 save_data_and_model("scale", x, model)
 
+class ScaleBroadcast(nn.Module):
+    def __init__(self, *args, **kwargs):
+        super(ScaleBroadcast, self).__init__()
+
+    def forward(self, x0, x1, x2):
+        return torch.mul(torch.mul(x0, x1), x2)
+
+model = ScaleBroadcast()
+input_0 = Variable(torch.ones(2, 1, 4, 5, dtype=torch.float32))
+input_1 = Variable(torch.ones(1, 4, 1, dtype=torch.float32))
+input_2 = Variable(torch.ones(2, 1, 4, 1, dtype=torch.float32))
+save_data_and_model_multy_inputs("scale_broadcast", model, input_0, input_1, input_2)
+
 x = Variable(torch.randn(1, 3, 25))
 conv1d = nn.Conv1d(3, 2, kernel_size=3, padding=2, stride=2, dilation=2, bias=False)
 save_data_and_model("conv1d", x, conv1d)
@@ -1268,6 +1281,19 @@ def forward(self, x):
 save_data_and_model("average_pooling_dynamic_axes", input, ave_pool)
 postprocess_model("models/average_pooling_dynamic_axes.onnx", [[1, 3, 'height', 'width']])
 
+class DynamicBatch(nn.Module):
+    def __init__(self, *args, **kwargs):
+        super(DynamicBatch, self).__init__()
+        self.pool = nn.MaxPool2d(2, stride=2)
+
+    def forward(self, x):
+        return torch.cat((self.pool(x), torch.ones(2, 3, 1, 2)))
+
+model = DynamicBatch()
+input_ = Variable(torch.ones(2, 3, 3, 4, dtype=torch.float32))
+save_data_and_model("dynamic_batch", input_, model, export_params=True)
+postprocess_model("models/dynamic_batch.onnx", [['batch_size', 3, 3, 4]])
+
 x = Variable(torch.randn(1, 3, 10))
 max_pool = nn.MaxPool1d(kernel_size=(5), stride=1, padding=2, dilation=1)
 save_data_and_model("maxpooling_1d", x, max_pool)
diff --git a/testdata/dnn/onnx/models/dynamic_batch.onnx b/testdata/dnn/onnx/models/dynamic_batch.onnx
new file mode 100644
index 0000000000000000000000000000000000000000..c2a3b49ae96d227b0f666415192d6d0625a010fa
GIT binary patch
literal 340
zcmb79K}y6x5KPaGo7r{`aZr&FL;`{gtR&f!A{g`{g5b%6moS?hNHioNGb_=PPsvC8
zn~57AAWhLlby0M&DR=7+TQzAWCgH{9)i=Rsgn&aRW{;|Us;VLhe0QYs0OxPIDRq&U
zOw~FjG3mgSdq``wHnFpvLGsWww#nDpj3!JuL;~-s(pXj6B=p%Im-o2yu8NJ8*Nm_U
z14cY}2ps3@=0E6w;yFTy5OG+FAwE?$iTvr7$z7zX&CQY10pZ?n>&0ts7rL#h#=e{l
r#ZU2s0T(#uxKcLF5|e*w?7td8SH}oTeu(?M&Fe)v9-!akmUy!t?vqD>

literal 0
HcmV?d00001

diff --git a/testdata/dnn/onnx/models/scale_broadcast.onnx b/testdata/dnn/onnx/models/scale_broadcast.onnx
new file mode 100644
index 000000000..c45a4f5fa
--- /dev/null
+++ b/testdata/dnn/onnx/models/scale_broadcast.onnx
@@ -0,0 +1,30 @@
+pytorch1.9:°

+
+
0
+
1
3Mul_0"Mul
+
+
3
+
2
4Mul_1"Multorch-jit-exportZ
+
0
+

+
+

+
+Z
+
1
+

+

+
+
Z
+
2
+

+
+

+
+
b
+
4
+

+
+

+
+B	
\ No newline at end of file