Fix sharded decoding name convention.
Fix sharded decoding name convention.

PiperOrigin-RevId: 228604730

tensorflow/tensor2tensor,tensorflow/tensor2tensor,tensorflow/tensor2tensor,tensorflow/tensor2tensor,tensorflow/tensor2tensor