Home
last modified time | relevance | path

Searched refs:replication_mode (Results 1 – 18 of 18) sorted by relevance

/external/tensorflow/tensorflow/python/distribute/
Ddistribute_lib.py364 replication_mode=InputReplicationMode.PER_WORKER): argument
399 if replication_mode != InputReplicationMode.PER_WORKER:
401 "Input replication mode not supported: %r" % replication_mode)
404 input_fn, replication_mode=replication_mode)
988 def _make_input_fn_iterator(self, input_fn, replication_mode): argument
1577 replication_mode=InputReplicationMode.PER_WORKER): argument
Done_device_strategy.py85 replication_mode=distribute_lib.InputReplicationMode.PER_WORKER): argument
Dcollective_all_reduce_strategy.py338 replication_mode=distribute_lib.InputReplicationMode.PER_WORKER): argument
Dparameter_server_strategy.py262 replication_mode=distribute_lib.InputReplicationMode.PER_WORKER): argument
Dtpu_strategy.py265 replication_mode=distribute_lib.InputReplicationMode.PER_WORKER): argument
Dmirrored_strategy.py576 replication_mode=distribute_lib.InputReplicationMode.PER_WORKER): argument
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.distribute.-strategy.pbtxt51 …argspec: "args=[\'self\', \'input_fn\', \'replication_mode\'], varargs=None, keywords=None, defaul…
Dtensorflow.distribute.experimental.-parameter-server-strategy.pbtxt52 …argspec: "args=[\'self\', \'input_fn\', \'replication_mode\'], varargs=None, keywords=None, defaul…
Dtensorflow.distribute.experimental.-multi-worker-mirrored-strategy.pbtxt52 …argspec: "args=[\'self\', \'input_fn\', \'replication_mode\'], varargs=None, keywords=None, defaul…
Dtensorflow.distribute.-one-device-strategy.pbtxt52 …argspec: "args=[\'self\', \'input_fn\', \'replication_mode\'], varargs=None, keywords=None, defaul…
Dtensorflow.distribute.experimental.-t-p-u-strategy.pbtxt56 …argspec: "args=[\'self\', \'input_fn\', \'replication_mode\'], varargs=None, keywords=None, defaul…
Dtensorflow.distribute.-mirrored-strategy.pbtxt52 …argspec: "args=[\'self\', \'input_fn\', \'replication_mode\'], varargs=None, keywords=None, defaul…
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.distribute.-strategy.pbtxt51 …argspec: "args=[\'self\', \'input_fn\', \'replication_mode\'], varargs=None, keywords=None, defaul…
Dtensorflow.distribute.experimental.-t-p-u-strategy.pbtxt56 …argspec: "args=[\'self\', \'input_fn\', \'replication_mode\'], varargs=None, keywords=None, defaul…
Dtensorflow.distribute.-mirrored-strategy.pbtxt52 …argspec: "args=[\'self\', \'input_fn\', \'replication_mode\'], varargs=None, keywords=None, defaul…
Dtensorflow.distribute.experimental.-multi-worker-mirrored-strategy.pbtxt52 …argspec: "args=[\'self\', \'input_fn\', \'replication_mode\'], varargs=None, keywords=None, defaul…
Dtensorflow.distribute.experimental.-parameter-server-strategy.pbtxt52 …argspec: "args=[\'self\', \'input_fn\', \'replication_mode\'], varargs=None, keywords=None, defaul…
Dtensorflow.distribute.-one-device-strategy.pbtxt52 …argspec: "args=[\'self\', \'input_fn\', \'replication_mode\'], varargs=None, keywords=None, defaul…