Neural networks in TensorFlow