Esempio n. 1
0
def _get_iterator(inputs, distribution_strategy=None):
    if distribution_strategy:
        return distributed_training_utils.get_iterator(inputs,
                                                       distribution_strategy)
    return training_utils.get_iterator(inputs)
Esempio n. 2
0
def _get_iterator(inputs, distribution_strategy=None):
  if distribution_strategy:
    return distributed_training_utils.get_iterator(
        inputs, distribution_strategy)
  return training_utils.get_iterator(inputs)