a8b6963c7786f861c4211a9779c945cb7ab006f9,official/resnet/ctl/ctl_imagenet_main.py,,get_input_dataset,#Any#Any#,71

Before Change

        parse_record_fn=imagenet_preprocessing.parse_record,
        dtype=dtype)

    if strategy:
      test_ds = strategy.experimental_distribute_dataset(test_ds)

  return train_ds, test_ds


def get_num_train_iterations(flags_obj):

After Change

          input_context=ctx)
      return test_ds

  if strategy:
    if isinstance(strategy, tf.distribute.experimental.TPUStrategy):
      test_ds = strategy.experimental_distribute_datasets_from_function(_test_data_fn)
    else:
      test_ds = strategy.experimental_distribute_dataset(_test_data_fn())
  else:
    test_ds = _test_data_fn()

  return train_ds, test_ds


def get_num_train_iterations(flags_obj):

Instances