Searched refs:get_cross_replica_context (Results 1 – 5 of 5) sorted by relevance
129 t.assertIs(None, ds_context.get_cross_replica_context())163 self.assertIs(None, ds_context.get_cross_replica_context())184 self.assertIs(dist, ds_context.get_cross_replica_context())264 self.assertIs(dist, ds_context.get_cross_replica_context())448 self.assertIs(dist, ds_context.get_cross_replica_context())
139 def get_cross_replica_context(): function
302 if distribution_strategy_context.get_cross_replica_context():
113 strategy = distribution_strategy_context.get_cross_replica_context()
403 distribution = distribution_strategy_context.get_cross_replica_context()