Improvements to partitioning algorithm

- Now passing all staging aliases through DEG unmodified
- Locality preferring: stay local if array inputs are local
- Logreg is being partitioned correctly