Tensorflow Neural Network Code