diff --git a/aidge_export_cpp/kernels/convolution.hpp b/aidge_export_cpp/kernels/convolution.hpp index 7df55abdbfd1297ddf31c0c9bf35c7fa9f076af2..1669ee86894e6834b4c37adc42d946bcb4acef8a 100644 --- a/aidge_export_cpp/kernels/convolution.hpp +++ b/aidge_export_cpp/kernels/convolution.hpp @@ -237,7 +237,7 @@ void convolution_depthwise_forward( const int iOffset = NB_CHANNELS * iPos; const int wOffset = (output*KERNEL_HEIGHT + sy) - * KERNEL_WIDTH * NB_CHANNELS; + * KERNEL_WIDTH; if (DILATION_X == 1 && ((PADDING_X == 0 && OUTPUTS_WIDTH == OUTPUTS_WIDTH_NOPAD) diff --git a/aidge_export_cpp/unit_tests/test_export.py b/aidge_export_cpp/unit_tests/test_export.py index 387c59519b3c41294444267a23c24e4ec704ee22..65db07e33c68e4900476f97d92879ad7ca3ed4e2 100644 --- a/aidge_export_cpp/unit_tests/test_export.py +++ b/aidge_export_cpp/unit_tests/test_export.py @@ -411,7 +411,7 @@ class test_operator_export(unittest.TestCase): self.unit_test_export(model, "Conv2D", [[1, 3, 12, 12]], False, False) def test_export_convDepthWise2D(self): - print("Conv2D") + print("ConvDepthWise2D") model = aidge_core.sequential([ aidge_core.ConvDepthWise2D(nb_channels=3, kernel_dims=(3, 3), name="conv") ])