Imported from https://github.com/lverwimp/tf-lm