What are the consequences of putting that in `fprop` rather than in `_linear_part`?