Searched refs:call_for_each_replica (Results 1 – 25 of 26) sorted by relevance
12
103 distribution.extended.call_for_each_replica(run_fn)110 result = distribution.extended.call_for_each_replica(_replica_id)135 result = distribution.extended.call_for_each_replica(replica_squared_fn)384 result = distribution.extended.call_for_each_replica(model_fn)406 in_model_fn = distribution.extended.call_for_each_replica(model_fn)414 in_model_fn = distribution.extended.call_for_each_replica(model_fn)427 result = distribution.extended.call_for_each_replica(model_fn)442 return distribution.extended.call_for_each_replica(model_fn)468 distribution.extended.call_for_each_replica(model_fn)487 distribution.extended.call_for_each_replica(model_fn)[all …]
112 v2 = distribution.extended.call_for_each_replica(model_fn)133 distribution.extended.call_for_each_replica(model_fn))148 result = distribution.extended.call_for_each_replica(model_fn)159 result = distribution.extended.call_for_each_replica(model_fn)172 result = distribution.extended.call_for_each_replica(model_fn)188 result = distribution.extended.call_for_each_replica(model_fn)206 result = distribution.extended.call_for_each_replica(model_fn)236 result = distribution.extended.call_for_each_replica(model_fn)274 result = distribution.extended.call_for_each_replica(model_fn)306 v0, v1 = distribution.extended.call_for_each_replica(create_fn)[all …]
87 dist.extended.call_for_each_replica(_raise_exception_fn)101 dist.extended.call_for_each_replica(_merge_raises_fn)183 g_v = d.extended.call_for_each_replica(grad_fn, args=(one,))240 g_v = d.extended.call_for_each_replica(grad_fn, args=(one,))289 output = d.extended.call_for_each_replica(run_fn)314 d.extended.call_for_each_replica(mark_devices_fn)321 dist.extended.call_for_each_replica(_raise_exception_fn)323 dist.extended.call_for_each_replica(_merge_raises_fn)325 dist.extended.call_for_each_replica(_merge_call_raises_fn)327 dist.extended.call_for_each_replica(_merge_call_merge_raises_fn)[all …]
95 distribution.extended.call_for_each_replica(init_and_verify, args=[g])129 distribution.extended.call_for_each_replica(init_and_verify, [g])
66 var, assign = distribution.extended.call_for_each_replica(replica_fn)92 var, assign_op = distribution.extended.call_for_each_replica(replica_fn)176 var, assign = distribution.extended.call_for_each_replica(replica_fn)
98 grads_and_vars = self.distribution.extended.call_for_each_replica(
206 y, z, f = d.extended.call_for_each_replica(model_fn)255 x = d.extended.call_for_each_replica(model_fn)356 y, z, f = d.extended.call_for_each_replica(model_fn)403 x, y, z, train_op = d.extended.call_for_each_replica(model_fn)476 g_v = d.extended.call_for_each_replica(grad_fn, args=(one,))885 strategy.extended.call_for_each_replica(f)
70 train_op = distribution.extended.call_for_each_replica(72 predict_op = distribution.extended.call_for_each_replica(
43 def call_for_each_replica(strategy, fn, args=None, kwargs=None): function80 python_function=functools.partial(call_for_each_replica, strategy,
117 t_out = strategy.extended.call_for_each_replica(variable_fn)
114 results = strat.extended.call_for_each_replica(fn=f)142 results = strat.extended.call_for_each_replica(fn=f, args=gens)
100 value, update = distribution.extended.call_for_each_replica(115 value, update = distribution.extended.call_for_each_replica(
175 g_v = distribution.extended.call_for_each_replica(grad_fn, args=[one])226 x = distribution.extended.call_for_each_replica(model_fn)550 result = strategy.extended.call_for_each_replica(_replica_id_f32)
171 dist.extended.call_for_each_replica(run_fn)173 dist.extended.call_for_each_replica(run_fn)
918 return mirrored_run.call_for_each_replica(self._container_strategy(), fn,
492 return mirrored_run.call_for_each_replica(self._container_strategy(), fn,
1315 return self._extended.call_for_each_replica(fn, args=args, kwargs=kwargs)2843 def call_for_each_replica(self, fn, args=(), kwargs=None): member in StrategyExtendedV1
58 return strategy.extended.call_for_each_replica(fn, args, kwargs)
758 distributed_model = strategy.extended.call_for_each_replica(809 distributed_model = strategy.extended.call_for_each_replica(940 grouped_session_args) = strategy.extended.call_for_each_replica(987 grouped = strategy.extended.call_for_each_replica(
47 name: "call_for_each_replica"
227 update_ops = distribution.extended.call_for_each_replica(
377 is_finite_float = distribution.extended.call_for_each_replica(
118 return distribution.extended.call_for_each_replica(get)
230 is_finite_per_replica = distribution.extended.call_for_each_replica(650 return distribution.extended.call_for_each_replica(
86 grouped_session_args) = strategy.extended.call_for_each_replica(