Searched refs:in_cross_replica_context (Results 1 – 9 of 9) sorted by relevance
66 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())
161 def in_cross_replica_context(): function
566 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():
619 if distribution_strategy_context.in_cross_replica_context():645 if distribution_strategy_context.in_cross_replica_context():
76 name: "in_cross_replica_context"
80 name: "in_cross_replica_context"
481 if distribution_strategy_context.in_cross_replica_context():675 if distribution_strategy_context.in_cross_replica_context():
242 if distribution_strategy_context.in_cross_replica_context():1973 distribution_strategy_context.in_cross_replica_context()):
543 if distribute_ctx.in_cross_replica_context():