message: Fix the resnet50_graph_test benchmark
diff: @@ -104,7 +104,7 @@ class ResNet50Benchmarks(tf.test.Benchmark):
  with tf.Graph().as_default():
  np_images, np_labels = random_batch(batch_size)
  dataset = tf.data.Dataset.from_tensors((np_images, np_labels)).repeat()
  - images, labels = tf.compat.v1.data.make_one_shot_iterator(
  + images, labels = tf.data.make_one_shot_iterator(
  dataset).get_next()
  model = resnet50.ResNet50(data_format())
  