Skip to content
Snippets Groups Projects
Commit 05524c28 authored by Olivier BICHLER's avatar Olivier BICHLER
Browse files

Added PaddedConvDepthWise tests

parent ce9b1582
No related branches found
No related tags found
2 merge requests!710.4.0,!60[Fix] Graph's input memory offset
......@@ -870,6 +870,38 @@ class test_operator_export(unittest.TestCase):
initFiller(model)
self.unit_test_export(model, "PaddedConv_dilated_asym", [[1, 3, 19, 19]])
def test_PaddedConvDW(self):
print("PaddedConvDW")
model = aidge_core.sequential([
aidge_core.PaddedConvDepthWise2D(3, kernel_dims=[7, 7], name="InputNode", stride_dims=[2, 2], padding_dims=[3, 3, 3, 3])
])
initFiller(model)
self.unit_test_export(model, "PaddedConvDW", [[1, 3, 19, 19]])
def test_PaddedConvDW_asym(self):
print("PaddedConvDW_asym")
model = aidge_core.sequential([
aidge_core.PaddedConvDepthWise2D(3, kernel_dims=[3, 7], name="InputNode", stride_dims=[2, 1], padding_dims=[3, 1, 2, 3])
])
initFiller(model)
self.unit_test_export(model, "PaddedConvDW_asym", [[1, 3, 19, 19]])
def test_PaddedConvDW_dilated(self):
print("PaddedConvDW_dilated")
model = aidge_core.sequential([
aidge_core.PaddedConvDepthWise2D(3, kernel_dims=[7, 7], name="InputNode", stride_dims=[2, 2], padding_dims=[3, 3, 3, 3], dilation_dims=[2, 2])
])
initFiller(model)
self.unit_test_export(model, "PaddedConvDW_dilated", [[1, 3, 19, 19]])
def test_PaddedConvDW_dilated_asym(self):
print("PaddedConvDW_dilated_asym")
model = aidge_core.sequential([
aidge_core.PaddedConvDepthWise2D(3, kernel_dims=[7, 5], name="InputNode", stride_dims=[1, 2], padding_dims=[1, 3, 3, 0], dilation_dims=[1, 2])
])
initFiller(model)
self.unit_test_export(model, "PaddedConvDW_dilated_asym", [[1, 3, 19, 19]])
def test_Conv2(self):
print("Conv2")
model = aidge_core.sequential([
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment