Searched defs:make_dataset (Results 1 – 9 of 9) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/python/data/benchmarks/ |
| H A D | interleave_benchmark.py | 44 def make_dataset(time_us, num_elements): function
80 def make_dataset(self, member in ParallelInterleaveBenchmark
|
| H A D | from_tensor_slices_benchmark.py | 116 def make_dataset(): function
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/grappler/ |
| H A D | datasets_test.py | 310 def make_dataset(tensor): function
344 def make_dataset(tensor): function
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/data/kernel_tests/ |
| H A D | cache_test.py | 184 def make_dataset(i): function
349 def make_dataset(): function
|
| H A D | tf_record_dataset_test.py | 195 def make_dataset(self, member in TFRecordDatasetCheckpointTest
|
| H A D | shuffle_test.py | 306 def make_dataset(): function
|
| H A D | snapshot_test.py | 361 def make_dataset(): function
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/data/experimental/benchmarks/ |
| H A D | map_and_batch_benchmark.py | 75 def make_dataset(map_num_calls, element_size, batch_size, batch_num_calls, function
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/ |
| H A D | dense_to_ragged_batch_test.py | 112 def testBasic(self, make_dataset, nrows, batch_size, drop_remainder): argument
|