Searched refs:flat_map_fn (Results 1 – 4 of 4) sorted by relevance
59 def flat_map_fn(_): function66 return dataset_ops.Dataset.range(5).flat_map(flat_map_fn)103 def flat_map_fn(_): function111 return dataset_ops.Dataset.range(5).flat_map(flat_map_fn)
176 def flat_map_fn(_): function184 dataset = dataset.flat_map(flat_map_fn)194 def flat_map_fn(_): function203 dataset = dataset.flat_map(flat_map_fn)213 def flat_map_fn(_): function223 dataset = dataset.flat_map(flat_map_fn)
374 def flat_map_fn(elem): function384 dataset = dataset_ops.Dataset.range(5).flat_map(flat_map_fn)391 def flat_map_fn(elem, message): function404 )).flat_map(flat_map_fn)
1006 def flat_map_fn(dummy_arg): function1029 return id_dataset.flat_map(flat_map_fn)