Searched refs:flat_map (Results 1 - 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/contrib/data/python/kernel_tests/
H A Dflat_map_dataset_op_test.py44 return dataset_ops.Dataset.range(start, start + 5 * 5, 5).flat_map(map_fn)
59 return dataset_ops.Dataset.range(5).flat_map(flat_map_fn)
75 return dataset_ops.Dataset.range(100).flat_map(map_fn)
84 return dataset_ops.Dataset.range(5).flat_map(
101 return dataset_ops.Dataset.range(5).flat_map(flat_map_fn)
116 return dataset_ops.Dataset.range(10).map(_map_fn).flat_map(_flat_map_fn)
/external/tensorflow/tensorflow/python/data/kernel_tests/
H A Dflat_map_dataset_op_test.py41 .flat_map(lambda x: dataset_ops.Dataset.from_tensors([x]).repeat(x))
59 .flat_map(lambda x: dataset_ops.Dataset.from_tensor_slices(x)
60 .flat_map(lambda y: dataset_ops.Dataset.from_tensors(y)
80 .flat_map(lambda x: dataset_ops.Dataset.from_tensor_slices(x)
81 .flat_map(lambda y: dataset_ops.Dataset.from_tensors(y)
110 .flat_map(lambda d: dataset_ops.Dataset.from_tensors(d["foo"])
135 dataset_ops.Dataset.range(10).map(_map_fn).flat_map(_flat_map_fn)
H A Dcache_dataset_op_test.py208 dataset = dataset_ops.Dataset.range(3).flat_map(
H A Ddataset_constructor_op_test.py379 dataset = dataset.flat_map(
456 dataset = dataset.flat_map(
/external/tensorflow/tensorflow/contrib/data/python/ops/
H A Dbatching.py103 return dataset.flat_map(map_func=unbatch_map)
H A Ddataset_ops.py517 def flat_map(self, map_func): member in class:Dataset
550 identical results = to @{tf.data.Dataset.flat_map}. In general,
/external/tensorflow/tensorflow/python/grappler/
H A Ddatasets_test.py320 dataset = dataset.flat_map(make_dataset(test_case['tensor']))
/external/tensorflow/tensorflow/python/data/ops/
H A Ddataset_ops.py434 # into a flat_map here enables multiple repetitions and/or nested
437 return id_dataset.flat_map(flat_map_fn)
794 def flat_map(self, map_func): member in class:Dataset
827 identical results = to @{tf.data.Dataset.flat_map}. In general,
1652 """See `Dataset.flat_map()` for details."""

Completed in 180 milliseconds