Communication-Aware DNN Pruning