diff --git a/src/finn/transformation/fpgadataflow/convert_to_hw_layers.py b/src/finn/transformation/fpgadataflow/convert_to_hw_layers.py index ea5025a098..25a2032aeb 100644 --- a/src/finn/transformation/fpgadataflow/convert_to_hw_layers.py +++ b/src/finn/transformation/fpgadataflow/convert_to_hw_layers.py @@ -1251,8 +1251,8 @@ def apply(self, model): class InferStreamingEltwise(Transformation): - """Convert eltwise Sub or Sub -> Abs to StreamingEltwise layer - with SubEltwise or AbsDiffEltwise op.""" + """Convert eltwise Add, Sub or Sub -> Abs to StreamingEltwise layer + with AddEltwise, SubEltwise or AbsDiffEltwise op.""" def apply(self, model): graph = model.graph