Home
last modified time | relevance | path

Searched refs:get_replica_context (Results 1 – 20 of 20) sorted by relevance

/external/tensorflow/tensorflow/python/distribute/
Ddistribute_lib_test.py64 ds_context.get_replica_context())
78 replica_context = ds_context.get_replica_context()
100 self.assertIs(None, ds_context.get_replica_context())
136 self.assertIs(None, ds_context.get_replica_context())
143 replica_ctx = ds_context.get_replica_context()
Dsummary_op_util.py40 replica_context = distribution_strategy_context.get_replica_context()
Ddistribution_strategy_context.py88 def get_replica_context(): function
Dvalues.py247 replica_context = distribution_strategy_context.get_replica_context()
499 replica_context = distribution_strategy_context.get_replica_context()
695 return distribution_strategy_context.get_replica_context().merge_call(
800 replica_context = distribution_strategy_context.get_replica_context()
969 return distribution_strategy_context.get_replica_context().merge_call(
1219 replica_context = distribution_strategy_context.get_replica_context()
1491 replica_context = distribution_strategy_context.get_replica_context()
Dinput_lib.py635 distribution_strategy_context.get_replica_context().merge_call(
653 distribution_strategy_context.get_replica_context().merge_call(
/external/tensorflow/tensorflow/contrib/distribute/python/
Dmirrored_strategy_multigpu_test.py240 ds_context.get_replica_context().merge_call(lambda _: _)
299 ds_context.get_replica_context().merge_call(merge_fn)
327 ds_context.get_replica_context().merge_call(lambda _: _)
343 ds_context.get_replica_context().merge_call(lambda _: _)
353 ds_context.get_replica_context().merge_call(lambda _: _)
365 ds_context.get_replica_context().merge_call(lambda _: _)
380 ds_context.get_replica_context().merge_call(lambda _: _)
397 ds_context.get_replica_context().merge_call(lambda _: _)
414 ds_context.get_replica_context().merge_call(lambda _: _)
442 ds_context.get_replica_context().merge_call(lambda _: _)
[all …]
Dstrategy_test_lib.py66 ds_context.get_replica_context().merge_call(_raise_exception_fn)
80 ds_context.get_replica_context().merge_call(_call_raises_fn)
95 ds_context.get_replica_context().merge_call(_call_merge_raises_fn)
223 replica_id = ds_context.get_replica_context().replica_id_in_sync_group
256 ds_context.get_replica_context().replica_id_in_sync_group)
476 lambda: ds_context.get_replica_context().replica_id_in_sync_group + 1)
566 ctx = ds_context.get_replica_context()
571 ctx = ds_context.get_replica_context()
Dkeras_optimizer_v2_test.py138 replica_id = ds_context.get_replica_context().replica_id_in_sync_group
Dparameter_server_strategy_test.py64 replica_id = ds_context.get_replica_context().replica_id_in_sync_group
/external/tensorflow/tensorflow/python/kernel_tests/
Dtemplate_mirrored_strategy_test.py37 ds_context.get_replica_context().merge_call(lambda _: ())
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.distribute.pbtxt64 name: "get_replica_context"
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.distribute.pbtxt68 name: "get_replica_context"
/external/tensorflow/tensorflow/contrib/metrics/python/metrics/
Dclassification.py177 best_f1 = distribution_strategy_context.get_replica_context().merge_call(
/external/tensorflow/tensorflow/python/training/
Dcheckpoint_utils.py191 distribution_strategy_context.get_replica_context().merge_call(
Dmoving_averages.py94 replica_context = distribution_strategy_context.get_replica_context()
Doptimizer.py548 return distribute_ctx.get_replica_context().merge_call(
/external/tensorflow/tensorflow/python/keras/utils/
Dmetrics_utils.py100 replica_context = distribution_strategy_context.get_replica_context()
/external/tensorflow/tensorflow/contrib/optimizer_v2/
Doptimizer_v2.py637 replica_context = distribute_ctx.get_replica_context()
877 return distribute_ctx.get_replica_context().merge_call(
/external/tensorflow/tensorflow/python/keras/optimizer_v2/
Doptimizer_v2.py410 return distribute_ctx.get_replica_context().merge_call(
/external/tensorflow/tensorflow/python/ops/
Dmetrics_impl.py311 return distribution_strategy_context.get_replica_context().merge_call(