| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/serialization/ |
| D | parallel_map_dataset_serialization_test.py | 47 def _map_fn(x, y, z): function
51 _map_fn, num_parallel_calls=3).repeat(self._num_epochs))
58 def _map_fn(x, y, z): function
62 _map_fn, num_parallel_calls=3).repeat(self._num_epochs).prefetch(5))
75 def _map_fn(x): function
80 _map_fn, num_parallel_calls=2).prefetch(2)
|
| D | map_and_batch_dataset_serialization_test.py | 44 def _map_fn(x): function
50 map_func=_map_fn,
71 def _map_fn(x): function
77 map_func=_map_fn,
|
| D | numa_map_and_batch_dataset_serialization_test.py | 43 def _map_fn(x): function
49 map_func=_map_fn,
73 def _map_fn(x): function
79 map_func=_map_fn,
|
| D | map_dataset_serialization_test.py | 48 def _map_fn(x, y, z): function
52 dataset_ops.Dataset.from_tensor_slices(components).map(_map_fn)
65 def _map_fn(x): function
69 return dataset_ops.Dataset.range(100).map(_map_fn)
|
| D | filter_dataset_serialization_test.py | 53 def _map_fn(i): function
60 return dataset_ops.Dataset.range(10).map(_map_fn).filter(_filter_fn).map(
|
| D | interleave_dataset_serialization_test.py | 67 def _map_fn(i): function
76 return dataset_ops.Dataset.range(10).map(_map_fn).interleave(
|
| D | parallel_interleave_dataset_serialization_test.py | 85 def _map_fn(i): function
94 return dataset_ops.Dataset.range(10).map(_map_fn).apply(
|
| D | flat_map_dataset_serialization_test.py | 107 def _map_fn(i): function
116 return dataset_ops.Dataset.range(10).map(_map_fn).flat_map(_flat_map_fn)
|
| /external/tensorflow/tensorflow/python/data/kernel_tests/ |
| D | filter_test_base.py | 43 def _map_fn(x, y, z): function
48 _map_fn).repeat(count)
105 def _map_fn(i): function
114 dataset = dataset_ops.Dataset.range(10).map(_map_fn)
118 dataset, expected_output=[_map_fn(i * 2)[0] for i in range(5)])
|
| D | map_test.py | 93 def _map_fn(x, y, z): function
97 _map_fn).repeat(count)
158 def _map_fn(x, y, z): function
162 _map_fn, num_parallel_calls=num_parallel_calls).prefetch(
309 def _map_fn(x): function
313 return dataset_ops.Dataset.range(10).map(_map_fn)
637 def _map_fn(x): function
641 dataset = dataset_ops.Dataset.range(100).map(_map_fn).prefetch(
662 dataset = dataset_ops.Dataset.range(100).map(_map_fn).prefetch(
691 def _map_fn(x_tensor): function
[all …]
|
| D | flat_map_test.py | 109 def _map_fn(i): function
117 dataset = dataset_ops.Dataset.range(10).map(_map_fn).flat_map(_flat_map_fn)
|
| D | batch_test.py | 61 def _map_fn(x, y, z): function
65 _map_fn).repeat(count).batch(batch_size, drop_remainder)
|
| D | iterator_cluster_test.py | 164 def _map_fn(x, y, z): function
168 dataset_ops.Dataset.from_tensor_slices(components).map(_map_fn)
|
| D | interleave_test.py | 189 def _map_fn(i): function
197 dataset = dataset_ops.Dataset.range(10).map(_map_fn).interleave(
|
| 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))
140 def _map_fn(x, y, z): function
145 .map(_map_fn).repeat(14))
|
| D | window_test.py | 68 def _map_fn(x, y, z): function
77 _map_fn).repeat(count).window(
|
| D | padded_batch_test.py | 195 def _map_fn(i): function
200 _ = dataset_ops.Dataset.range(10).map(_map_fn).padded_batch(10)
|
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/ |
| D | group_by_window_test.py | 56 def _map_fn(v): function
61 math_ops.range(32)).map(_map_fn)
85 def _map_fn(v): function
90 math_ops.range(64)).map(_map_fn)
137 def _map_fn(v): function
155 128)).map(_map_fn).filter(lambda d: math_ops.equal(d["x"] % 2, 0))
|
| D | map_and_batch_test.py | 65 def _map_fn(x, y, z): function
72 map_func=_map_fn,
329 def _map_fn(x, y, z): function
333 1000).apply(batching.map_and_batch(_map_fn, batch_size=100))
|
| D | parallel_interleave_test.py | 584 def _map_fn(i): function
592 dataset = dataset_ops.Dataset.range(10).map(_map_fn).apply(
|
| /external/tensorflow/tensorflow/contrib/data/python/kernel_tests/ |
| D | slide_dataset_op_test.py | 62 def _map_fn(x, y, z): function
69 dataset_ops.Dataset.from_tensor_slices(components).map(_map_fn)
127 def _map_fn(x, y, z): function
133 dataset_ops.Dataset.from_tensor_slices(components).map(_map_fn)
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tpu_estimator.py | 3548 def _map_fn(*args): function
3578 return _map_fn
|