I changed it to `all_nonzero_shape` which is more similar `to numpy.all`. 
This API is used in `distarray.update`. It first finds out the the overlap regions, then it has to check if the overlap regions are all nonzero. So this is a necessary step.