Skip to content
Snippets Groups Projects
Commit d909b06c authored by Cyril Moineau's avatar Cyril Moineau
Browse files

Fix Conv op export ways of getting nb_in_chan

parent a07a87bb
No related branches found
No related tags found
3 merge requests!27v0.2.0,!22v0.4.0,!15Export refactor
......@@ -144,8 +144,8 @@ class ConvCPP(ExportNode):
# Use PaddedConv to add padding attribute
self.padding = [0, 0]
self.nb_channels = node.get_operator().get_attr("InChannels")
self.nb_outputs = node.get_operator().get_attr("OutChannels")
self.nb_channels = node.get_operator().in_channels()
self.nb_outputs = node.get_operator().out_channels()
if len(self.inputs_dims[0]) == 4:
# if dims == [batch, nb_channels, height, width]
......@@ -187,7 +187,6 @@ class ConvCPP(ExportNode):
if not self.is_last:
list_actions.append(set_up_output(self.name, "float"))
print(self.inputs[0])
list_actions.append(generate_str(
str(ROOT / "templates" / "kernel_forward" / "convolution_forward.jinja"),
name=self.name,
......
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