Make pix2pix flexible so that cyclegan can use it.
Make pix2pix flexible so that cyclegan can use it.

PiperOrigin-RevId: 253589693

tensorflow/examples,tensorflow/examples,tensorflow/examples,tensorflow/examples,tensorflow/examples,tensorflow/examples,tensorflow/examples,tensorflow/examples,tensorflow/examples