Story Tag: https://github.com/tensorflow/tpu/issues/1140