Searched refs:_get_default_replica_context (Results 1 – 3 of 3) sorted by relevance
77 _get_default_replica_context())332 def _get_default_replica_context(): function
127 t.assertIs(ds_context._get_default_replica_context(),455 self.assertIs(ds_context._get_default_replica_context(), replica_ctx)
93 …with distribution_strategy_context._get_default_replica_context(): # pylint: disable=protected-ac…