From b67584eea5a66f303f780429c662b203569db3e5 Mon Sep 17 00:00:00 2001 From: SamFC10 Date: Tue, 16 Feb 2021 20:45:55 +0530 Subject: [PATCH] Added data for exp layer test --- testdata/dnn/onnx/data/input_exp.npy | Bin 0 -> 144 bytes testdata/dnn/onnx/data/output_exp.npy | Bin 0 -> 144 bytes testdata/dnn/onnx/generate_onnx_models.py | 8 ++++++++ testdata/dnn/onnx/models/exp.onnx | 11 +++++++++++ 4 files changed, 19 insertions(+) create mode 100644 testdata/dnn/onnx/data/input_exp.npy create mode 100644 testdata/dnn/onnx/data/output_exp.npy create mode 100644 testdata/dnn/onnx/models/exp.onnx diff --git a/testdata/dnn/onnx/data/input_exp.npy b/testdata/dnn/onnx/data/input_exp.npy new file mode 100644 index 0000000000000000000000000000000000000000..09071c0f9feaeac2ad103c80f9aa6ee1478b71f3 GIT binary patch literal 144 zcmbR27wQ`j$;eQ~P_3SlTAW;@Zl$1ZlV+l>qoAIaUsO_*m=~X4l#&V(cT3DEP6dh= oXCxM+0{I$7ItoUbItsN4WCO1Mhws>3NxEpyvS;c3Y37&g0b$N0j{pDw literal 0 HcmV?d00001 diff --git a/testdata/dnn/onnx/data/output_exp.npy b/testdata/dnn/onnx/data/output_exp.npy new file mode 100644 index 0000000000000000000000000000000000000000..d7e7b1bef6daa9111a2787a9b41303723d4f4adf GIT binary patch literal 144 zcmbR27wQ`j$;eQ~P_3SlTAW;@Zl$1ZlV+l>qoAIaUsO_*m=~X4l#&V(cT3DEP6dh= oXCxM+0{I$7ItoUbItsN4WCN~BhNJe8?n@k+oqOzjB9}S<06>``h5!Hn literal 0 HcmV?d00001 diff --git a/testdata/dnn/onnx/generate_onnx_models.py b/testdata/dnn/onnx/generate_onnx_models.py index f017ad775..83777c983 100644 --- a/testdata/dnn/onnx/generate_onnx_models.py +++ b/testdata/dnn/onnx/generate_onnx_models.py @@ -958,6 +958,14 @@ def forward(self, x): model = Power(2) save_data_and_model("pow2", x, model) +class Exp(nn.Module): + def forward(self, x): + return x.exp() + +x = Variable(torch.randn(2, 2)) +model = Exp() +save_data_and_model("exp", x, model) + class ReduceMaxGlobal(nn.Module): def forward(self, x): out = torch.max(x) diff --git a/testdata/dnn/onnx/models/exp.onnx b/testdata/dnn/onnx/models/exp.onnx new file mode 100644 index 000000000..2990fc312 --- /dev/null +++ b/testdata/dnn/onnx/models/exp.onnx @@ -0,0 +1,11 @@ +pytorch1.7:P + +01Exp_0"Exptorch-jit-exportZ +0 +  + +b +1 +  + +B \ No newline at end of file