Home
last modified time | relevance | path

Searched defs:replication_mode (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/distribute/
H A Ddistribute_lib.py975 replication_mode=InputReplicationMode.PER_WORKER): argument
1905 replication_mode=InputReplicationMode.PER_WORKER): argument
2284 def _make_input_fn_iterator(self, input_fn, replication_mode): argument
3668 replication_mode=InputReplicationMode.PER_WORKER): argument
H A Dcollective_all_reduce_strategy.py707 replication_mode=distribute_lib.InputReplicationMode.PER_WORKER): argument
H A Done_device_strategy.py297 replication_mode=distribute_lib.InputReplicationMode.PER_WORKER): argument
H A Dparameter_server_strategy.py357 replication_mode=distribute_lib.InputReplicationMode.PER_WORKER): argument
H A Ddistribute_lib_test.py83 replication_mode=distribute_lib.InputReplicationMode.PER_WORKER): argument
H A Dmirrored_strategy.py552 replication_mode=distribute_lib.InputReplicationMode.PER_WORKER): argument
H A Dtpu_strategy.py947 replication_mode=distribute_lib.InputReplicationMode.PER_WORKER): argument