Revert "Don't re-flatten parameters if they are already flattened"
Revert "Don't re-flatten parameters if they are already flattened"

jhjin/nn,jonathantompson/nn,sagarwaghmare69/nn,eriche2016/nn,xianjiec/nn,apaszke/nn,colesbury/nn,PraveerSINGH/nn,elbamos/nn,caldweln/nn,kmul00/nn,joeyhng/nn,andreaskoepf/nn,nicholas-leonard/nn,diz-vara/nn