| /external/tensorflow/tensorflow/python/data/kernel_tests/ |
| filter_dataset_op_test.py | 44 def _map_fn(x, y, z): function in function:FilterDatasetTest.testFilterDataset
48 dataset_ops.Dataset.from_tensor_slices(components).map(_map_fn)
135 def _map_fn(i): function in function:FilterDatasetTest.testSparse
145 dataset_ops.Dataset.range(10).map(_map_fn).filter(_filter_fn).map(
155 self.assertSparseValuesEqual(actual, _map_fn(i * 2)[0])
|
| flat_map_dataset_op_test.py | 126 def _map_fn(i): function in function:FlatMapDatasetTest.testSparse
135 dataset_ops.Dataset.range(10).map(_map_fn).flat_map(_flat_map_fn)
|
| interleave_dataset_op_test.py | 181 def _map_fn(i): function in function:InterleaveDatasetTest.testSparse
190 dataset_ops.Dataset.range(10).map(_map_fn).interleave(
|
| iterator_ops_cluster_test.py | 157 def _map_fn(x, y, z): function in function:IteratorClusterTest.testImplicitDisposeParallelMapDataset
161 dataset_ops.Dataset.from_tensor_slices(components).map(_map_fn)
|
| map_dataset_op_test.py | 49 def _map_fn(x, y, z): function in function:MapDatasetTest._buildMapDataset
51 return (dataset_ops.Dataset.from_tensor_slices(components).map(_map_fn)
110 def _map_fn(x, y, z): function in function:MapDatasetTest._buildParallelMapDataset
113 .map(_map_fn, num_parallel_calls=num_parallel_calls)
472 def _map_fn(x): function in function:MapDatasetTest.testPrefetch
478 .map(_map_fn)
531 def _map_fn(x_tensor): function in function:MapDatasetTest.testMultiOutputPyFunc
538 .map(_map_fn)
|
| batch_dataset_op_test.py | 51 def _map_fn(x, y, z): function in function:BatchDatasetTest.testBatchDataset
55 dataset_ops.Dataset.from_tensor_slices(components).map(_map_fn)
367 def _map_fn(i): function in function:BatchDatasetTest.testPaddedBatchSparseError
372 _ = dataset_ops.Dataset.range(10).map(_map_fn).padded_batch(10)
|
| iterator_ops_test.py | 78 def _map_fn(x, y, z): function in function:IteratorTest.testOneShotIterator
81 iterator = (dataset_ops.Dataset.from_tensor_slices(components).map(_map_fn)
103 def _map_fn(x, y, z): function in function:IteratorTest.testOneShotIteratorCaptureByValue
107 .map(_map_fn).repeat(14).make_one_shot_iterator())
128 def _map_fn(x, y, z): function in function:IteratorTest.testOneShotIteratorInsideContainer.within_container
131 .map(_map_fn).repeat(14).make_one_shot_iterator())
|
| /external/tensorflow/tensorflow/contrib/data/python/kernel_tests/ |
| map_dataset_op_test.py | 120 def _map_fn(x): function in function:MapDatasetTest.testCaptureResourceInMapFn._build_ds
124 return dataset_ops.Dataset.range(10).map(_map_fn)
159 def _map_fn(x, y, z): function in function:MapDatasetSerializationTest._build_ds
163 dataset_ops.Dataset.from_tensor_slices(components).map(_map_fn)
176 def _map_fn(x): function in function:MapDatasetSerializationTest.testSaveStatefulFunction._build_ds
180 return dataset_ops.Dataset.range(100).map(_map_fn)
263 def _map_fn(x, y, z): function in function:ParallelMapDatasetSerializationTest._build_ds
267 _map_fn, num_parallel_calls=3).repeat(self._num_epochs))
274 def _map_fn(x, y, z): function in function:ParallelMapDatasetSerializationTest._build_ds_with_prefetch
278 _map_fn, num_parallel_calls=3).repeat(self._num_epochs).prefetch(5)
291 def _map_fn(x): function in function:ParallelMapDatasetSerializationTest.testSaveStatefulFunction._build_ds
[all...] |
| filter_dataset_op_test.py | 55 def _map_fn(i): function in function:FilterDatasetSerializationTest._build_sparse_filter
62 return dataset_ops.Dataset.range(10).map(_map_fn).filter(_filter_fn).map(
|
| flat_map_dataset_op_test.py | 107 def _map_fn(i): function in function:FlatMapDatasetSerializationTest.testSparseCore
116 return dataset_ops.Dataset.range(10).map(_map_fn).flat_map(_flat_map_fn)
|
| bucketing_test.py | 211 def _map_fn(v): function in function:BucketTest.testSingleBucket
216 dataset_ops.Dataset.from_tensor_slices(math_ops.range(32)).map(_map_fn))
246 def _map_fn(v): function in function:BucketTest.testEvenOddBuckets
251 dataset_ops.Dataset.from_tensor_slices(math_ops.range(64)).map(_map_fn))
303 def _map_fn(v): function in function:BucketTest.testEvenOddBucketsFilterOutAllOdd
321 dataset_ops.Dataset.from_tensor_slices(math_ops.range(128)).map(_map_fn)
|
| batch_dataset_op_test.py | 281 def _map_fn(i): function in function:BatchDatasetTest.testPaddedBatchAndDropRemainderSparseError
286 _ = dataset_ops.Dataset.range(10).map(_map_fn).apply(
325 def _map_fn(x, y, z): function in function:BatchDatasetTest._testBatchAndMapDatasetHelper
331 map_func=_map_fn,
|
| interleave_dataset_op_test.py | 82 def _map_fn(i): function in function:InterleaveDatasetSerializationTest.testSparseCore
91 return dataset_ops.Dataset.range(10).map(_map_fn).interleave(
681 def _map_fn(i): function in function:ParallelInterleaveDatasetTest.testSparse
689 dataset = dataset_ops.Dataset.range(10).map(_map_fn)
|
