From b6f7db264d453872fbbde84f841b3c79c4ff7e2a Mon Sep 17 00:00:00 2001 From: SamFC10 Date: Wed, 30 Jun 2021 21:40:53 +0530 Subject: [PATCH] data for concat const blobs test --- .../onnx/data/input_concat_const_blobs.npy | Bin 0 -> 160 bytes .../onnx/data/output_concat_const_blobs.npy | Bin 0 -> 176 bytes testdata/dnn/onnx/generate_onnx_models.py | 18 ++++++++++++++++++ .../dnn/onnx/models/concat_const_blobs.onnx | Bin 0 -> 441 bytes 4 files changed, 18 insertions(+) create mode 100644 testdata/dnn/onnx/data/input_concat_const_blobs.npy create mode 100644 testdata/dnn/onnx/data/output_concat_const_blobs.npy create mode 100644 testdata/dnn/onnx/models/concat_const_blobs.onnx diff --git a/testdata/dnn/onnx/data/input_concat_const_blobs.npy b/testdata/dnn/onnx/data/input_concat_const_blobs.npy new file mode 100644 index 0000000000000000000000000000000000000000..988e07f97802437f841f496397dc4ff6cae54753 GIT binary patch literal 160 zcmbR27wQ`j$;eQ~P_3SlTAW;@Zl$1ZlV+l>qoAIaUsO_*m=~X4l#&V(cT3DEP6dh= zXCxM+0{I$-ItoS*q^YA&t3U?ey7!#fKIGPkeS#BG_S@{bWcy)F_I`!JY5V51IP81P HT4fIa_Tns@ literal 0 HcmV?d00001 diff --git a/testdata/dnn/onnx/data/output_concat_const_blobs.npy b/testdata/dnn/onnx/data/output_concat_const_blobs.npy new file mode 100644 index 0000000000000000000000000000000000000000..3840e62878d3ee63b31497853a9c577429e465d1 GIT binary patch literal 176 zcmbR27wQ`j$;eQ~P_3SlTAW;@Zl$1ZlV+l>qoAIaUsO_*m=~X4l#&V(cT3DEP6dh= zXCxM+0{I$-Its=*3PwPrsiRPmvZMj=!yS0ILUy{<& ZeLkh1?9QG!V|(_D&%RkRXYK>C?Euz#HH!cM literal 0 HcmV?d00001 diff --git a/testdata/dnn/onnx/generate_onnx_models.py b/testdata/dnn/onnx/generate_onnx_models.py index b09ca5fb5..234118f73 100644 --- a/testdata/dnn/onnx/generate_onnx_models.py +++ b/testdata/dnn/onnx/generate_onnx_models.py @@ -162,6 +162,24 @@ def forward(self, x): save_data_and_model("concatenation", input, model) +class ConcatConstBlob(nn.Module): + + def __init__(self): + super(ConcatConstBlob, self).__init__() + self.squeeze = nn.Conv2d(2, 2, kernel_size=1, stride=1, padding=0) + + def forward(self, x): + x = self.squeeze(x) + y = torch.tensor([[[[0.1, -0.2], [-0.3, 0.4]]]], dtype=torch.float32) + return torch.cat([x, y], axis=1) + + +input = Variable(torch.randn(1, 2, 2, 2)) +model = ConcatConstBlob() +model.eval() +save_data_and_model("concat_const_blob", input, model) + + class Mul(nn.Module): def __init__(self): diff --git a/testdata/dnn/onnx/models/concat_const_blobs.onnx b/testdata/dnn/onnx/models/concat_const_blobs.onnx new file mode 100644 index 0000000000000000000000000000000000000000..fc223e764f0d7a47526d38a122eaaead864f20cd GIT binary patch literal 441 zcmd;J6Jjr@EXglQ&X8g@)U&W!#LP8?k&87mub{Mqi?6t_G&QvRWxLDJR@=FUOfIKEG zQ7)eB)S|r9ocQ95#DY|qB0(;eg2a?!2L=cP=@Z~$FD@y{Oi6{A#N{l+Xd=Z0w5Yfw zF|Q=vP>BP^*3#f&ElbQPP1Ta&U<5)Y4kiIcFM+dX&e)zkHGh#Jo7N*L2vqo%?GWB@Yc^Au%oy d4n`pXE}+3c%ml+p(p-$DXp+n@Nhc-&E&u~uaXtV5 literal 0 HcmV?d00001