Added IAF Flow layer to latent_layers.py
Added IAF Flow layer to latent_layers.py

PiperOrigin-RevId: 213349934

tensorflow/tensor2tensor,tensorflow/tensor2tensor,tensorflow/tensor2tensor,tensorflow/tensor2tensor,tensorflow/tensor2tensor