Searched defs:num_replicas_in_sync (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/python/distribute/ |
| H A D | input_lib_test.py | 69 num_replicas_in_sync, argument
107 num_replicas_in_sync, argument
169 num_replicas_in_sync=None, argument
712 num_replicas_in_sync, distribution, argument
755 num_replicas_in_sync, distribution, argument
|
| H A D | distribute_lib.py | 496 num_replicas_in_sync=1): argument
510 def num_replicas_in_sync(self): member in InputContext
588 num_replicas_in_sync=1): argument
600 def num_replicas_in_sync(self): member in ValueContext
1570 def num_replicas_in_sync(self): member in StrategyBase
3118 def num_replicas_in_sync(self): member in ReplicaContextBase
|
| H A D | input_lib.py | 1003 num_replicas_in_sync=None, argument
1120 num_replicas_in_sync): argument
1162 def _make_rebatch_fn(self, dataset, num_workers, num_replicas_in_sync): argument
|
| H A D | input_ops.py | 26 def auto_shard_dataset(dataset, num_shards, index, num_replicas_in_sync=None): argument
|
| H A D | input_util.py | 25 num_replicas_in_sync=None, argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/distribute/v1/ |
| H A D | input_lib.py | 35 num_replicas_in_sync=None, argument
234 num_replicas_in_sync=None, argument
|