Tensorflow Simple Neural Network Diagram