Home
last modified time | relevance | path

Searched refs:replica_context (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/python/distribute/
Dsummary_op_util.py40 replica_context = distribution_strategy_context.get_replica_context()
41 if not replica_context:
45 replica_id = replica_context.replica_id_in_sync_group
Ddistribution_strategy_context.py42 self.replica_context = replica
123 return _get_per_thread_mode().replica_context
206 return (per_thread_mode.strategy, per_thread_mode.replica_context)
Dvalues.py79 def select_for_current_replica(self, values, replica_context): argument
138 def select_for_current_replica(self, values, replica_context): argument
140 del replica_context
199 def select_for_current_replica(self, values, replica_context): argument
201 replica_id = replica_context.replica_id_in_sync_group
247 replica_context = distribution_strategy_context.get_replica_context()
248 if replica_context:
250 self._values, replica_context)
499 replica_context = distribution_strategy_context.get_replica_context()
500 if replica_context:
[all …]
Ddistribute_lib_test.py78 replica_context = ds_context.get_replica_context()
79 self.assertTrue(replica_context is not None)
84 self.assertEqual("foo", replica_context.merge_call(None, test_arg="foo"))
Ddistribute_lib.py128 if context.replica_context is replica_ctx: return
130 if context.replica_context is None:
/external/tensorflow/tensorflow/python/keras/utils/
Dmetrics_utils.py100 replica_context = distribution_strategy_context.get_replica_context()
101 if replica_context is None: # if in cross replica context already
122 result_t = replica_context.merge_call(
/external/tensorflow/tensorflow/python/training/
Dmoving_averages.py94 replica_context = distribution_strategy_context.get_replica_context()
95 if replica_context:
103 return replica_context.merge_call(merge_fn, args=(variable, value))
/external/tensorflow/tensorflow/contrib/optimizer_v2/
Doptimizer_v2.py637 replica_context = distribute_ctx.get_replica_context()
638 if replica_context is None:
646 self._per_graph_state = replica_context.merge_call(lambda _: {})