Searched refs:distribution_strategy_context (Results 1 – 25 of 77) sorted by relevance
1234
52 from tensorflow.python.distribute import distribution_strategy_context88 strategy = distribution_strategy_context.get_strategy()111 strategy = distribution_strategy_context.get_strategy()
21 from tensorflow.python.distribute import distribution_strategy_context as ds_context
29 distribution_strategy_context = LazyLoader( variable98 distribution_strategy_context.get_strategy_and_replica_context())115 distribution_strategy_context.get_strategy_and_replica_context())134 distribution_strategy_context.get_strategy_and_replica_context())
21 from tensorflow.python.distribute import distribution_strategy_context166 if distribution_strategy_context.in_cross_replica_context():173 replica_context = distribution_strategy_context.get_replica_context()245 if distribution_strategy_context.has_strategy():
24 from tensorflow.python.distribute import distribution_strategy_context42 default_strategy_fn = distribution_strategy_context.get_strategy111 if not distribution_strategy_context.has_strategy():116 distribution_strategy_context.get_replica_context()121 distribution = distribution_strategy_context.get_strategy()
20 from tensorflow.python.distribute import distribution_strategy_context178 if distribution_strategy_context.in_cross_replica_context():184 replica_context = distribution_strategy_context.get_replica_context()
24 from tensorflow.python.distribute import distribution_strategy_context402 if distribution_strategy_context.has_strategy():403 distribution = distribution_strategy_context.get_cross_replica_context()
22 from tensorflow.python.distribute import distribution_strategy_context40 replica_context = distribution_strategy_context.get_replica_context()
205 from tensorflow.python.distribute import distribution_strategy_context282 …if not distribution_strategy_context.get_strategy()._scale_loss_for_estimator: # pylint: disable=…319 if not distribution_strategy_context.has_strategy():376 …self._context = distribution_strategy_context._CrossReplicaThreadMode( # pylint: disable=protecte…388 if distribution_strategy_context.has_strategy():2414 replica_context = distribution_strategy_context.get_replica_context()2917 …self._thread_context = distribution_strategy_context._InReplicaThreadMode( # pylint: disable=prot…2990 …distribution_strategy_context._CrossReplicaThreadMode(self._strategy)) # pylint: disable=protecte…3386 if distribution_strategy_context.has_strategy():3596 if distribution_strategy_context.has_strategy():[all …]
43 from tensorflow.python.distribute import distribution_strategy_context150 distribution_strategy = distribution_strategy_context.get_strategy()209 distribution_strategy = distribution_strategy_context.get_strategy()
21 from tensorflow.python.distribute import distribution_strategy_context102 replica_context = distribution_strategy_context.get_replica_context()113 strategy = distribution_strategy_context.get_cross_replica_context()183 assert distribution_strategy_context.in_cross_replica_context(), (
25 from tensorflow.python.distribute import distribution_strategy_context302 if distribution_strategy_context.get_cross_replica_context():305 distribution_strategy_context.get_replica_context().merge_call(
22 from tensorflow.python.distribute.distribution_strategy_context import *
25 from tensorflow.python.distribute import distribution_strategy_context194 strategy = distribution_strategy_context.get_strategy()
21 from tensorflow.python.distribute import distribution_strategy_context261 distribution_strategy = distribution_strategy_context.get_strategy()
27 from tensorflow.python.distribute import distribution_strategy_context74 strategy = distribution_strategy_context.get_strategy()82 and not distribution_strategy_context.in_cross_replica_context()):118 has_strategy = distribution_strategy_context.has_strategy()119 replica_context = distribution_strategy_context.get_replica_context()
21 from tensorflow.python.distribute import distribution_strategy_context329 distribution_strategy_context.get_strategy().num_replicas_in_sync)
23 from tensorflow.python.distribute import distribution_strategy_context93 …with distribution_strategy_context._get_default_replica_context(): # pylint: disable=protected-ac…
21 from tensorflow.python.distribute import distribution_strategy_context343 if distribution_strategy_context.has_strategy():344 distribution = distribution_strategy_context.get_strategy()703 if distribution_strategy_context.in_cross_replica_context():710 return distribution_strategy_context.get_replica_context().merge_call(798 strategy = distribution_strategy_context.get_strategy()1172 if not distribution_strategy_context.has_strategy():1174 strategy = distribution_strategy_context.get_strategy()
22 from tensorflow.python.distribute import distribution_strategy_context41 return distribution_strategy_context.get_strategy() # The default strategy
23 from tensorflow.python.distribute import distribution_strategy_context280 _is_tpu_strategy(distribution_strategy_context.get_strategy())):
22 from tensorflow.python.distribute import distribution_strategy_context118 strategy = distribution_strategy_context.get_strategy()
28 from tensorflow.python.distribute import distribution_strategy_context149 distribution_strategy_context.has_strategy()):151 distribution_strategy_context.get_strategy())352 if distribution_strategy_context.has_strategy():357 if distribution_strategy_context.in_cross_replica_context():359 distribution_strategy_context.get_strategy())1064 distribution_strategy_context.in_cross_replica_context()):1146 distribution_strategy_context.in_cross_replica_context()):1201 distribution_strategy_context.in_cross_replica_context()):2811 if not strategy and distribution_strategy_context.has_strategy():[all …]
24 from tensorflow.python.distribute import distribution_strategy_context250 distribution_strategy_context.in_cross_replica_context() and801 self._distribute_strategy = distribution_strategy_context.get_strategy()
20 from tensorflow.python.distribute import distribution_strategy_context as ds_context