Searched defs:global_batch_size (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/python/data/experimental/ops/ |
| H A D | distribute.py | 365 def batch_sizes_for_worker(global_batch_size, num_workers, argument
|
| /aosp_15_r20/external/pytorch/test/distributed/ |
| H A D | test_c10d_common.py | 338 global_batch_size, argument
363 global_batch_size, argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/ |
| H A D | rebatch_dataset_test.py | 38 def _test(self, global_batch_size, num_workers, num_replicas_per_worker, argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ |
| H A D | nn_impl.py | 413 global_batch_size=None): argument
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/distributed/ |
| H A D | distributed_test.py | 5226 global_batch_size, argument
5249 global_batch_size, argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/distribute/ |
| H A D | distribute_lib.py | 524 def get_per_replica_batch_size(self, global_batch_size): argument
|