From f46c715d3eaa499244a3490401c620f9b1d0db50 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Cl=C3=A9ment=20Fisher?= <clement.fisher@cea.fr>
Date: Tue, 13 Feb 2024 14:59:07 +0000
Subject: [PATCH] Update operators.py
Correction of the output values used for the layers.h template in convolution
---
aidge_export_cpp/operators.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/aidge_export_cpp/operators.py b/aidge_export_cpp/operators.py
index de6c964..c275a08 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,
--
GitLab