| /external/tensorflow/tensorflow/python/data/kernel_tests/ |
| D | filter_test.py | 61 def _map_fn(x, y, z): function
66 _map_fn).repeat(count)
125 def _map_fn(i): function
134 dataset = dataset_ops.Dataset.range(10).map(_map_fn)
138 dataset, expected_output=[_map_fn(i * 2)[0] for i in range(5)])
196 def _map_fn(i): function
203 return dataset_ops.Dataset.range(10).map(_map_fn).filter(_filter_fn).map(
244 def _map_fn(i): function
251 return dataset_ops.Dataset.range(10).map(_map_fn).filter(_filter_fn).map(
|
| D | flat_map_test.py | 118 def _map_fn(i): function
126 dataset = dataset_ops.Dataset.range(10).map(_map_fn).flat_map(_flat_map_fn)
135 def _map_fn(i): function
146 dataset = dataset_ops.Dataset.range(10).map(_map_fn).flat_map(_flat_map_fn)
158 def _map_fn(i): function
165 dataset = dataset_ops.Dataset.range(10).map(_map_fn).flat_map(_flat_map_fn)
282 def _map_fn(i): function
291 return dataset_ops.Dataset.range(10).map(_map_fn).flat_map(_flat_map_fn)
|
| D | interleave_test.py | 238 def _map_fn(i): function
246 dataset = dataset_ops.Dataset.range(10).map(_map_fn).interleave(
433 def _map_fn(i): function
442 return dataset_ops.Dataset.range(10).map(_map_fn).interleave(
493 def _map_fn(i): function
502 return dataset_ops.Dataset.range(10).map(_map_fn).interleave(
|
| D | group_by_window_test.py | 53 def _map_fn(v): function
59 math_ops.range(32)).map(_map_fn)
84 def _map_fn(v): function
90 math_ops.range(64)).map(_map_fn)
138 def _map_fn(v): function
156 128)).map(_map_fn).filter(lambda d: math_ops.equal(d["x"] % 2, 0))
|
| D | map_test.py | 138 def _map_fn(x, y, z): function
142 dataset = apply_map(dataset, _map_fn).repeat(count)
205 def _map_fn(x, y, z): function
209 dataset = apply_map(dataset, _map_fn, num_parallel_calls=num_parallel_calls)
369 def _map_fn(x): function
373 return apply_map(dataset_ops.Dataset.range(10), _map_fn)
744 def _map_fn(x): function
754 dataset = apply_map(dataset, _map_fn).prefetch(buffer_size)
780 def _map_fn(x_tensor): function
787 dataset = apply_map(dataset, _map_fn)
[all …]
|
| D | iterator_cluster_test.py | 162 def _map_fn(x, y, z): function
166 dataset_ops.Dataset.from_tensor_slices(components).map(_map_fn)
|
| D | iterator_test.py | 86 def _map_fn(x, y, z): function
90 dataset_ops.Dataset.from_tensor_slices(components).map(_map_fn)
113 def _map_fn(x, y, z): function
118 .map(_map_fn).repeat(14))
141 def _map_fn(x, y, z): function
146 .map(_map_fn).repeat(14))
|
| D | window_test.py | 55 def _map_fn(x, y, z): function
64 _map_fn).repeat(count).window(
|
| D | batch_test.py | 73 def _map_fn(x, y, z): function
77 _map_fn).repeat(count).batch(batch_size, drop_remainder,
|
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/optimization/ |
| D | filter_parallelization_test.py | 71 def _map_fn(x, y, z): function
76 _map_fn).repeat(count)
148 def _map_fn(i): function
157 dataset = dataset_ops.Dataset.range(10).map(_map_fn)
163 dataset, expected_output=[_map_fn(i * 2)[0] for i in range(5)])
270 def _map_fn(i): function
277 dataset = dataset_ops.Dataset.range(10).map(_map_fn)
|
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/ |
| D | map_and_batch_test.py | 60 def _map_fn(x, y, z): function
67 map_func=_map_fn,
266 def _map_fn(x, y, z): function
270 1000).apply(batching.map_and_batch(_map_fn, batch_size=100))
440 def _map_fn(x): function
447 map_func=_map_fn,
474 def _map_fn(x): function
481 map_func=_map_fn,
|
| D | parallel_interleave_test.py | 565 def _map_fn(i): function
573 dataset = dataset_ops.Dataset.range(10).map(_map_fn).apply(
812 def _map_fn(i): function
821 return dataset_ops.Dataset.range(10).map(_map_fn).apply(
|
| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| D | pfor.py | 3993 def _map_fn(i): function
4000 output = map_fn.map_fn(_map_fn, pfor_input.pfor.all_indices)
4161 def _map_fn(i): function
4169 output = map_fn.map_fn(_map_fn, pfor_input.pfor.all_indices)
|