Searched defs:global_batch_size (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/python/keras/integration_test/ |
| D | multi_worker_tutorial_test.py | 97 def dataset_fn(self, global_batch_size, input_context): argument
|
| /external/tensorflow/tensorflow/python/data/experimental/ops/ |
| D | distribute.py | 353 def batch_sizes_for_worker(global_batch_size, num_workers, argument
|
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/ |
| D | rebatch_dataset_test.py | 41 def _test(self, global_batch_size, num_workers, num_replicas_per_worker, argument
|
| /external/tensorflow/tensorflow/python/keras/distribute/ |
| D | keras_correctness_test_base.py | 152 def get_batch_size(global_batch_size, distribution): argument
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | nn_impl.py | 395 global_batch_size=None): argument
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | distribute_lib.py | 500 def get_per_replica_batch_size(self, global_batch_size): argument
|