Searched refs:reader_num_threads (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/ |
| H A D | make_batched_features_dataset_test.py | 169 reader_num_threads=[2, 4],
173 reader_num_threads, parser_num_threads): argument
180 reader_num_threads=reader_num_threads,
186 interleave_cycle_length=reader_num_threads)
195 reader_num_threads=reader_num_threads,
200 interleave_cycle_length=reader_num_threads)
|
| H A D | parse_example_dataset_test.py | 1171 reader_num_threads=5,
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/data/experimental/ops/ |
| H A D | readers.py | 927 reader_num_threads=None, argument
1024 if reader_num_threads is None:
1025 reader_num_threads = 1
1045 if reader_num_threads == dataset_ops.AUTOTUNE:
1048 num_parallel_calls=reader_num_threads)
1058 cycle_length=reader_num_threads,
1111 reader_num_threads=None, argument
1118 prefetch_buffer_size, reader_num_threads, parser_num_threads,
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/data/kernel_tests/ |
| H A D | tf_record_test_base.py | 44 reader_num_threads=1, argument
67 reader_num_threads=reader_num_threads,
|
| /aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v2/ |
| H A D | tensorflow.data.experimental.pbtxt | 165 …uffle_buffer_size\', \'shuffle_seed\', \'prefetch_buffer_size\', \'reader_num_threads\', \'parser_…
|
| /aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v1/ |
| H A D | tensorflow.data.experimental.pbtxt | 189 …uffle_buffer_size\', \'shuffle_seed\', \'prefetch_buffer_size\', \'reader_num_threads\', \'parser_…
|