[Transform] Add comment to streamingeltwise conversion

Xilinx · Aug 22, 2024 · e1d1f63 · e1d1f63
1 parent ec7be72
commit e1d1f63
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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