c1db25d9281a9374005d5c8b4e183b8e6d7d2ace,cloud_tpu/models/resnet/estimator_resnet.py,,input_fn,#Any#Any#,193

Before Change

  // TODO(xiejw,saeta): Consider removing the sharding dimension below.
  images_shape = images.get_shape().as_list()
  if images_shape[0] is None:
    images_shape[0] = batch_size
    images = tf.reshape(
        images, images_shape, name="InputPipeline/images/reshape")
  labels_shape = labels.get_shape().as_list()

After Change

  images.set_shape(images.get_shape().merge_with(
      tf.TensorShape([batch_size, None, None, None])))
  labels.set_shape(
      labels.get_shape().merge_with(tf.TensorShape([batch_size, None])))
  images = pipeline_outputs_transform(images)
  return images, labels

Instances