Searched refs:flat_map_fn (Results 1 - 2 of 2) sorted by relevance
/external/tensorflow/tensorflow/contrib/data/python/kernel_tests/ | ||
H A D | flat_map_dataset_op_test.py | 52 def flat_map_fn(_): function in function:FlatMapDatasetSerializationTest.testMapThenFlatMap.build_ds 59 return dataset_ops.Dataset.range(5).flat_map(flat_map_fn) 93 def flat_map_fn(_): function in function:FlatMapDatasetSerializationTest.testDisallowCapturingStatefulOps.build_ds 101 return dataset_ops.Dataset.range(5).flat_map(flat_map_fn) |
/external/tensorflow/tensorflow/python/data/ops/ | ||
H A D | dataset_ops.py | 413 def flat_map_fn(iterator_id_t): function in function:Dataset.from_generator 437 return id_dataset.flat_map(flat_map_fn) |
Completed in 156 milliseconds