A BED LOAD TRANSPORT MODEL FOR NON-UNIFORM FLOWS