Imported from https://github.com/skiler07/mst-tf