Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/python/distribute/
Ddistribute_lib_test.py131 t.assertIs(ds_context._get_default_strategy(), ds_context.get_strategy())
166 self.assertIs(dist, ds_context.get_strategy())
187 self.assertIs(dist, ds_context.get_strategy())
202 self.assertIs(dist, ds_context.get_strategy())
218 self.assertIs(dist, ds_context.get_strategy())
235 self.assertIs(dist, ds_context.get_strategy())
267 self.assertIs(dist, ds_context.get_strategy())
274 self.assertIs(dist2, ds_context.get_strategy())
301 self.assertIs(dist, ds_context.get_strategy())
304 self.assertIs(dist, ds_context.get_strategy())
[all …]
Ddistribution_strategy_context.py181 def get_strategy(): function
213 return get_strategy() is not _get_default_strategy()
306 current_strategy = get_strategy()
354 get_distribution_strategy = get_strategy
/external/tensorflow/tensorflow/python/keras/optimizer_v2/
Dutils.py94 if isinstance(distribute_ctx.get_strategy(),
115 if isinstance(distribute_ctx.get_strategy(),
136 if isinstance(distribute_ctx.get_strategy(),
Doptimizer_v2.py406 self._distribution_strategy = distribute_ctx.get_strategy()
652 strategy = distribute_ctx.get_strategy()
897 strategy = distribute_ctx.get_strategy()
/external/tensorflow/tensorflow/python/keras/distribute/
Ddistributed_file_utils.py88 strategy = distribution_strategy_context.get_strategy()
111 strategy = distribution_strategy_context.get_strategy()
Ddistributed_training_utils.py53 strategy = ds_context.get_strategy()
/external/tensorflow/tensorflow/python/training/
Dslot_creator.py150 distribution_strategy = distribution_strategy_context.get_strategy()
209 distribution_strategy = distribution_strategy_context.get_strategy()
Dsession_manager.py194 strategy = distribution_strategy_context.get_strategy()
Dsync_replicas_optimizer.py261 distribution_strategy = distribution_strategy_context.get_strategy()
Doptimizer.py533 num_replicas = distribute_ctx.get_strategy().num_replicas_in_sync
821 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.pbtxt100 name: "get_strategy"
/external/tensorflow/tensorflow/python/training/experimental/
Dloss_scale_test.py42 default_strategy_fn = distribution_strategy_context.get_strategy
121 distribution = distribution_strategy_context.get_strategy()
Dloss_scale_optimizer_test.py44 default_strategy_fn = distribution_strategy_context.get_strategy
/external/tensorflow/tensorflow/python/ops/
Dstateful_random_ops.py453 self._distribution_strategy = ds_context.get_strategy()
465 strat_name = type(ds_context.get_strategy()).__name__
615 return ds_context.get_strategy().extended.update(
Dcontrol_flow_util_v2.py280 _is_tpu_strategy(distribution_strategy_context.get_strategy())):
/external/tensorflow/tensorflow/python/keras/mixed_precision/
Dloss_scale_benchmark.py41 return distribution_strategy_context.get_strategy() # The default strategy
Dloss_scale_optimizer.py344 distribution = distribution_strategy_context.get_strategy()
798 strategy = distribution_strategy_context.get_strategy()
1174 strategy = distribution_strategy_context.get_strategy()
Dloss_scale_optimizer_test.py53 default_strategy_fn = distribution_strategy_context.get_strategy
790 distribution_strategy_context.get_strategy(), var, opt, 2)()
856 distribution_strategy_context.get_strategy(), var, opt, 2)()
926 distribution_strategy_context.get_strategy(), var, opt, 2)()
/external/tensorflow/tensorflow/python/keras/layers/
Dnormalization.py298 distribution_strategy_context.get_strategy().extended,
473 with distribution_strategy_context.get_strategy(
507 with distribution_strategy_context.get_strategy(
511 with distribution_strategy_context.get_strategy(
/external/tensorflow/tensorflow/python/keras/engine/
Dinput_layer.py118 strategy = distribution_strategy_context.get_strategy()
Dtraining.py306 self._distribution_strategy = ds_context.get_strategy()
727 return self._distribution_strategy or ds_context.get_strategy()
2581 strategy = ds_context.get_strategy()
Dbase_layer_utils.py801 self._distribute_strategy = distribution_strategy_context.get_strategy()
/external/tensorflow/tensorflow/python/keras/utils/
Dlosses_utils.py329 distribution_strategy_context.get_strategy().num_replicas_in_sync)
Dmetrics_utils.py74 strategy = distribution_strategy_context.get_strategy()

12