Searched refs:drop_final_batch (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/ |
| D | make_tf_record_dataset_test.py | 38 num_parallel_reads=1, drop_final_batch=False, parser_fn=False): argument
56 drop_final_batch=drop_final_batch,
64 drop_final_batch=drop_final_batch,
90 self._read_test(batch_size, num_epochs, 0, drop_final_batch=True)
93 self._read_test(batch_size, num_epochs, drop_final_batch=True)
97 drop_final_batch=True)
103 for drop_final_batch in [False, True]:
105 drop_final_batch=drop_final_batch)
107 parser_fn=True, drop_final_batch=drop_final_batch)
|
| D | reader_dataset_ops_test_base.py | 81 drop_final_batch=False): argument
102 drop_final_batch=drop_final_batch)
337 cycle_length, drop_final_batch, use_parser_fn): argument
365 if record_batch and not drop_final_batch:
369 interleave_cycle_length, drop_final_batch, use_parser_fn): argument
380 drop_final_batch, use_parser_fn):
|
| D | make_batched_features_dataset_test.py | 206 drop_final_batch=True)
|
| D | auto_shard_dataset_test.py | 258 drop_final_batch=True,
268 drop_final_batch=True,
|
| D | stats_dataset_ops_test.py | 417 drop_final_batch=False)
|
| /external/tensorflow/tensorflow/python/data/experimental/ops/ |
| D | readers.py | 233 drop_final_batch=False): argument
306 drop_final_batch = drop_final_batch or num_epochs is None
309 dataset = dataset.batch(batch_size, drop_remainder=drop_final_batch)
313 dataset = dataset.batch(batch_size, drop_remainder=drop_final_batch)
930 drop_final_batch=False): argument
1081 batch_size, drop_remainder=drop_final_batch or num_epochs is None)
1114 drop_final_batch=False): argument
1119 sloppy_ordering, drop_final_batch))
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.data.experimental.pbtxt | 157 …eader_num_threads\', \'parser_num_threads\', \'sloppy_ordering\', \'drop_final_batch\'], varargs=N…
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.data.experimental.pbtxt | 181 …eader_num_threads\', \'parser_num_threads\', \'sloppy_ordering\', \'drop_final_batch\'], varargs=N…
|