Get rid of ae_input, since do_compress_attend is doing the same thing
Get rid of ae_input, since do_compress_attend is doing the same thing

PiperOrigin-RevId: 187102333

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