Coupling rheology and segregation in granular flows. (29th December 2020)