Searched refs:flat_map_fn (Results 1 – 4 of 4) sorted by relevance
52 def flat_map_fn(_): function59 return dataset_ops.Dataset.range(5).flat_map(flat_map_fn)93 def flat_map_fn(_): function101 return dataset_ops.Dataset.range(5).flat_map(flat_map_fn)
147 def flat_map_fn(_): function155 dataset = dataset.flat_map(flat_map_fn)164 def flat_map_fn(_): function173 dataset = dataset.flat_map(flat_map_fn)
318 def flat_map_fn(elem): function328 dataset = dataset_ops.Dataset.range(5).flat_map(flat_map_fn)334 def flat_map_fn(elem, message): function347 )).flat_map(flat_map_fn)
508 def flat_map_fn(dummy_arg): function531 return id_dataset.flat_map(flat_map_fn)