Helper functions used for Tensorflow