| /aosp_15_r20/external/tensorflow/tensorflow/python/data/kernel_tests/ |
| H A D | interleave_test.py | 404 def _build_dataset(): function
411 verify_fn(self, _build_dataset, num_outputs)
441 def _build_dataset(): function
445 verify_fn(self, _build_dataset, num_outputs=20)
464 def _build_dataset(): function
471 verify_fn(self, _build_dataset, num_outputs)
501 def _build_dataset(): function
505 verify_fn(self, _build_dataset, num_outputs=20)
|
| H A D | group_by_window_test.py | 376 def _build_dataset(self, components): member in GroupByWindowCheckpointTest
389 lambda: self._build_dataset(components),
393 lambda: self._build_dataset(components),
397 lambda: self._build_dataset(components),
405 def _build_dataset(self, components): member in GroupByWindowCheckpointTest
418 lambda: self._build_dataset(components),
422 lambda: self._build_dataset(components),
426 lambda: self._build_dataset(components),
|
| H A D | window_test.py | 260 def _build_dataset(self): member in WindowCheckpointTest
269 verify_fn(self, self._build_dataset, num_outputs=42)
275 def _build_dataset(self): member in WindowCheckpointTest
284 verify_fn(self, self._build_dataset, num_outputs=42)
|
| H A D | take_while_test.py | 130 def _build_dataset(self, num_elements, upper_bound): member in TakeWhileCheckpointTest
140 verify_fn(self, lambda: self._build_dataset(num_elements, upper_bound),
|
| H A D | shard_test.py | 104 def _build_dataset(self, num_elements, num_shards, index): member in ShardCheckpointTest
116 lambda: self._build_dataset(elems, num_shards, index),
|
| H A D | fixed_length_record_dataset_test.py | 233 def _build_dataset(self, num_epochs, compression_type=None): member in FixedLengthRecordDatasetCheckpointTest
245 verify_fn(self, lambda: self._build_dataset(num_epochs), num_outputs)
|
| H A D | scan_test.py | 303 def _build_dataset(self, num_elements): member in ScanCheckpointTest
315 self, lambda: self._build_dataset(num_outputs), num_outputs=num_outputs)
|
| H A D | zip_test.py | 140 def _build_dataset(self, arr): member in ZipCheckpointTest
159 verify_fn(self, lambda: self._build_dataset(elements), len(elements))
|
| H A D | bucket_by_sequence_length_test.py | 412 def _build_dataset(): function
461 dataset = _build_dataset()
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/ |
| H A D | dense_to_sparse_batch_test.py | 112 def _build_dataset(self, components): member in DenseToSparseBatchCheckpointTest
124 verify_fn(self, lambda: self._build_dataset(components), num_outputs)
|
| H A D | group_by_reducer_test.py | 199 def _build_dataset(self, components): member in GroupByReducerCheckpointTest
215 lambda: self._build_dataset(components),
|
| H A D | directed_interleave_dataset_test.py | 336 def _build_dataset(self, probs, num_samples): member in SampleFromDatasetsCheckpointTest
350 self, lambda: self._build_dataset([0.5, 0.5], 100), num_outputs=100)
|
| H A D | sql_dataset_test.py | 595 def _build_dataset(self, num_repeats): member in SqlDatasetCheckpointTest
611 verify_fn(self, lambda: self._build_dataset(num_repeats), num_outputs)
|
| H A D | parallel_interleave_test.py | 820 def _build_dataset(): function
824 verify_fn(self, _build_dataset, num_outputs=20)
|