Searched refs:get_replica_context (Results 1 – 20 of 20) sorted by relevance
64 ds_context.get_replica_context())78 replica_context = ds_context.get_replica_context()100 self.assertIs(None, ds_context.get_replica_context())136 self.assertIs(None, ds_context.get_replica_context())143 replica_ctx = ds_context.get_replica_context()
40 replica_context = distribution_strategy_context.get_replica_context()
88 def get_replica_context(): function
247 replica_context = distribution_strategy_context.get_replica_context()499 replica_context = distribution_strategy_context.get_replica_context()695 return distribution_strategy_context.get_replica_context().merge_call(800 replica_context = distribution_strategy_context.get_replica_context()969 return distribution_strategy_context.get_replica_context().merge_call(1219 replica_context = distribution_strategy_context.get_replica_context()1491 replica_context = distribution_strategy_context.get_replica_context()
635 distribution_strategy_context.get_replica_context().merge_call(653 distribution_strategy_context.get_replica_context().merge_call(
240 ds_context.get_replica_context().merge_call(lambda _: _)299 ds_context.get_replica_context().merge_call(merge_fn)327 ds_context.get_replica_context().merge_call(lambda _: _)343 ds_context.get_replica_context().merge_call(lambda _: _)353 ds_context.get_replica_context().merge_call(lambda _: _)365 ds_context.get_replica_context().merge_call(lambda _: _)380 ds_context.get_replica_context().merge_call(lambda _: _)397 ds_context.get_replica_context().merge_call(lambda _: _)414 ds_context.get_replica_context().merge_call(lambda _: _)442 ds_context.get_replica_context().merge_call(lambda _: _)[all …]
66 ds_context.get_replica_context().merge_call(_raise_exception_fn)80 ds_context.get_replica_context().merge_call(_call_raises_fn)95 ds_context.get_replica_context().merge_call(_call_merge_raises_fn)223 replica_id = ds_context.get_replica_context().replica_id_in_sync_group256 ds_context.get_replica_context().replica_id_in_sync_group)476 lambda: ds_context.get_replica_context().replica_id_in_sync_group + 1)566 ctx = ds_context.get_replica_context()571 ctx = ds_context.get_replica_context()
138 replica_id = ds_context.get_replica_context().replica_id_in_sync_group
64 replica_id = ds_context.get_replica_context().replica_id_in_sync_group
37 ds_context.get_replica_context().merge_call(lambda _: ())
64 name: "get_replica_context"
68 name: "get_replica_context"
177 best_f1 = distribution_strategy_context.get_replica_context().merge_call(
191 distribution_strategy_context.get_replica_context().merge_call(
94 replica_context = distribution_strategy_context.get_replica_context()
548 return distribute_ctx.get_replica_context().merge_call(
100 replica_context = distribution_strategy_context.get_replica_context()
637 replica_context = distribute_ctx.get_replica_context()877 return distribute_ctx.get_replica_context().merge_call(
410 return distribute_ctx.get_replica_context().merge_call(
311 return distribution_strategy_context.get_replica_context().merge_call(