Home
last modified time | relevance | path

Searched refs:flat_map_fn (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/serialization/
Dflat_map_dataset_serialization_test.py52 def flat_map_fn(_): function
59 return dataset_ops.Dataset.range(5).flat_map(flat_map_fn)
93 def flat_map_fn(_): function
101 return dataset_ops.Dataset.range(5).flat_map(flat_map_fn)
/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/optimization/
Doptimize_dataset_test.py147 def flat_map_fn(_): function
155 dataset = dataset.flat_map(flat_map_fn)
164 def flat_map_fn(_): function
173 dataset = dataset.flat_map(flat_map_fn)
/external/tensorflow/tensorflow/python/data/kernel_tests/
Dfrom_generator_test.py318 def flat_map_fn(elem): function
328 dataset = dataset_ops.Dataset.range(5).flat_map(flat_map_fn)
334 def flat_map_fn(elem, message): function
347 )).flat_map(flat_map_fn)
/external/tensorflow/tensorflow/python/data/ops/
Ddataset_ops.py508 def flat_map_fn(dummy_arg): function
531 return id_dataset.flat_map(flat_map_fn)