Searched refs:split_batch_by (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/contrib/distribute/python/ |
| D | input_lib_test.py | 39 devices, split_batch_by): argument
57 split_batch_by)
66 split_batch_by=None): argument
69 input_type, dataset_fn, worker_device_pairs, devices, split_batch_by)
162 split_batch_by=[None, 2],
164 def testBatchSplitting(self, input_type, split_batch_by): argument
170 batch_size // split_batch_by if split_batch_by else batch_size)
177 split_batch_by=split_batch_by)
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | input_lib.py | 282 def __init__(self, dataset, input_workers, split_batch_by=None): argument
309 if split_batch_by:
310 … dataset = batching._RebatchDataset(dataset, split_batch_by) # pylint: disable=protected-access
|