Home
last modified time | relevance | path

Searched refs:replica_ctx (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/python/distribute/
Ddistribution_strategy_context.py53 def __init__(self, replica_ctx): argument
54 _ThreadMode.__init__(self, replica_ctx.strategy, None, replica_ctx)
Ddistribute_lib_test.py143 replica_ctx = ds_context.get_replica_context()
144 self.assertIs(ds_context._get_default_replica_context(), replica_ctx)
145 self.assertEqual("foo_bar", replica_ctx.merge_call(merge_fn, args=("bar",)))
Ddistribute_lib.py125 def require_replica_context(replica_ctx): argument
128 if context.replica_context is replica_ctx: return
132 if context.strategy is replica_ctx.strategy:
137 (context.strategy, replica_ctx.strategy))