TensorFlow code structure