Allow for decompress outside do_mask. It seems to be independent of that.
Allow for decompress outside do_mask. It seems to be independent of that.

PiperOrigin-RevId: 195744845

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