tensorflow_gan/python/estimator/tpu_gan_estimator_test_lib.py [371:379]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
      data = np.zeros([input_dim], dtype=np.float32)
      ds = (
          tf.data.Dataset.from_tensors((data, data)).repeat().batch(
              params['batch_size'], drop_remainder=True))
      if return_ds:
        return ds
      else:
        x, y = tf.compat.v1.data.make_one_shot_iterator(ds).get_next()
        return x, y
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



tensorflow_gan/python/estimator/tpu_gan_estimator_test_lib.py [382:390]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
      data = np.zeros([input_dim], dtype=np.float32)
      ds = (
          tf.data.Dataset.from_tensors((data, data)).repeat().batch(
              params['batch_size'], drop_remainder=True))
      if return_ds:
        return ds
      else:
        x, y = tf.compat.v1.data.make_one_shot_iterator(ds).get_next()
        return x, y
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -