Linear regression in TensorFlow