Data model in TensorFlow