| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/serialization/ |
| D | parallel_map_dataset_serialization_test.py | 51 def _map_fn(x, y, z): function
55 _map_fn, num_parallel_calls=3).repeat(self._num_epochs))
62 def _map_fn(x, y, z): function
66 _map_fn, num_parallel_calls=3).repeat(self._num_epochs).prefetch(5))
77 def _map_fn(x): function
82 _map_fn, num_parallel_calls=2).prefetch(2)
|
| D | map_and_batch_dataset_serialization_test.py | 50 def _map_fn(x): function
56 map_func=_map_fn,
76 def _map_fn(x): function
82 map_func=_map_fn,
|
| D | map_dataset_serialization_test.py | 52 def _map_fn(x, y, z): function
56 dataset_ops.Dataset.from_tensor_slices(components).map(_map_fn)
68 def _map_fn(x): function
72 return dataset_ops.Dataset.range(100).map(_map_fn)
|
| D | interleave_dataset_serialization_test.py | 65 def _map_fn(i): function
74 return dataset_ops.Dataset.range(10).map(_map_fn).interleave(
|
| D | filter_dataset_serialization_test.py | 59 def _map_fn(i): function
66 return dataset_ops.Dataset.range(10).map(_map_fn).filter(_filter_fn).map(
|
| D | parallel_interleave_dataset_serialization_test.py | 90 def _map_fn(i): function
99 return dataset_ops.Dataset.range(10).map(_map_fn).apply(
|
| D | flat_map_dataset_serialization_test.py | 118 def _map_fn(i): function
127 return dataset_ops.Dataset.range(10).map(_map_fn).flat_map(_flat_map_fn)
|
| /external/tensorflow/tensorflow/python/data/kernel_tests/ |
| D | flat_map_test.py | 117 def _map_fn(i): function
125 dataset = dataset_ops.Dataset.range(10).map(_map_fn).flat_map(_flat_map_fn)
134 def _map_fn(i): function
145 dataset = dataset_ops.Dataset.range(10).map(_map_fn).flat_map(_flat_map_fn)
157 def _map_fn(i): function
164 dataset = dataset_ops.Dataset.range(10).map(_map_fn).flat_map(_flat_map_fn)
|
| D | filter_test.py | 64 def _map_fn(x, y, z): function
69 _map_fn).repeat(count)
128 def _map_fn(i): function
137 dataset = dataset_ops.Dataset.range(10).map(_map_fn)
141 dataset, expected_output=[_map_fn(i * 2)[0] for i in range(5)])
|
| D | map_test.py | 166 def _map_fn(x, y, z): function
170 dataset = apply_map(dataset, _map_fn).repeat(count)
233 def _map_fn(x, y, z): function
237 dataset = apply_map(dataset, _map_fn, num_parallel_calls=num_parallel_calls)
397 def _map_fn(x): function
401 return apply_map(dataset_ops.Dataset.range(10), _map_fn)
741 def _map_fn(x): function
751 dataset = apply_map(dataset, _map_fn).prefetch(buffer_size)
777 def _map_fn(x_tensor): function
784 dataset = apply_map(dataset, _map_fn)
[all …]
|
| D | iterator_cluster_test.py | 166 def _map_fn(x, y, z): function
170 dataset_ops.Dataset.from_tensor_slices(components).map(_map_fn)
|
| D | batch_test.py | 68 def _map_fn(x, y, z): function
72 _map_fn).repeat(count).batch(batch_size, drop_remainder,
|
| 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 | 58 def _map_fn(x, y, z): function
67 _map_fn).repeat(count).window(
|
| D | interleave_test.py | 234 def _map_fn(i): function
242 dataset = dataset_ops.Dataset.range(10).map(_map_fn).interleave(
|
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/ |
| D | group_by_window_test.py | 57 def _map_fn(v): function
62 math_ops.range(32)).map(_map_fn)
87 def _map_fn(v): function
92 math_ops.range(64)).map(_map_fn)
140 def _map_fn(v): function
158 128)).map(_map_fn).filter(lambda d: math_ops.equal(d["x"] % 2, 0))
|
| 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))
|
| D | parallel_interleave_test.py | 566 def _map_fn(i): function
574 dataset = dataset_ops.Dataset.range(10).map(_map_fn).apply(
|
| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| D | pfor.py | 3744 def _map_fn(i): function
3751 output = map_fn.map_fn(_map_fn, pfor_input.pfor.all_indices)
3912 def _map_fn(i): function
3920 output = map_fn.map_fn(_map_fn, pfor_input.pfor.all_indices)
|