message: apply test input_lib_test reviews
diff: @@ -1256,11 +1256,19 @@ class DistributedIteratorPerReplicaTest(test.TestCase, parameterized.TestCase):
  combinations.combine(
  mode=["eager"],
  distribution=[
  - strategy_combinations.mirrored_strategy_with_two_gpus]))
  + strategy_combinations.mirrored_strategy_with_two_gpus,
  + strategy_combinations.mirrored_strategy_with_cpu_1_and_2,
  + strategy_combinations.mirrored_strategy_with_gpu_and_cpu,
  + ]))
  +
  + def setUp(self):
  + super(DistributedIteratorPerReplicaTest, self).setUp()
  + strategy_combinations.set_virtual_cpus_to_at_least(3)
  +
  def testInputSignatureForPerReplicaValues(self, distribution):
  - with distribution.scope():
  def dataset_fn(input_context):
  - return tf.data.Dataset.from_tensor_slices(np.zeros([4, 4])).apply(tf.data.experimental.prefetch_to_device('gpu'))
  + return tf.data.Dataset.from_tensor_slices(
  + np.array(np.arange(0, 16)).reshape(4, 4))
  input_options = tf.distribute.InputOptions(replication_mode = tf.distribute.InputReplicationMode.PER_REPLICA)
  