Searched defs:num_replicas_in_sync (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/python/distribute/ |
| D | input_ops.py | 30 def auto_shard_dataset(dataset, num_shards, index, num_replicas_in_sync=None): argument
|
| D | input_lib_test.py | 69 num_replicas_in_sync, argument
109 num_replicas_in_sync, argument
171 num_replicas_in_sync=None, argument
735 num_replicas_in_sync, distribution, argument
776 num_replicas_in_sync, distribution, argument
|
| D | input_lib.py | 67 num_replicas_in_sync=None, argument
966 num_replicas_in_sync=None, argument
1040 def _make_rebatch_fn(self, dataset, num_workers, num_replicas_in_sync): argument
1151 num_replicas_in_sync=None, argument
1428 num_replicas_in_sync=None, argument
|
| D | distribute_lib.py | 472 num_replicas_in_sync=1): argument
486 def num_replicas_in_sync(self): member in InputContext
564 num_replicas_in_sync=1): argument
576 def num_replicas_in_sync(self): member in ValueContext
1514 def num_replicas_in_sync(self): member in StrategyBase
2997 def num_replicas_in_sync(self): member in ReplicaContextBase
|