Home
last modified time | relevance | path

Searched refs:replica_id_in_sync_group (Results 1 – 25 of 30) sorted by relevance

12

/external/tensorflow/tensorflow/python/distribute/
Dstrategy_gather_test.py128 1, shape=(ctx.replica_id_in_sync_group + 1, 1))
150 1, shape=(1, ctx.replica_id_in_sync_group + 1))
178 1, shape=(1, ctx.replica_id_in_sync_group + 1))
225 return array_ops.ones(shape=(range(1, ctx.replica_id_in_sync_group + 2)))
365 1, shape=(ctx.replica_id_in_sync_group + 1, 1))
393 1, shape=(1, ctx.replica_id_in_sync_group + 1))
422 1, shape=(1, ctx.replica_id_in_sync_group + 1))
490 1, shape=(1, ctx.replica_id_in_sync_group + 1))
542 return array_ops.ones(shape=(range(1, ctx.replica_id_in_sync_group + 2)))
603 return array_ops.constant([x[ctx.replica_id_in_sync_group]])
[all …]
Ddistribute_lib.py587 replica_id_in_sync_group=0, argument
596 self._replica_id_in_sync_group = replica_id_in_sync_group
605 def replica_id_in_sync_group(self): member in ValueContext
612 .format(self.replica_id_in_sync_group, self.num_replicas_in_sync))
3028 def __init__(self, strategy, replica_id_in_sync_group): argument
3040 if not (replica_id_in_sync_group is None or
3041 tensor_util.is_tf_type(replica_id_in_sync_group) or
3042 isinstance(replica_id_in_sync_group, int)):
3045 self._replica_id_in_sync_group = replica_id_in_sync_group
3050 replica_id_in_sync_group)
[all …]
Done_device_strategy.py474 def _get_local_replica_id(self, replica_id_in_sync_group): argument
475 return replica_id_in_sync_group
483 self, strategy, replica_id_in_sync_group=0)
Dtpu_strategy.py1584 with _TPUReplicaContext(strategy, replica_id_in_sync_group=replica_id):
1661 def _get_local_replica_id(self, replica_id_in_sync_group): argument
1662 return replica_id_in_sync_group
1695 def __init__(self, strategy, replica_id_in_sync_group=0): argument
1697 self, strategy, replica_id_in_sync_group=replica_id_in_sync_group)
1703 replica_id = tensor_util.constant_value(self.replica_id_in_sync_group)
1773 self.replica_id_in_sync_group, self.num_replicas_in_sync)
1801 self.replica_id_in_sync_group, [1])
Dsummary_op_util.py41 replica_id = replica_context.replica_id_in_sync_group
Dvars_test.py355 replica_id = ctx.replica_id_in_sync_group
491 replica_id = ctx.replica_id_in_sync_group
516 replica_id = ctx.replica_id_in_sync_group
537 replica_id = ctx.replica_id_in_sync_group
558 replica_id = ctx.replica_id_in_sync_group
982 replica_id = ctx.replica_id_in_sync_group
1014 replica_id = ctx.replica_id_in_sync_group
1047 replica_id = ctx.replica_id_in_sync_group
1108 replica_id = ctx.replica_id_in_sync_group
1151 math_ops.equal(replica_ctx.replica_id_in_sync_group,
Dvalues_test.py146 per_replica.append(val * ctx.replica_id_in_sync_group)
192 return multiple_values[ctx.replica_id_in_sync_group]
213 return multiple_values[ctx.replica_id_in_sync_group]
273 worker_device_id = ctx.replica_id_in_sync_group % len(worker_devices)
302 return multiple_values[ctx.replica_id_in_sync_group]
Dstrategy_common_test.py54 return ds_context.get_replica_context().replica_id_in_sync_group
179 ds_context.get_replica_context().replica_id_in_sync_group)
192 ds_context.get_replica_context().replica_id_in_sync_group)
427 replica_id = replica_context.replica_id_in_sync_group
Ddistribution_strategy_context.py392 _get_default_strategy(), replica_id_in_sync_group=0)
Dmirrored_strategy_test.py424 return ds_context.get_replica_context().replica_id_in_sync_group
438 return ds_context.get_replica_context().replica_id_in_sync_group
771 ds_context.get_replica_context().replica_id_in_sync_group,
852 ds_context.get_replica_context().replica_id_in_sync_group,
915 ds_context.get_replica_context().replica_id_in_sync_group,
1463 replica_id = ds_context.get_replica_context().replica_id_in_sync_group
1470 replica_id = ds_context.get_replica_context().replica_id_in_sync_group
Dmirrored_run.py298 self.replica_id_in_sync_group = (
380 self.replica_id_in_sync_group), \
Ddistribute_lib_test.py73 self._container_strategy(), replica_id_in_sync_group=0):
119 def _get_local_replica_id(self, replica_id_in_sync_group): argument
120 return replica_id_in_sync_group
Dparameter_server_strategy.py688 def _get_local_replica_id(self, replica_id_in_sync_group): argument
689 return replica_id_in_sync_group
Dstrategy_test_lib.py275 replica_id = ds_context.get_replica_context().replica_id_in_sync_group
309 ds_context.get_replica_context().replica_id_in_sync_group)
631 lambda: ds_context.get_replica_context().replica_id_in_sync_group + 1,
Dcollective_all_reduce_strategy.py988 def _get_local_replica_id(self, replica_id_in_sync_group): argument
989 return (replica_id_in_sync_group -
Dmirrored_strategy.py903 def _get_local_replica_id(self, replica_id_in_sync_group): argument
904 return replica_id_in_sync_group
Dcollective_all_reduce_strategy_test.py720 return replica_ctx.replica_id_in_sync_group, replica_ctx._replica_id
768 .replica_id_in_sync_group, dtypes.float32)
Dparameter_server_strategy_v2_test.py365 ).replica_id_in_sync_group
398 ).replica_id_in_sync_group
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.distribute.experimental.-value-context.pbtxt10 name: "replica_id_in_sync_group"
15 …argspec: "args=[\'self\', \'replica_id_in_sync_group\', \'num_replicas_in_sync\'], varargs=None, k…
Dtensorflow.distribute.-replica-context.pbtxt15 name: "replica_id_in_sync_group"
24 …argspec: "args=[\'self\', \'strategy\', \'replica_id_in_sync_group\'], varargs=None, keywords=None…
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.distribute.-replica-context.pbtxt15 name: "replica_id_in_sync_group"
24 …argspec: "args=[\'self\', \'strategy\', \'replica_id_in_sync_group\'], varargs=None, keywords=None…
/external/tensorflow/tensorflow/python/tpu/tests/
Dtpu_embedding_v2_valid_input_test.py143 core_id = ctx.replica_id_in_sync_group
/external/tensorflow/tensorflow/python/training/experimental/
Dloss_scale_test.py113 .replica_id_in_sync_group)
/external/tensorflow/tensorflow/python/ops/numpy_ops/g3doc/
DTensorFlow_NumPy_Keras_and_Distribution_Strategy.ipynb261 " replica_id = tf.distribute.get_replica_context().replica_id_in_sync_group\n",
/external/tensorflow/tensorflow/python/ops/
Dstateful_random_ops.py206 return rctx.replica_id_in_sync_group

12