diff --git a/aidge_export_cpp/operators.py b/aidge_export_cpp/operators.py index de6c9642c66f3ae92a022cacbcb8e9b99ea056d7..c275a085fcc0dadfbf46faebf3568dc47e71fc0d 100644 --- a/aidge_export_cpp/operators.py +++ b/aidge_export_cpp/operators.py @@ -154,7 +154,7 @@ class ConvCPP(ExportNode): CONFIGURATIONS.CONV, name=self.name, input_dims=self.inputs_dims[0][1:], - output_dims=self.outputs_dims[0], + output_dims=self.outputs_dims[0][1:], kernel=self.kernel, stride=self.stride, padding=self.padding, diff --git a/aidge_export_cpp/templates/kernel_forward/convolution_forward.jinja b/aidge_export_cpp/templates/kernel_forward/convolution_forward.jinja index e0a53836070cb574fc17396db55bf5031721b2f2..fa253e6af0003cfece3fd8515ab105bfbe16d829 100644 --- a/aidge_export_cpp/templates/kernel_forward/convolution_forward.jinja +++ b/aidge_export_cpp/templates/kernel_forward/convolution_forward.jinja @@ -1,16 +1,16 @@ -conv_forward<{{name|upper}}_NB_CHANNELS, - {{name|upper}}_CHANNELS_HEIGHT, - {{name|upper}}_CHANNELS_WIDTH, - {{name|upper}}_NB_OUTPUTS, - {{name|upper}}_OUTPUTS_HEIGHT, - {{name|upper}}_OUTPUTS_WIDTH, - {{name|upper}}_PADDING_Y, - {{name|upper}}_PADDING_X, - {{name|upper}}_STRIDE_Y, - {{name|upper}}_STRIDE_X, - {{name|upper}}_DILATION_Y, - {{name|upper}}_DILATION_X, - {{name|upper}}_KERNEL_HEIGHT, - {{name|upper}}_KERNEL_WIDTH, - {{name|upper}}_ACTIVATION> - ({{input_name}}, {{output_name}}, {{weights_name}}, {{biases_name}}, {{name|upper}}_RESCALING); \ No newline at end of file +convolution_forward<{{name|upper}}_NB_CHANNELS, + {{name|upper}}_CHANNELS_HEIGHT, + {{name|upper}}_CHANNELS_WIDTH, + {{name|upper}}_NB_OUTPUTS, + {{name|upper}}_OUTPUTS_HEIGHT, + {{name|upper}}_OUTPUTS_WIDTH, + {{name|upper}}_PADDING_Y, + {{name|upper}}_PADDING_X, + {{name|upper}}_STRIDE_Y, + {{name|upper}}_STRIDE_X, + {{name|upper}}_DILATION_Y, + {{name|upper}}_DILATION_X, + {{name|upper}}_KERNEL_HEIGHT, + {{name|upper}}_KERNEL_WIDTH, + {{name|upper}}_ACTIVATION> + ({{input_name}}, {{output_name}}, {{weights_name}}, {{biases_name}}, {{name|upper}}_RESCALING);