Merge pull request #835 from l-bat:lb/onnx_mish_subgraph

Author: alalek

testdata/dnn/onnx/data/input_mish.npy

@@ -1224,4 +1224,12 @@ def forward(self, x):
 
 x = Variable(torch.randn(1, 2, 20, 20))
 model = Conv1ResizePoold()
-save_data_and_model("conv_resize_pool_1d", x, model)
+save_data_and_model("conv_resize_pool_1d", x, model)
+
+class Mish(nn.Module):
+    def forward(self, x):
+        return x * (torch.tanh(F.softplus(x)))
+
+x = Variable(torch.randn([1, 2, 2, 2]))
+model = Mish()
+save_data_and_model("mish", x, model)

testdata/dnn/onnx/data/output_mish.npy

@@ -0,0 +1,21 @@
+pytorch1.7:�

+
+
0
1
+Softplus_0"Softplus
+
+
1
2Tanh_1"Tanh
+
+
0
+
2
3Mul_2"Multorch-jit-exportZ
+
0
+

+

+
+
+b
+
3
+

+

+
+
+B