Searched refs:replica_id_in_sync_group (Results 1 – 13 of 13) sorted by relevance
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.distribute.-replica-context.pbtxt | 14 name: "replica_id_in_sync_group"
23 …argspec: "args=[\'self\', \'strategy\', \'replica_id_in_sync_group\'], varargs=None, keywords=None…
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.distribute.-replica-context.pbtxt | 14 name: "replica_id_in_sync_group"
23 …argspec: "args=[\'self\', \'strategy\', \'replica_id_in_sync_group\'], varargs=None, keywords=None…
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | tpu_strategy.py | 169 with _TPUReplicaContext(self, replica_id_in_sync_group=replica_id):
629 def __init__(self, strategy, replica_id_in_sync_group=None): argument
630 if replica_id_in_sync_group is None:
631 replica_id_in_sync_group = constant_op.constant(0, dtypes.int32)
633 self, strategy, replica_id_in_sync_group=replica_id_in_sync_group)
|
| D | summary_op_util.py | 45 replica_id = replica_context.replica_id_in_sync_group
|
| D | distribution_strategy_context.py | 230 _get_default_strategy(), replica_id_in_sync_group=0)
|
| D | distribute_lib.py | 1385 def __init__(self, strategy, replica_id_in_sync_group): argument
1389 self._replica_id_in_sync_group = replica_id_in_sync_group
1459 def replica_id_in_sync_group(self): member in ReplicaContext
1602 replica_id_in_sync_group=constant_op.constant(0, dtypes.int32)):
|
| D | distribute_lib_test.py | 54 replica_id_in_sync_group=constant_op.constant(0, dtypes.int32)):
|
| D | one_device_strategy.py | 215 self, strategy, replica_id_in_sync_group=zero)
|
| D | values.py | 201 replica_id = replica_context.replica_id_in_sync_group
|
| /external/tensorflow/tensorflow/contrib/distribute/python/ |
| D | strategy_test_lib.py | 223 replica_id = ds_context.get_replica_context().replica_id_in_sync_group
256 ds_context.get_replica_context().replica_id_in_sync_group)
476 lambda: ds_context.get_replica_context().replica_id_in_sync_group + 1)
|
| D | keras_optimizer_v2_test.py | 138 replica_id = ds_context.get_replica_context().replica_id_in_sync_group
|
| D | mirrored_strategy_multigpu_test.py | 1015 ds_context.get_replica_context().replica_id_in_sync_group,
1076 ds_context.get_replica_context().replica_id_in_sync_group,
1129 ds_context.get_replica_context().replica_id_in_sync_group,
1577 replica_id = ds_context.get_replica_context().replica_id_in_sync_group
|
| D | parameter_server_strategy_test.py | 64 replica_id = ds_context.get_replica_context().replica_id_in_sync_group
|