Searched refs:distribution_strategy_context (Results 1 – 25 of 35) sorted by relevance
12
29 distribution_strategy_context = LazyLoader( variable65 distribution_strategy_context.get_strategy_and_replica_context())81 distribution_strategy_context.get_strategy_and_replica_context())99 distribution_strategy_context.get_strategy_and_replica_context())
28 from tensorflow.python.distribute import distribution_strategy_context115 if not distribution_strategy_context.has_strategy():172 …self._context = distribution_strategy_context._CrossReplicaThreadMode( # pylint: disable=protecte…864 if distribution_strategy_context.has_strategy():1387 …self._thread_context = distribution_strategy_context._InReplicaThreadMode( # pylint: disable=prot…1447 …distribution_strategy_context._CrossReplicaThreadMode(self._strategy)) # pylint: disable=protecte…1554 if distribution_strategy_context.has_strategy():1689 if distribution_strategy_context.has_strategy():1702 _push_per_thread_mode = distribution_strategy_context._push_per_thread_mode # pylint: disable=prot…1703 _get_per_thread_mode = distribution_strategy_context._get_per_thread_mode # pylint: disable=protec…[all …]
22 from tensorflow.python.distribute import distribution_strategy_context40 replica_context = distribution_strategy_context.get_replica_context()
28 from tensorflow.python.distribute import distribution_strategy_context247 replica_context = distribution_strategy_context.get_replica_context()414 if not distribution_strategy_context.has_strategy():418 current_strategy = distribution_strategy_context.get_strategy()427 if not distribution_strategy_context.has_strategy():499 replica_context = distribution_strategy_context.get_replica_context()566 if distribution_strategy_context.in_cross_replica_context():666 if distribution_strategy_context.in_cross_replica_context():695 return distribution_strategy_context.get_replica_context().merge_call(723 if distribution_strategy_context.in_cross_replica_context():[all …]
26 from tensorflow.python.distribute import distribution_strategy_context619 if distribution_strategy_context.in_cross_replica_context():624 distribution = distribution_strategy_context.get_strategy()635 distribution_strategy_context.get_replica_context().merge_call(645 if distribution_strategy_context.in_cross_replica_context():653 distribution_strategy_context.get_replica_context().merge_call(
25 from tensorflow.python.distribute import distribution_strategy_context
22 from tensorflow.python.distribute import distribution_strategy_context as ds_context
127 "distribution_strategy_context.py",
42 from tensorflow.python.distribute import distribution_strategy_context124 distribution_strategy = distribution_strategy_context.get_strategy()161 distribution_strategy = distribution_strategy_context.get_strategy()
24 from tensorflow.python.distribute import distribution_strategy_context188 if distribution_strategy_context.get_cross_replica_context():191 distribution_strategy_context.get_replica_context().merge_call(
20 from tensorflow.python.distribute import distribution_strategy_context94 replica_context = distribution_strategy_context.get_replica_context()105 strategy = distribution_strategy_context.get_cross_replica_context()
22 from tensorflow.python.distribute.distribution_strategy_context import *
24 from tensorflow.python.distribute import distribution_strategy_context189 strategy = distribution_strategy_context.get_strategy()
22 from tensorflow.python.distribute import distribution_strategy_context263 distribution_strategy = distribution_strategy_context.get_strategy()
21 from tensorflow.python.distribute import distribution_strategy_context180 distribution_strategy_context.get_strategy().num_replicas_in_sync)240 distribution_strategy_context.get_strategy().num_replicas_in_sync)
27 from tensorflow.python.distribute import distribution_strategy_context100 replica_context = distribution_strategy_context.get_replica_context()
21 from tensorflow.python.distribute import distribution_strategy_context177 best_f1 = distribution_strategy_context.get_replica_context().merge_call(
21 from tensorflow.python.distribute import distribution_strategy_context409 with distribution_strategy_context.get_strategy(417 with distribution_strategy_context.get_strategy(481 if distribution_strategy_context.in_cross_replica_context():482 strategy = distribution_strategy_context.get_strategy()675 if distribution_strategy_context.in_cross_replica_context():676 strategy = distribution_strategy_context.get_strategy()
39 from tensorflow.python.distribute.distribution_strategy_context import *
20 from tensorflow.python.distribute import distribution_strategy_context as ds_context
53 from tensorflow.python.distribute import distribution_strategy_context350 distribution_strategy_context._get_default_strategy, # pylint: disable=protected-access
26 from tensorflow.python.distribute import distribution_strategy_context as ds_context
21 from tensorflow.python.distribute import distribution_strategy_context194 distribution_strategy_context.get_strategy().num_replicas_in_sync)
28 from tensorflow.python.distribute import distribution_strategy_context237 if distribution_strategy_context.has_strategy():242 if distribution_strategy_context.in_cross_replica_context():244 distribution_strategy_context.get_strategy())1973 distribution_strategy_context.in_cross_replica_context()):
25 from tensorflow.python.distribute import distribution_strategy_context735 if distribution_strategy_context.has_strategy():