Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/python/distribute/
Dmirrored_variable_test.py45 replica_id = ds_context.get_replica_context().replica_id_in_sync_group
106 ds_context.get_replica_context().merge_call(lambda _: _)
126 ds_context.get_replica_context().merge_call(lambda _: _)
143 ds_context.get_replica_context().merge_call(lambda _: _)
154 ds_context.get_replica_context().merge_call(lambda _: _)
167 ds_context.get_replica_context().merge_call(lambda _: _)
183 ds_context.get_replica_context().merge_call(lambda _: _)
201 ds_context.get_replica_context().merge_call(lambda _: _)
217 ds_context.get_replica_context().merge_call(lambda _: _)
256 ds_context.get_replica_context().merge_call(lambda _: _)
[all …]
Dstrategy_gather_test.py283 ctx = ds_context.get_replica_context()
355 ctx = ds_context.get_replica_context()
382 ctx = ds_context.get_replica_context()
410 ctx = ds_context.get_replica_context()
449 ctx = ds_context.get_replica_context()
468 ctx = ds_context.get_replica_context()
504 ctx = ds_context.get_replica_context()
532 ctx = ds_context.get_replica_context()
561 ctx = ds_context.get_replica_context()
599 ctx = ds_context.get_replica_context()
[all …]
Dmirrored_strategy_test.py349 ds_context.get_replica_context().merge_call(lambda _: _)
400 return ds_context.get_replica_context().replica_id_in_sync_group
414 return ds_context.get_replica_context().replica_id_in_sync_group
437 return ds_context.get_replica_context().merge_call(merge_fn, args=(i,))
456 return ds_context.get_replica_context().merge_call(merge_fn, args=(t,))
471 ds_context.get_replica_context().merge_call(merge_fn)
514 ds_context.get_replica_context().merge_call(lambda _: _)
532 ds_context.get_replica_context().merge_call(lambda _: _)
558 c = ds_context.get_replica_context().merge_call(in_cross_replica)
587 c = ds_context.get_replica_context().merge_call(in_cross_replica)
[all …]
Ddistribute_lib_test.py128 ds_context.get_replica_context())
161 replica_context = ds_context.get_replica_context()
183 self.assertIs(None, ds_context.get_replica_context())
263 self.assertIs(None, ds_context.get_replica_context())
447 self.assertIs(None, ds_context.get_replica_context())
454 replica_ctx = ds_context.get_replica_context()
468 replica_ctx = ds_context.get_replica_context()
Dstrategy_test_lib.py81 ds_context.get_replica_context().merge_call(_raise_exception_fn)
95 ds_context.get_replica_context().merge_call(_call_raises_fn)
110 ds_context.get_replica_context().merge_call(_call_merge_raises_fn)
258 replica_id = ds_context.get_replica_context().replica_id_in_sync_group
292 ds_context.get_replica_context().replica_id_in_sync_group)
614 lambda: ds_context.get_replica_context().replica_id_in_sync_group + 1,
801 ctx = ds_context.get_replica_context()
806 ctx = ds_context.get_replica_context()
Dsummary_op_util.py40 replica_context = distribution_strategy_context.get_replica_context()
Dvars_test.py336 ctx = ds_context.get_replica_context()
464 ctx = ds_context.get_replica_context()
488 ctx = ds_context.get_replica_context()
509 ctx = ds_context.get_replica_context()
530 ctx = ds_context.get_replica_context()
922 ctx = ds_context.get_replica_context()
958 ctx = ds_context.get_replica_context()
991 ctx = ds_context.get_replica_context()
1049 ctx = ds_context.get_replica_context()
1092 replica_ctx = ds_context.get_replica_context()
Ddistribution_strategy_context.py92 def get_replica_context(): function
Dvalues_util.py290 replica_context = ds_context.get_replica_context()
316 replica_context = ds_context.get_replica_context()
Dps_values.py100 replica_context = ds_context.get_replica_context()
Dtpu_strategy_test.py640 ctx = distribution_strategy_context.get_replica_context()
1125 replica_ctx = distribution_strategy_context.get_replica_context()
1159 replica_ctx = distribution_strategy_context.get_replica_context()
/external/tensorflow/tensorflow/python/kernel_tests/
Dtemplate_mirrored_strategy_test.py43 ds_context.get_replica_context().merge_call(lambda _: ())
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.distribute.pbtxt76 name: "get_replica_context"
/external/tensorflow/tensorflow/python/keras/optimizer_v2/
Dutils.py45 reduced = distribute_ctx.get_replica_context().merge_call(
/external/tensorflow/tensorflow/python/keras/distribute/
Dmirrored_variable_test.py85 ds_context.get_replica_context().merge_call(lambda _: _)
Dkeras_optimizer_v2_test.py138 replica_id = ds_context.get_replica_context().replica_id_in_sync_group
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.distribute.pbtxt96 name: "get_replica_context"
/external/tensorflow/tensorflow/python/training/experimental/
Dloss_scaling_gradient_tape.py173 replica_context = distribution_strategy_context.get_replica_context()
Dloss_scale_optimizer.py184 replica_context = distribution_strategy_context.get_replica_context()
Dloss_scale_test.py116 distribution_strategy_context.get_replica_context()
/external/tensorflow/tensorflow/python/keras/layers/
Dnormalization_v2.py145 replica_ctx = ds.get_replica_context()
/external/tensorflow/tensorflow/python/distribute/integration_test/
Dmwms_peer_failure_test.py197 ctx = tf.distribute.get_replica_context()
/external/tensorflow/tensorflow/python/training/
Dcheckpoint_utils.py305 distribution_strategy_context.get_replica_context().merge_call(
Dmoving_averages.py102 replica_context = distribution_strategy_context.get_replica_context()
/external/tensorflow/tensorflow/python/keras/engine/
Dtraining_distributed_v1.py332 (distribution_strategy_context.get_replica_context().merge_call(
479 (distribution_strategy_context.get_replica_context().merge_call(

12