Enabling Uncertainty Estimation in Iterative Neural Networks