From d868dfb623108e0c650d1ccde4624c34e3f471bf Mon Sep 17 00:00:00 2001
From: Liubov Batanina <piccione-mail@yandex.ru>
Date: Tue, 17 Nov 2020 14:43:39 +0300
Subject: [PATCH] Added testdata for ReduceMax

---
 .../dnn/onnx/data/input_reduce_max_axis_0.npy | Bin 0 -> 176 bytes
 .../dnn/onnx/data/input_reduce_max_axis_1.npy | Bin 0 -> 176 bytes
 .../onnx/data/output_reduce_max_axis_0.npy    | Bin 0 -> 176 bytes
 .../onnx/data/output_reduce_max_axis_1.npy    | Bin 0 -> 144 bytes
 testdata/dnn/onnx/generate_onnx_models.py     |  22 ++++++++++++++++--
 .../dnn/onnx/models/reduce_max_axis_0.onnx    | Bin 0 -> 157 bytes
 .../dnn/onnx/models/reduce_max_axis_1.onnx    | Bin 0 -> 157 bytes
 7 files changed, 20 insertions(+), 2 deletions(-)
 create mode 100644 testdata/dnn/onnx/data/input_reduce_max_axis_0.npy
 create mode 100644 testdata/dnn/onnx/data/input_reduce_max_axis_1.npy
 create mode 100644 testdata/dnn/onnx/data/output_reduce_max_axis_0.npy
 create mode 100644 testdata/dnn/onnx/data/output_reduce_max_axis_1.npy
 create mode 100644 testdata/dnn/onnx/models/reduce_max_axis_0.onnx
 create mode 100644 testdata/dnn/onnx/models/reduce_max_axis_1.onnx

diff --git a/testdata/dnn/onnx/data/input_reduce_max_axis_0.npy b/testdata/dnn/onnx/data/input_reduce_max_axis_0.npy
new file mode 100644
index 0000000000000000000000000000000000000000..9469ba8fb20a9fe27527df84dac8a61c6d6ac300
GIT binary patch
literal 176
zcmbR27wQ`j$;eQ~P_3SlTAW;@Zl$1ZlV+l>qoAIaUsO_*m=~X4l#&V(cT3DEP6dh=
zXCxM+0{I$-Its=*3PwPrsiRP<KnCE7u|H}bU^8u>XCe22Z7qWK3omx<*Ja(jzpDA1
a-P4XRdlyBE{oKvx_DSrO*nceD$_@aeO)+Kw

literal 0
HcmV?d00001

diff --git a/testdata/dnn/onnx/data/input_reduce_max_axis_1.npy b/testdata/dnn/onnx/data/input_reduce_max_axis_1.npy
new file mode 100644
index 0000000000000000000000000000000000000000..9469ba8fb20a9fe27527df84dac8a61c6d6ac300
GIT binary patch
literal 176
zcmbR27wQ`j$;eQ~P_3SlTAW;@Zl$1ZlV+l>qoAIaUsO_*m=~X4l#&V(cT3DEP6dh=
zXCxM+0{I$-Its=*3PwPrsiRP<KnCE7u|H}bU^8u>XCe22Z7qWK3omx<*Ja(jzpDA1
a-P4XRdlyBE{oKvx_DSrO*nceD$_@aeO)+Kw

literal 0
HcmV?d00001

diff --git a/testdata/dnn/onnx/data/output_reduce_max_axis_0.npy b/testdata/dnn/onnx/data/output_reduce_max_axis_0.npy
new file mode 100644
index 0000000000000000000000000000000000000000..42b704542cc9c34ecbf46e84399ba77ee30dd1e9
GIT binary patch
literal 176
zcmbR27wQ`j$;eQ~P_3SlTAW;@Zl$1ZlV+l>qoAIaUsO_*m=~X4l#&V(cT3DEP6dh=
zXCxM+0{I%oItoTWq^YA&t3V#$im^XxA7C?WpJyTWfo&~<_6sj|?bl`9yuYgXoZZuo
XFnbq8i~Zcq=k`hLmDqnQ-O3ICnUyg~

literal 0
HcmV?d00001

diff --git a/testdata/dnn/onnx/data/output_reduce_max_axis_1.npy b/testdata/dnn/onnx/data/output_reduce_max_axis_1.npy
new file mode 100644
index 0000000000000000000000000000000000000000..034e23a74303a4259c2e060852cb95ef09218ad3
GIT binary patch
literal 144
zcmbR27wQ`j$;eQ~P_3SlTAW;@Zl$1ZlV+l>qoAIaUsO_*m=~X4l#&V(cT3DEP6dh=
pXCxM+0{I$-ItoTWq^YA&t3V#$im^XxA7C?WUsdxtyQdvt_5e~sBN6}r

literal 0
HcmV?d00001

diff --git a/testdata/dnn/onnx/generate_onnx_models.py b/testdata/dnn/onnx/generate_onnx_models.py
index 9b6f45604..9ca637ce0 100644
--- a/testdata/dnn/onnx/generate_onnx_models.py
+++ b/testdata/dnn/onnx/generate_onnx_models.py
@@ -944,15 +944,33 @@ def forward(self, x):
 model = Power(2)
 save_data_and_model("pow2", x, model)
 
-class ReduceMax(nn.Module):
+class ReduceMaxGlobal(nn.Module):
   def forward(self, x):
     out = torch.max(x)
     return torch.unsqueeze(out, 0)
 
 x = Variable(torch.randn(1, 3, 2, 2))
-model = ReduceMax()
+model = ReduceMaxGlobal()
 save_data_and_model("reduce_max", x, model)
 
+class ReduceMax(nn.Module):
+      def __init__(self, axes):
+    super(ReduceMax, self).__init__()
+    self.axes = axes
+
+  def forward(self, x):
+    # torch.return_types.max(values, indices)
+    out = torch.max(x, dim=self.axes, keepdim=False)[0]
+    return out
+
+x = Variable(torch.randn(1, 3, 2, 2))
+
+model = ReduceMax(axes=0)
+save_data_and_model("reduce_max_axis_0", x, model)
+
+model = ReduceMax(axes=1)
+save_data_and_model("reduce_max_axis_1", x, model)
+
 class ResizeConv(nn.Module):
     def __init__(
             self,
diff --git a/testdata/dnn/onnx/models/reduce_max_axis_0.onnx b/testdata/dnn/onnx/models/reduce_max_axis_0.onnx
new file mode 100644
index 0000000000000000000000000000000000000000..f844fd01669a6593e56b0983dfa1359967fd0874
GIT binary patch
literal 157
zcmd;J6Jjr@EXglQ&X8g@)HAngW8|{oVl)t9G?d~FN=+$EPW4T!h&NE;gmbjGxmXe_
zQi~lJ7BI4F@pExxr=}L9Wabu200o$Y1i*&rW@VP>rdAZ>7nMXwgUl1-65(JJ65wLu
eU<6`jAZ7yLBylcALm?q90iYxgvV;?p04D%k5FRlA

literal 0
HcmV?d00001

diff --git a/testdata/dnn/onnx/models/reduce_max_axis_1.onnx b/testdata/dnn/onnx/models/reduce_max_axis_1.onnx
new file mode 100644
index 0000000000000000000000000000000000000000..0c34b9fefe7ac17011dfd640a80a928198fb9f59
GIT binary patch
literal 157
zcmd;J6Jjr@EXglQ&X8g@)HAngW8|{oVl)t9G?d~FN=+$EPW4T!h&NE;gmbjGxmXe_
zQi~lJ7cjDG@pExxr=}L9Wabu2Ff3qX5)uF#rkj;nqMKS#kY7|1B@Hr9j7x-rQAmJ`
hiGvY{nSq!Igp<U%7!8GlxCDTbJWxrngcFkhCjeYm9xwm^

literal 0
HcmV?d00001