Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/keras/layers/
Dnormalization_v2.py145 replica_ctx = ds.get_replica_context()
146 if replica_ctx:
154 y_sum = replica_ctx.all_reduce(reduce_util.ReduceOp.SUM, local_sum)
155 y_squared_sum = replica_ctx.all_reduce(reduce_util.ReduceOp.SUM,
157 global_batch_size = replica_ctx.all_reduce(reduce_util.ReduceOp.SUM,
/external/tensorflow/tensorflow/python/distribute/
Ddistribution_strategy_context.py57 def __init__(self, replica_ctx): argument
58 _ThreadMode.__init__(self, replica_ctx.strategy, None, replica_ctx)
Ddistribute_lib_test.py454 replica_ctx = ds_context.get_replica_context()
455 self.assertIs(ds_context._get_default_replica_context(), replica_ctx)
456 self.assertEqual("foo_bar", replica_ctx.merge_call(merge_fn, args=("bar",)))
468 replica_ctx = ds_context.get_replica_context()
469 replica_ctx.merge_call(merge_fn, args=("bar",))
Dtpu_strategy_test.py1125 replica_ctx = distribution_strategy_context.get_replica_context()
1126 with replica_ctx.experimental_logical_device(0):
1128 with replica_ctx.experimental_logical_device(1):
1159 replica_ctx = distribution_strategy_context.get_replica_context()
1160 with replica_ctx.experimental_logical_device(0):
1162 with replica_ctx.experimental_logical_device(1):
Dcollective_all_reduce_strategy_test.py562 replica_ctx = distribution_strategy_context.get_replica_context()
563 return replica_ctx.replica_id_in_sync_group, replica_ctx._replica_id
Ddistribute_lib.py329 def require_replica_context(replica_ctx): argument
332 if context.replica_context is replica_ctx: return
336 if context.strategy is replica_ctx.strategy:
341 (context.strategy, replica_ctx.strategy))
Dvars_test.py1092 replica_ctx = ds_context.get_replica_context()
1094 math_ops.equal(replica_ctx.replica_id_in_sync_group,