Tensorflow Graph Neural Network Code