2d074b5ddfbfbc486506384993bb5b269903e144,scripts/tf_cnn_benchmarks/benchmark_cnn.py,BenchmarkCNN,_build_input_processing,#BenchmarkCNN#Any#,2190

Before Change

        input_producer_stages.append(staging_area)
        for group_index in xrange(self.batch_group_size):
          batch_index = group_index + device_num * self.batch_group_size
          put_op = staging_area.put(
              [images_splits[batch_index], labels_splits[batch_index]])
          input_producer_op.append(put_op)
      assert input_producer_op

    return input_processing_info._replace(

After Change

      for device_num in range(len(self.devices)):
        staging_area = data_flow_ops.StagingArea(
            [parts[0].dtype for parts in input_list],
            shapes=[parts[0].get_shape() for parts in input_list],
            shared_name="input_producer_staging_area_%d" % device_num)
        input_producer_stages.append(staging_area)
        for group_index in xrange(self.batch_group_size):

Instances