Applying dropout operations with TensorFlow