Searched refs:call_for_each_replica (Results 1 – 20 of 20) sorted by relevance
100 distribution.extended.call_for_each_replica(run_fn)104 result = distribution.extended.call_for_each_replica(_replica_id)251 result = distribution.extended.call_for_each_replica(model_fn)271 in_model_fn = distribution.extended.call_for_each_replica(model_fn)279 in_model_fn = distribution.extended.call_for_each_replica(model_fn)290 result = distribution.extended.call_for_each_replica(model_fn)305 distribution.extended.call_for_each_replica(model_fn)332 v2 = distribution.extended.call_for_each_replica(model_fn)347 result = distribution.extended.call_for_each_replica(model_fn)357 result = distribution.extended.call_for_each_replica(model_fn)[all …]
73 dist.extended.call_for_each_replica(_raise_exception_fn)87 dist.extended.call_for_each_replica(_merge_raises_fn)135 g_v = d.extended.call_for_each_replica(grad_fn, args=(one,))188 g_v = d.extended.call_for_each_replica(grad_fn, args=(one,))236 output = d.extended.call_for_each_replica(run_fn)261 d.extended.call_for_each_replica(mark_devices_fn)268 dist.extended.call_for_each_replica(_raise_exception_fn)270 dist.extended.call_for_each_replica(_merge_raises_fn)272 dist.extended.call_for_each_replica(_merge_call_raises_fn)274 dist.extended.call_for_each_replica(_merge_call_merge_raises_fn)[all …]
67 distribution.extended.call_for_each_replica(106 distribution.extended.call_for_each_replica(158 distribution.extended.call_for_each_replica(224 distribution.extended.call_for_each_replica(319 distribution.extended.call_for_each_replica(402 (train_op, loss) = distribution.extended.call_for_each_replica(425 distribution.extended.call_for_each_replica(initial_loss))
56 var, assign = distribution.extended.call_for_each_replica(replica_fn)82 var, assign_op = distribution.extended.call_for_each_replica(replica_fn)155 var, assign = distribution.extended.call_for_each_replica(replica_fn)
94 distribution.extended.call_for_each_replica(init_and_verify, args=[g])127 distribution.extended.call_for_each_replica(init_and_verify, [g])
239 y, z, f = d.extended.call_for_each_replica(model_fn)289 x = d.extended.call_for_each_replica(model_fn)390 y, z, f = d.extended.call_for_each_replica(model_fn)441 x, y, z, train_op = d.extended.call_for_each_replica(model_fn)522 g_v = d.extended.call_for_each_replica(grad_fn, args=(one,))924 strategy.extended.call_for_each_replica(f)
102 grads_and_vars = self.distribution.extended.call_for_each_replica(
49 distribution.extended.call_for_each_replica(
86 strategy.extended.call_for_each_replica(
74 train_op, optimizer = distribution.extended.call_for_each_replica(
101 value, update = distribution.extended.call_for_each_replica(116 value, update = distribution.extended.call_for_each_replica(
184 g_v = d.extended.call_for_each_replica(grad_fn, args=[one])268 train_op = d.extended.call_for_each_replica(model_fn)295 x = distribution.extended.call_for_each_replica(model_fn)
91 dist.extended.call_for_each_replica(run_fn)93 dist.extended.call_for_each_replica(run_fn)
499 return self._extended.call_for_each_replica(fn, args=args, kwargs=kwargs)1080 def call_for_each_replica(self, fn, args=(), kwargs=None): member in DistributionStrategyExtended
674 distributed_model = strategy.extended.call_for_each_replica(725 distributed_model = strategy.extended.call_for_each_replica(777 grouped_session_args) = strategy.extended.call_for_each_replica(824 grouped = strategy.extended.call_for_each_replica(
233 grouped_session_args) = strategy.extended.call_for_each_replica(
1975 return self._distribution_strategy.extended.call_for_each_replica(
46 name: "call_for_each_replica"
307 strategy.extended.call_for_each_replica(336 train_op = strategy.extended.call_for_each_replica(