Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/distribute/
Ddistribute_lib_test.py66 t.assertFalse(ds_context.in_cross_replica_context())
81 self.assertFalse(ds_context.in_cross_replica_context())
102 self.assertTrue(ds_context.in_cross_replica_context())
138 self.assertTrue(ds_context.in_cross_replica_context())
Ddistribution_strategy_context.py161 def in_cross_replica_context(): function
Dvalues.py566 if distribution_strategy_context.in_cross_replica_context():
666 if distribution_strategy_context.in_cross_replica_context():
723 if distribution_strategy_context.in_cross_replica_context():
936 if distribution_strategy_context.in_cross_replica_context():
1103 if distribution_strategy_context.in_cross_replica_context():
1246 if distribution_strategy_context.in_cross_replica_context():
1271 if distribution_strategy_context.in_cross_replica_context():
1480 if distribution_strategy_context.in_cross_replica_context():
Dinput_lib.py619 if distribution_strategy_context.in_cross_replica_context():
645 if distribution_strategy_context.in_cross_replica_context():
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.distribute.pbtxt76 name: "in_cross_replica_context"
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.distribute.pbtxt80 name: "in_cross_replica_context"
/external/tensorflow/tensorflow/python/keras/layers/
Dnormalization.py481 if distribution_strategy_context.in_cross_replica_context():
675 if distribution_strategy_context.in_cross_replica_context():
/external/tensorflow/tensorflow/python/keras/engine/
Dtraining.py242 if distribution_strategy_context.in_cross_replica_context():
1973 distribution_strategy_context.in_cross_replica_context()):
/external/tensorflow/tensorflow/python/training/
Doptimizer.py543 if distribute_ctx.in_cross_replica_context():