Searched refs:get_cross_replica_context (Results 1 – 4 of 4) sorted by relevance
65 t.assertIs(None, ds_context.get_cross_replica_context())80 self.assertIs(None, ds_context.get_cross_replica_context())101 self.assertIs(dist, ds_context.get_cross_replica_context())137 self.assertIs(dist, ds_context.get_cross_replica_context())
126 def get_cross_replica_context(): function
188 if distribution_strategy_context.get_cross_replica_context():
105 strategy = distribution_strategy_context.get_cross_replica_context()