Fixed a bug in pad_to_multiple_2d. The height and width were incorrect.
Fixed a bug in pad_to_multiple_2d. The height and width were incorrect.

PiperOrigin-RevId: 168060083

tensorflow/tensor2tensor,tensorflow/tensor2tensor,vthorsteinsson/tensor2tensor,waterblue13/tensor2tensor,tensorflow/tensor2tensor,rsepassi/tensor2tensor,vthorsteinsson/tensor2tensor,tensorflow/tensor2tensor,waterblue13/tensor2tensor,vthorsteinsson/tensor2tensor,rsepassi/tensor2tensor,rsepassi/tensor2tensor,rsepassi/tensor2tensor,vthorsteinsson/tensor2tensor,tensorflow/tensor2tensor,waterblue13/tensor2tensor