From 58fbc3f216123aa64ad0a10cfc8ac42051da3300 Mon Sep 17 00:00:00 2001
From: Smirnov Egor <s.e.a.98@yandex.ru>
Date: Mon, 11 Oct 2021 18:59:15 +0300
Subject: [PATCH] add tests for Sub

---
 .../dnn/onnx/data/input_sub_from_const1.npy   | Bin 0 -> 224 bytes
 .../data/input_sub_from_const_broadcast.npy   | Bin 0 -> 152 bytes
 .../data/input_sub_from_const_eltwise.npy     | Bin 0 -> 224 bytes
 .../dnn/onnx/data/output_sub_from_const1.npy  | Bin 0 -> 224 bytes
 .../data/output_sub_from_const_broadcast.npy  | Bin 0 -> 152 bytes
 .../data/output_sub_from_const_eltwise.npy    | Bin 0 -> 224 bytes
 testdata/dnn/onnx/generate_onnx_models.py     |  35 ++++++++++++++++++
 testdata/dnn/onnx/models/sub_from_const1.onnx | Bin 0 -> 171 bytes
 .../onnx/models/sub_from_const_broadcast.onnx |  15 ++++++++
 .../onnx/models/sub_from_const_eltwise.onnx   |  19 ++++++++++
 10 files changed, 69 insertions(+)
 create mode 100644 testdata/dnn/onnx/data/input_sub_from_const1.npy
 create mode 100644 testdata/dnn/onnx/data/input_sub_from_const_broadcast.npy
 create mode 100644 testdata/dnn/onnx/data/input_sub_from_const_eltwise.npy
 create mode 100644 testdata/dnn/onnx/data/output_sub_from_const1.npy
 create mode 100644 testdata/dnn/onnx/data/output_sub_from_const_broadcast.npy
 create mode 100644 testdata/dnn/onnx/data/output_sub_from_const_eltwise.npy
 create mode 100644 testdata/dnn/onnx/models/sub_from_const1.onnx
 create mode 100644 testdata/dnn/onnx/models/sub_from_const_broadcast.onnx
 create mode 100644 testdata/dnn/onnx/models/sub_from_const_eltwise.onnx

diff --git a/testdata/dnn/onnx/data/input_sub_from_const1.npy b/testdata/dnn/onnx/data/input_sub_from_const1.npy
new file mode 100644
index 0000000000000000000000000000000000000000..7ebf75d8a96fc360bb01ad51c2ab371ff7807f96
GIT binary patch
literal 224
zcmbR27wQ`j$;eQ~P_3SlTAW;@Zl$1ZlV+l>qoAIaUsO_*m=~X4l#&V(cT3DEP6dh=
zXCxM+0{I$-ItoTQ3dTALCYm}5wF+bat{C0@d)prA+h3a<wqNyRrd51fxV`(uGyBAO
zTkUVeZ?|J(nr3U!*k>QwI>GL8&^<fd0^j{BPb}U4l-pr{-Mi)c&%_1Wt=qY3-|YU|
a`;?mU?9MRw*ekf(?)%uhcVFYHS^EKKgiBrk

literal 0
HcmV?d00001

diff --git a/testdata/dnn/onnx/data/input_sub_from_const_broadcast.npy b/testdata/dnn/onnx/data/input_sub_from_const_broadcast.npy
new file mode 100644
index 0000000000000000000000000000000000000000..5f34483a08b566f3c7d8c55f9cb19c7488e3b4dc
GIT binary patch
literal 152
zcmbR27wQ`j$;eQ~P_3SlTAW;@Zl$1ZlV+l>qoAIaUsO_*m=~X4l#&V(cT3DEP6dh=
xXCxM+0{I$7Its>`ItsN4WCN}#hyU5x|0%P*I+JbxcY(wEDlW~l-|}G1egK!)Db4@@

literal 0
HcmV?d00001

diff --git a/testdata/dnn/onnx/data/input_sub_from_const_eltwise.npy b/testdata/dnn/onnx/data/input_sub_from_const_eltwise.npy
new file mode 100644
index 0000000000000000000000000000000000000000..38408b6ed04cf85e4f6996690c49311e37eb611a
GIT binary patch
literal 224
zcmbR27wQ`j$;eQ~P_3SlTAW;@Zl$1ZlV+l>qoAIaUsO_*m=~X4l#&V(cT3DEP6dh=
zXCxM+0{I$-ItoTQ3dTALCYm}5wF+bauHPXn`wq;D-+Ni;qTL~#3HIOCf49H<yTM*^
zMuWZIx4Zjp@e1sZ)eqfowQ$q^jDj=!(l^Z8C$%tjuTCk8eQn^G{gsEe*?r~|J`nsr
bZhw#4wf#vd0`{I8w(p;+WnsU(Wc_{sT@OtK

literal 0
HcmV?d00001

diff --git a/testdata/dnn/onnx/data/output_sub_from_const1.npy b/testdata/dnn/onnx/data/output_sub_from_const1.npy
new file mode 100644
index 0000000000000000000000000000000000000000..a4a4aa6675e25fb2a3363b5b90172c2b69720575
GIT binary patch
literal 224
zcmbR27wQ`j$;eQ~P_3SlTAW;@Zl$1ZlV+l>qoAIaUsO_*m=~X4l#&V(cT3DEP6dh=
zXCxM+0{I$-ItoTQ3dTALCYm}5wF+bau4|vV?Y~$puzT(M+P?2-y?x6wMY|xA&Gru#
zDee^sP_f@{pKISB{9<p|UUU0t*97dvc098Wc_QTSiS>cK{dZvpUkz{joOP@0pFiGa
aUlH$Of8me6-IEp5?f)dKw|6MzbpQZ#)<}i`

literal 0
HcmV?d00001

diff --git a/testdata/dnn/onnx/data/output_sub_from_const_broadcast.npy b/testdata/dnn/onnx/data/output_sub_from_const_broadcast.npy
new file mode 100644
index 0000000000000000000000000000000000000000..c7ad16fec1697428d30a9bc67b8383d09c608200
GIT binary patch
literal 152
zcmbR27wQ`j$;eQ~P_3SlTAW;@Zl$1ZlV+l>qoAIaUsO_*m=~X4l#&V(cT3DEP6dh=
xXCxM+0{I$7Its>`ItsN4WCJeGzaQ+I<Sy-PSoL(D*-it8I-4c?mn}J92LO@vD4_rV

literal 0
HcmV?d00001

diff --git a/testdata/dnn/onnx/data/output_sub_from_const_eltwise.npy b/testdata/dnn/onnx/data/output_sub_from_const_eltwise.npy
new file mode 100644
index 0000000000000000000000000000000000000000..013f17ad3f5dd31fc485e623fa3e06a24a6192f8
GIT binary patch
literal 224
zcmbR27wQ`j$;eQ~P_3SlTAW;@Zl$1ZlV+l>qoAIaUsO_*m=~X4l#&V(cT3DEP6dh=
zXCxM+0{I$-ItoTQ3dTALCYm}5wF+bau2>QG{f9d^_iNm5-Zx<f*8#Jyng@z({0{WU
z+}N+JU2K2gTKV3l%d+<KG&1akuTQt{Ja*Mi^=IS$vw=$cPk4#gTXG!Vzh;G@!=%>j
b`%WyGZKuOnZLd+V+5Y;fiTj=Q_SpjfgH%b|

literal 0
HcmV?d00001

diff --git a/testdata/dnn/onnx/generate_onnx_models.py b/testdata/dnn/onnx/generate_onnx_models.py
index af8032550..d1965c2f4 100644
--- a/testdata/dnn/onnx/generate_onnx_models.py
+++ b/testdata/dnn/onnx/generate_onnx_models.py
@@ -1523,3 +1523,38 @@ def save_data_and_paddle_model(model, name, input_data):
 input_shape = [1, 2, 3, 4]
 x = paddle.rand(input_shape, dtype="float32")
 save_data_and_paddle_model(Resize_HumanSeg(), "resize_humanseg", x)
+
+class SubFromConstBroadcast(nn.Module):
+    def __init__(self, *args, **kwargs):
+        super(SubFromConstBroadcast, self).__init__()
+        self.const = torch.randn(1, 3, dtype=torch.float32)
+
+    def forward(self, x):
+        return self.const - x
+
+model = SubFromConstBroadcast()
+input_ = Variable(torch.randn(2, 3, dtype=torch.float32))
+save_data_and_model("sub_from_const_broadcast", input_, model)
+
+class SubFromConstEltWise(nn.Module):
+    def __init__(self, *args, **kwargs):
+        super(SubFromConstEltWise, self).__init__()
+        self.const = torch.randn(1, 2, 3, 4, dtype=torch.float32)
+
+    def forward(self, x):
+        return self.const - x
+
+model = SubFromConstEltWise()
+input_ = Variable(torch.randn(1, 2, 3, 4, dtype=torch.float32))
+save_data_and_model("sub_from_const_eltwise", input_, model)
+
+class SubFromConst1(nn.Module):
+    def __init__(self, *args, **kwargs):
+        super(SubFromConst1, self).__init__()
+
+    def forward(self, x):
+        return 1 - x
+
+model = SubFromConst1()
+input_ = Variable(torch.randn(1, 2, 3, 4, dtype=torch.float32))
+save_data_and_model("sub_from_const1", input_, model)
diff --git a/testdata/dnn/onnx/models/sub_from_const1.onnx b/testdata/dnn/onnx/models/sub_from_const1.onnx
new file mode 100644
index 0000000000000000000000000000000000000000..4809de5f209239cb0e53b0790726398d0811e871
GIT binary patch
literal 171
zcmd;J6Jjr@EXglQ&X8g@)U&jj!pNmB#AqnR<(!{aT#}eq5^tcy0b^^4aIuyp=9H#t
zaR@Mau`n<+*e_sY;S%LyG~{Be5MnfvVht`$iZ@hZ22w%-U}JT&GD~z*D+=<9N}{Ae
is>HZNI2eTlxR^K?ftU%1nSq!k2`FiVOVWu+fD-`v&L4LG

literal 0
HcmV?d00001

diff --git a/testdata/dnn/onnx/models/sub_from_const_broadcast.onnx b/testdata/dnn/onnx/models/sub_from_const_broadcast.onnx
new file mode 100644
index 000000000..feabfa849
--- /dev/null
+++ b/testdata/dnn/onnx/models/sub_from_const_broadcast.onnx
@@ -0,0 +1,15 @@
+pytorch1.9:�

+;
1
+Constant_0"Constant* 
+value*

JW@�@-�oy��

+
+
1
+
0
2Sub_1"Subtorch-jit-exportZ
+
0
+

+
+b
+
2
+

+
+B	
\ No newline at end of file
diff --git a/testdata/dnn/onnx/models/sub_from_const_eltwise.onnx b/testdata/dnn/onnx/models/sub_from_const_eltwise.onnx
new file mode 100644
index 000000000..feb661f7c
--- /dev/null
+++ b/testdata/dnn/onnx/models/sub_from_const_eltwise.onnx
@@ -0,0 +1,19 @@
+pytorch1.9:�

+�

1
+Constant_0"Constant*x
+value*l

J`�)h����Vg>]D���t8���
��/�0?F��ko]��3���uI?�X�=.&����,��[W�����>a��Z]��ǒ�?�2V־�݂��

+
+
1
+
0
2Sub_1"Subtorch-jit-exportZ
+
0
+

+

+
+
+b
+
2
+

+

+
+
+B	
\ No newline at end of file