TensorFlow high-level API - loading data