Searched refs:_replica_id (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/python/keras/distribute/ |
| D | keras_optimizer_v2_test.py | 65 replica_id = _replica_id()
137 def _replica_id(): function
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | mirrored_variable_test.py | 44 def _replica_id(): function
199 replica_id = self.evaluate(_replica_id())
320 return math_ops.cast(_replica_id() + 1, dtype=dtypes.float32)
448 replica_id = self.evaluate(_replica_id())
|
| D | mirrored_strategy_test.py | 97 replica_id = int(self.evaluate(_replica_id()))
107 result = distribution.extended.call_for_each_replica(_replica_id)
339 replica_id_str = str(self.evaluate(_replica_id()))
1423 def _replica_id(): function
|
| D | distribute_lib.py | 255 self._replica_id = replica_id
260 _update_replica_id.current = self._replica_id
3031 def _replica_id(self): member in ReplicaContextBase
|
| D | values_util.py | 292 replica_id = replica_context._replica_id # pylint: disable=protected-access
|
| D | collective_all_reduce_strategy.py | 796 replica_context._replica_id, # pylint: disable=protected-access
|
| D | collective_all_reduce_strategy_test.py | 563 return replica_ctx.replica_id_in_sync_group, replica_ctx._replica_id
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tensor_tracer.py | 511 self._replica_id = None
596 self._replica_id = tpu_ops.tpu_replicated_input(
600 self._replica_id = 'unknown'
924 '@', self._replica_id,
1450 _flush_fun, cache_val, self._replica_id,
1453 flush_op = _flush_fun(cache_val, self._replica_id,
1920 replica_id=self._replica_id,
2040 self._replica_id = 0
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | ops_test.py | 1881 self._replica_id = replica_id
1900 with g.device("/job:worker/replica:{}".format(self._replica_id)):
1906 name="FloatOutput_{}".format(self._replica_id))
1946 name="FloatOutput_{}".format(self._replica_id))
2011 name="FloatOutput_{}".format(self._replica_id))
|