Use padding ratio instead of padding amount on the dimension to determine best sharding dim for per-stage vars.
Use padding ratio instead of padding amount on the dimension to determine best sharding dim for per-stage vars.

PiperOrigin-RevId: 377378333

tensorflow/lingvo,tensorflow/lingvo,tensorflow/lingvo,tensorflow/lingvo