Modelling flow-induced microstructural segregation in semi-solid metals