Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/python/keras/optimizer_v2/
Dutils.py38 …reduced = distribute_ctx.get_strategy().extended._replica_ctx_all_reduce( # pylint: disable=prote…
91 if isinstance(distribute_ctx.get_strategy(),
112 if isinstance(distribute_ctx.get_strategy(),
133 if isinstance(distribute_ctx.get_strategy(),
156 strategy = distribute_ctx.get_strategy()
Doptimizer_v2.py397 self._distribution_strategy = distribute_ctx.get_strategy()
654 strategy = distribute_ctx.get_strategy()
903 strategy = distribute_ctx.get_strategy()
/external/tensorflow/tensorflow/python/distribute/
Ddistribute_lib_test.py128 t.assertIs(ds_context._get_default_strategy(), ds_context.get_strategy())
163 self.assertIs(dist, ds_context.get_strategy())
184 self.assertIs(dist, ds_context.get_strategy())
199 self.assertIs(dist, ds_context.get_strategy())
215 self.assertIs(dist, ds_context.get_strategy())
232 self.assertIs(dist, ds_context.get_strategy())
264 self.assertIs(dist, ds_context.get_strategy())
271 self.assertIs(dist2, ds_context.get_strategy())
298 self.assertIs(dist, ds_context.get_strategy())
301 self.assertIs(dist, ds_context.get_strategy())
[all …]
Ddistribution_strategy_context.py234 def get_strategy(): function
266 return get_strategy() is not _get_default_strategy()
359 current_strategy = get_strategy()
407 get_distribution_strategy = get_strategy
Dmerge_call_interim.py25 strategy = distribution_strategy_context.get_strategy()
/external/tensorflow/tensorflow/python/keras/distribute/
Ddistributed_file_utils.py85 strategy = distribution_strategy_context.get_strategy()
108 strategy = distribution_strategy_context.get_strategy()
Ddistributed_training_utils.py52 strategy = ds_context.get_strategy()
/external/tensorflow/tensorflow/python/training/
Dslot_creator.py151 distribution_strategy = distribution_strategy_context.get_strategy()
210 distribution_strategy = distribution_strategy_context.get_strategy()
Dsession_manager.py213 strategy = distribution_strategy_context.get_strategy()
Dsync_replicas_optimizer.py258 distribution_strategy = distribution_strategy_context.get_strategy()
Doptimizer.py617 num_replicas = distribute_ctx.get_strategy().num_replicas_in_sync
914 distribution_strategy = distribute_ctx.get_strategy()
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.distribute.pbtxt80 name: "get_strategy"
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.distribute.pbtxt108 name: "get_strategy"
/external/tensorflow/tensorflow/python/training/experimental/
Dloss_scale_test.py38 default_strategy_fn = distribution_strategy_context.get_strategy
117 distribution = distribution_strategy_context.get_strategy()
Dloss_scale_optimizer_test.py39 default_strategy_fn = distribution_strategy_context.get_strategy
/external/tensorflow/tensorflow/python/keras/mixed_precision/
Dloss_scale_optimizer.py229 distribution = distribution_strategy_context.get_strategy()
716 strategy = distribution_strategy_context.get_strategy()
1112 strategy = distribution_strategy_context.get_strategy()
/external/tensorflow/tensorflow/python/keras/engine/
Dbase_layer_utils.py798 self._distribute_strategy = distribution_strategy_context.get_strategy()
871 self._distribute_strategy = distribution_strategy_context.get_strategy()
Dinput_layer.py115 strategy = distribution_strategy_context.get_strategy()
/external/tensorflow/tensorflow/python/ops/
Dcontrol_flow_util_v2.py288 _is_tpu_strategy(distribution_strategy_context.get_strategy())):
Dstateful_random_ops.py439 self._distribution_strategy = ds_context.get_strategy()
613 return ds_context.get_strategy().extended.update(
/external/tensorflow/tensorflow/python/keras/utils/
Dmetrics_utils.py76 strategy = distribution_strategy_context.get_strategy()
144 not distribution_strategy_context.get_strategy(
Dlosses_utils.py338 distribution_strategy_context.get_strategy().num_replicas_in_sync)
/external/tensorflow/tensorflow/python/tpu/tests/
Dtpu_embedding_v2_optimizer_test.py149 strategy = distribution_strategy_context.get_strategy()
/external/tensorflow/tensorflow/python/tpu/
Dtpu_embedding_for_serving.py113 self._strategy = distribution_strategy_context.get_strategy()
Dtpu_embedding_v1.py134 self._strategy = distribution_strategy_context.get_strategy()

12