Simplify local_shape computation...

now that no-distribution dimensions have a map.