Hashed attention that uses top-k buckets
Hashed attention that uses top-k buckets

PiperOrigin-RevId: 270192777

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