Home
last modified time | relevance | path

Searched refs:in_cross_replica_context (Results 1 – 23 of 23) sorted by relevance

/external/tensorflow/tensorflow/python/distribute/
Dvalues.py670 if ds_context.in_cross_replica_context():
920 if ds_context.in_cross_replica_context():
1157 if (ds_context.in_cross_replica_context() and
1170 if (ds_context.in_cross_replica_context() and
1183 if (ds_context.in_cross_replica_context() and
1236 if (ds_context.in_cross_replica_context() and
1263 if ds_context.in_cross_replica_context():
1391 if (ds_context.in_cross_replica_context() and
1401 if ds_context.in_cross_replica_context():
1428 if (ds_context.in_cross_replica_context() and
[all …]
Dvalues_util.py187 if ds_context.in_cross_replica_context():
199 if ds_context.in_cross_replica_context():
212 if ds_context.in_cross_replica_context():
Ddistribute_lib_test.py130 t.assertFalse(ds_context.in_cross_replica_context())
164 self.assertFalse(ds_context.in_cross_replica_context())
185 self.assertTrue(ds_context.in_cross_replica_context())
265 self.assertTrue(ds_context.in_cross_replica_context())
449 self.assertTrue(ds_context.in_cross_replica_context())
Ddistribution_strategy_context.py156 def in_cross_replica_context(): function
Dps_values.py90 if ds_context.in_cross_replica_context():
Dinput_lib.py2064 if distribution_strategy_context.in_cross_replica_context():
2092 if distribution_strategy_context.in_cross_replica_context():
/external/tensorflow/tensorflow/python/keras/distribute/
Ddistributed_training_utils.py57 if ((not is_tpu) and strategy and ds_context.in_cross_replica_context()):
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.distribute.pbtxt88 name: "in_cross_replica_context"
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.distribute.pbtxt108 name: "in_cross_replica_context"
/external/tensorflow/tensorflow/python/training/experimental/
Dloss_scaling_gradient_tape.py166 if distribution_strategy_context.in_cross_replica_context():
Dloss_scale_optimizer.py178 if distribution_strategy_context.in_cross_replica_context():
/external/tensorflow/tensorflow/python/distribute/coordinator/
Dcluster_coordinator_test.py925 self.assertFalse(distribution_strategy_context.in_cross_replica_context())
927 self.assertTrue(distribution_strategy_context.in_cross_replica_context())
936 distribution_strategy_context.in_cross_replica_context())
/external/tensorflow/tensorflow/python/training/
Dmoving_averages.py183 assert distribution_strategy_context.in_cross_replica_context(), (
Doptimizer.py571 if distribute_ctx.in_cross_replica_context():
/external/tensorflow/tensorflow/python/keras/utils/
Dmetrics_utils.py82 and not distribution_strategy_context.in_cross_replica_context()):
/external/tensorflow/tensorflow/python/keras/engine/
Dtraining_v1.py357 if distribution_strategy_context.in_cross_replica_context():
1064 distribution_strategy_context.in_cross_replica_context()):
1146 distribution_strategy_context.in_cross_replica_context()):
1201 distribution_strategy_context.in_cross_replica_context()):
Dbase_layer_utils.py250 distribution_strategy_context.in_cross_replica_context() and
Dbase_layer_v1.py1184 ds_context.in_cross_replica_context() and
/external/tensorflow/tensorflow/python/ops/
Dnn_impl.py436 if ds.has_strategy() and ds.in_cross_replica_context():
479 if ds.has_strategy() and ds.in_cross_replica_context():
Dstateful_random_ops.py610 if ds_context.in_cross_replica_context():
/external/tensorflow/tensorflow/python/eager/
Dfunction.py3166 (parent_graph, device_functions, colocation_stack, in_cross_replica_context,
3170 colocation_stack, in_cross_replica_context, variable_policy,
3223 in_cross_replica_context = False
3225 …in_cross_replica_context = (strategy_stack[-1].replica_context is None) # pylint: disable=protect…
3236 in_cross_replica_context, variable_policy, xla_context_id)
/external/tensorflow/tensorflow/python/keras/mixed_precision/
Dloss_scale_optimizer.py703 if distribution_strategy_context.in_cross_replica_context():
/external/tensorflow/tensorflow/python/keras/optimizer_v2/
Doptimizer_v2.py646 if distribute_ctx.in_cross_replica_context():