Imported from https://github.com/taki0112/BigGAN-Tensorflow