Skip to content
Snippets Groups Projects
Commit b5213340 authored by Vincent Templier's avatar Vincent Templier
Browse files

Merge branch 'master' into 'master'

Correction of the convolution_forward.jinja file

See merge request !5
parents 2fb1c6c2 f46c715d
No related branches found
No related tags found
No related merge requests found
......@@ -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,
......
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);
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