Linear regression with TensorFlow