diff --git a/aidge_export_cpp/operators.py b/aidge_export_cpp/operators.py index b6121e4a8ca5666e2c449ee821d7119b40934362..cb7a09c9fa01eee433f2cbcdb09cc66dc0bb17a6 100644 --- a/aidge_export_cpp/operators.py +++ b/aidge_export_cpp/operators.py @@ -188,7 +188,7 @@ class PaddedConvCPP(ExportNodeCpp): _setup_conv2D(self) @ExportLibCpp.register("ConvDepthWise2D", aidge_core.ImplSpec(aidge_core.IOSpec(aidge_core.dtype.float32))) -class ConvCPP(ExportNodeCpp): +class ConvDepthWiseCPP(ExportNodeCpp): def __init__(self, node, mem_info): super().__init__(node, mem_info) self.attributes["depthwise"] = True @@ -200,7 +200,7 @@ class ConvCPP(ExportNodeCpp): _setup_conv2D(self) @ExportLibCpp.register_metaop("PaddedConvDepthWise2D", aidge_core.ImplSpec(aidge_core.IOSpec(aidge_core.dtype.float32))) -class PaddedConvCPP(ExportNodeCpp): +class PaddedConvDepthWiseCPP(ExportNodeCpp): def __init__(self, node, mem_info): super().__init__(node, mem_info) self.attributes["depthwise"] = True