Operations using `Non-trainable params` are not counted in `Mult-Adds`

[Yohan Poirier-Ginter]

This does not seem like the intended behavior? If it is, maybe consider adding a second count for non-traininable params