Searched refs:get_replica_context (Results 1 – 25 of 43) sorted by relevance
12
276 ctx = ds_context.get_replica_context()348 ctx = ds_context.get_replica_context()375 ctx = ds_context.get_replica_context()403 ctx = ds_context.get_replica_context()442 ctx = ds_context.get_replica_context()461 ctx = ds_context.get_replica_context()497 ctx = ds_context.get_replica_context()520 ctx = ds_context.get_replica_context()549 ctx = ds_context.get_replica_context()587 ctx = ds_context.get_replica_context()[all …]
46 replica_id = ds_context.get_replica_context().replica_id_in_sync_group107 ds_context.get_replica_context().merge_call(lambda _: _)127 ds_context.get_replica_context().merge_call(lambda _: _)144 ds_context.get_replica_context().merge_call(lambda _: _)155 ds_context.get_replica_context().merge_call(lambda _: _)168 ds_context.get_replica_context().merge_call(lambda _: _)184 ds_context.get_replica_context().merge_call(lambda _: _)202 ds_context.get_replica_context().merge_call(lambda _: _)218 ds_context.get_replica_context().merge_call(lambda _: _)257 ds_context.get_replica_context().merge_call(lambda _: _)[all …]
54 return ds_context.get_replica_context().replica_id_in_sync_group79 y = ds_context.get_replica_context().merge_call(merge_fn)179 ds_context.get_replica_context().replica_id_in_sync_group)192 ds_context.get_replica_context().replica_id_in_sync_group)285 replica_context = ds_context.get_replica_context()426 replica_context = ds_context.get_replica_context()475 rep_ctx = ds_context.get_replica_context()496 rep_ctx = ds_context.get_replica_context()529 rep_ctx = ds_context.get_replica_context()572 rep_ctx = ds_context.get_replica_context()
373 ds_context.get_replica_context().merge_call(lambda _: _)424 return ds_context.get_replica_context().replica_id_in_sync_group438 return ds_context.get_replica_context().replica_id_in_sync_group461 return ds_context.get_replica_context().merge_call(merge_fn, args=(i,))480 return ds_context.get_replica_context().merge_call(merge_fn, args=(t,))495 ds_context.get_replica_context().merge_call(merge_fn)538 ds_context.get_replica_context().merge_call(lambda _: _)556 ds_context.get_replica_context().merge_call(lambda _: _)582 c = ds_context.get_replica_context().merge_call(in_cross_replica)611 c = ds_context.get_replica_context().merge_call(in_cross_replica)[all …]
125 ds_context.get_replica_context())158 replica_context = ds_context.get_replica_context()180 self.assertIs(None, ds_context.get_replica_context())260 self.assertIs(None, ds_context.get_replica_context())464 self.assertIs(None, ds_context.get_replica_context())471 replica_ctx = ds_context.get_replica_context()485 replica_ctx = ds_context.get_replica_context()
80 ds_context.get_replica_context().merge_call(_raise_exception_fn)94 ds_context.get_replica_context().merge_call(_call_raises_fn)109 ds_context.get_replica_context().merge_call(_call_merge_raises_fn)275 replica_id = ds_context.get_replica_context().replica_id_in_sync_group309 ds_context.get_replica_context().replica_id_in_sync_group)631 lambda: ds_context.get_replica_context().replica_id_in_sync_group + 1,819 ctx = ds_context.get_replica_context()824 ctx = ds_context.get_replica_context()
36 replica_context = distribution_strategy_context.get_replica_context()
53 return distribution_strategy_context.get_replica_context().merge_call(
354 ctx = ds_context.get_replica_context()490 ctx = ds_context.get_replica_context()515 ctx = ds_context.get_replica_context()536 ctx = ds_context.get_replica_context()557 ctx = ds_context.get_replica_context()981 ctx = ds_context.get_replica_context()1013 ctx = ds_context.get_replica_context()1046 ctx = ds_context.get_replica_context()1107 ctx = ds_context.get_replica_context()1149 replica_ctx = ds_context.get_replica_context()
39 ds_context.get_replica_context().merge_call(lambda _: ())
286 replica_context = ds_context.get_replica_context()312 replica_context = ds_context.get_replica_context()
145 def get_replica_context(): function
99 replica_ctx = distribution_strategy_context.get_replica_context()136 replica_ctx = distribution_strategy_context.get_replica_context()
65 return ds_context.get_replica_context()._update( # pylint: disable=protected-access95 return ds_context.get_replica_context().merge_call(115 return ds_context.get_replica_context().merge_call(merge_fn, args=(value,))402 replica_context = ds_context.get_replica_context()1374 replica_context = ds_context.get_replica_context()
800 replica_context = distribution_strategy_context.get_replica_context()814 replica_context = distribution_strategy_context.get_replica_context()
2490 replica_context = distribution_strategy_context.get_replica_context()2506 replica_context = distribution_strategy_context.get_replica_context()2626 replica_context = distribution_strategy_context.get_replica_context()
76 name: "get_replica_context"
104 name: "get_replica_context"
42 reduced = distribute_ctx.get_replica_context().merge_call(
181 replica_context = distribution_strategy_context.get_replica_context()
112 distribution_strategy_context.get_replica_context()
204 ctx = tf.distribute.get_replica_context()
350 distribution_strategy_context.get_replica_context().merge_call(
99 replica_context = distribution_strategy_context.get_replica_context()
327 (distribution_strategy_context.get_replica_context().merge_call(474 (distribution_strategy_context.get_replica_context().merge_call(