Top tensor2tensor issues