Fix removal of nodes ingested by multiple downstream nodes
- Currently, when a node is removed (such as a linear activation), it is only replaced correctly in the first downstream node that uses it
- This PR replaces it in all downstream nodes
To do
-
Add a pytest
@nickshey