Function _maybe_map_fn

official/core/input_reader.py:31–35

Calls dataset.map if a valid function is passed in.

(dataset: tf.data.Dataset,
                  fn: Optional[Callable[..., Any]] = None)

def _maybe_map_fn(dataset: tf.data.Dataset,
                  fn: Optional[Callable[..., Any]] = None) -> tf.data.Dataset:
  """Calls dataset.map if a valid function is passed in."""
  return dataset if fn is None else dataset.map(
      fn, num_parallel_calls=tf.data.experimental.AUTOTUNE)


def match_files(input_path: Union[Sequence[str], str]) -> List[str]: