From 00d6a276db50a2476c6201c9e36d94271e2707ec Mon Sep 17 00:00:00 2001 From: Liubov Batanina Date: Mon, 1 Feb 2021 12:45:51 +0300 Subject: [PATCH] Added NormalizeL2 test data --- testdata/dnn/onnx/data/input_normalize_fusion.npy | Bin 0 -> 152 bytes .../dnn/onnx/data/output_normalize_fusion.npy | Bin 0 -> 152 bytes testdata/dnn/onnx/generate_onnx_models.py | 13 +++++++++++++ testdata/dnn/onnx/models/normalize_fusion.onnx | Bin 0 -> 322 bytes 4 files changed, 13 insertions(+) create mode 100644 testdata/dnn/onnx/data/input_normalize_fusion.npy create mode 100644 testdata/dnn/onnx/data/output_normalize_fusion.npy create mode 100644 testdata/dnn/onnx/models/normalize_fusion.onnx diff --git a/testdata/dnn/onnx/data/input_normalize_fusion.npy b/testdata/dnn/onnx/data/input_normalize_fusion.npy new file mode 100644 index 0000000000000000000000000000000000000000..8cdfd0bff462763fb3c1503ce71ff623fad7cd52 GIT binary patch literal 152 zcmbR27wQ`j$;eQ~P_3SlTAW;@Zl$1ZlV+l>qoAIaUsO_*m=~X4l#&V(cT3DEP6dh= wXCxM+0{I$7Its>`ItsN4WCN~)_q81O-JI;=l4k8wNq67BI+oYrfn?!+0Am6rasU7T literal 0 HcmV?d00001 diff --git a/testdata/dnn/onnx/data/output_normalize_fusion.npy b/testdata/dnn/onnx/data/output_normalize_fusion.npy new file mode 100644 index 0000000000000000000000000000000000000000..420aebbce5cb7b8114913fe3ab334aef0b7920b6 GIT binary patch literal 152 zcmbR27wQ`j$;eQ~P_3SlTAW;@Zl$1ZlV+l>qoAIaUsO_*m=~X4l#&V(cT3DEP6dh= xXCxM+0{I$7Its>`ItsN4WCJeO1GV;l?I+mgJ3rmK_1D~essH2cpEB>?2LOw5DEj~a literal 0 HcmV?d00001 diff --git a/testdata/dnn/onnx/generate_onnx_models.py b/testdata/dnn/onnx/generate_onnx_models.py index 8d5cc52c3..f017ad775 100644 --- a/testdata/dnn/onnx/generate_onnx_models.py +++ b/testdata/dnn/onnx/generate_onnx_models.py @@ -1245,3 +1245,16 @@ def forward(self, x): x = Variable(torch.randn([1, 1, 3, 4])) model = PadCalculation() save_data_and_model("calc_pads", x, model, version=11) + +class NormalizeFusion(nn.Module): + def forward(self, x): + mul = x * x + sum = torch.sum(mul, dim=(1), keepdim=True) + maximum = torch.clamp(sum, min=1e-8) + sqrt = torch.sqrt(maximum) + reciprocal = torch.reciprocal(sqrt) + return x * reciprocal + +x = Variable(torch.randn([2, 3])) +model = NormalizeFusion() +save_data_and_model("normalize_fusion", x, model) diff --git a/testdata/dnn/onnx/models/normalize_fusion.onnx b/testdata/dnn/onnx/models/normalize_fusion.onnx new file mode 100644 index 0000000000000000000000000000000000000000..196c6808cc16d31f97827d8de35179112694b239 GIT binary patch literal 322 zcmZ9HxlY4C5I{4&vYBvLOCq5yO=lE{ZwVoR2o(|y>1b9aSqXzLoOK8tpW>ch;iIs_ zQ6SCe&AjdrgZ9feSyYQhraC9l*@uE-glRhz9H!saDW6(u3p__~D7b8RCzka>%E8KQbb6Y=oO5{Xr%A%6@2VMOh?hv2^6{57OKD09WFRv7H@C9kj BL(l*K literal 0 HcmV?d00001