Skip to content

Commit fa87b1b

Browse files
committed
Merge branch 4.x
2 parents 7e5d87c + 1507165 commit fa87b1b

File tree

5 files changed

+20
-22
lines changed

5 files changed

+20
-22
lines changed

testdata/dnn/onnx/data/README.md

Lines changed: 0 additions & 22 deletions
This file was deleted.
Binary file not shown.
Binary file not shown.

testdata/dnn/onnx/generate_onnx_models.py

Lines changed: 20 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -689,6 +689,26 @@ def forward(self, x):
689689
model = PoolConv()
690690
save_data_and_model("pool_conv_3d", input, model)
691691

692+
class DepthWiseAdd(nn.Module):
693+
694+
def __init__(self):
695+
super(DepthWiseAdd, self).__init__()
696+
self.dconv1 = nn.Conv2d(8, 8, kernel_size=3, stride=1, padding=0, groups=8)
697+
self.dconv2 = nn.Conv2d(8, 8, kernel_size=3, stride=1, padding=0, groups=8)
698+
699+
def forward(self, x):
700+
a = self.dconv1(x)
701+
b = self.dconv2(x)
702+
z = a + b
703+
z = z * 2
704+
return z
705+
706+
707+
input = Variable(torch.randn(1, 8, 32, 32))
708+
model = DepthWiseAdd()
709+
model.eval()
710+
save_data_and_model("depthwiseconv_add", input, model)
711+
692712
class Clip(nn.Module):
693713

694714
def __init__(self):
1.29 KB
Binary file not shown.

0 commit comments

Comments
 (0)