Searched refs:call_for_each_replica (Results 1 – 25 of 30) sorted by relevance
12
| /external/tensorflow/tensorflow/python/distribute/ |
| D | mirrored_strategy_test.py | 103 distribution.extended.call_for_each_replica(run_fn)
107 result = distribution.extended.call_for_each_replica(_replica_id)
115 result = distribution.extended.call_for_each_replica(replica_squared_fn)
360 result = distribution.extended.call_for_each_replica(model_fn)
382 in_model_fn = distribution.extended.call_for_each_replica(model_fn)
390 in_model_fn = distribution.extended.call_for_each_replica(model_fn)
403 result = distribution.extended.call_for_each_replica(model_fn)
418 return distribution.extended.call_for_each_replica(model_fn)
444 distribution.extended.call_for_each_replica(model_fn)
463 distribution.extended.call_for_each_replica(model_fn)
[all …]
|
| D | mirrored_variable_test.py | 111 v2 = distribution.extended.call_for_each_replica(model_fn)
132 distribution.extended.call_for_each_replica(model_fn))
147 result = distribution.extended.call_for_each_replica(model_fn)
158 result = distribution.extended.call_for_each_replica(model_fn)
171 result = distribution.extended.call_for_each_replica(model_fn)
187 result = distribution.extended.call_for_each_replica(model_fn)
205 result = distribution.extended.call_for_each_replica(model_fn)
235 result = distribution.extended.call_for_each_replica(model_fn)
273 result = distribution.extended.call_for_each_replica(model_fn)
305 v0, v1 = distribution.extended.call_for_each_replica(create_fn)
[all …]
|
| D | strategy_test_lib.py | 88 dist.extended.call_for_each_replica(_raise_exception_fn)
102 dist.extended.call_for_each_replica(_merge_raises_fn)
166 g_v = d.extended.call_for_each_replica(grad_fn, args=(one,))
223 g_v = d.extended.call_for_each_replica(grad_fn, args=(one,))
272 output = d.extended.call_for_each_replica(run_fn)
297 d.extended.call_for_each_replica(mark_devices_fn)
304 dist.extended.call_for_each_replica(_raise_exception_fn)
306 dist.extended.call_for_each_replica(_merge_raises_fn)
308 dist.extended.call_for_each_replica(_merge_call_raises_fn)
310 dist.extended.call_for_each_replica(_merge_call_merge_raises_fn)
[all …]
|
| D | checkpoint_utils_test.py | 97 distribution.extended.call_for_each_replica(init_and_verify, args=[g])
129 distribution.extended.call_for_each_replica(init_and_verify, [g])
|
| D | moving_averages_test.py | 70 var, assign = distribution.extended.call_for_each_replica(replica_fn)
96 var, assign_op = distribution.extended.call_for_each_replica(replica_fn)
180 var, assign = distribution.extended.call_for_each_replica(replica_fn)
|
| D | step_fn.py | 102 grads_and_vars = self.distribution.extended.call_for_each_replica(
|
| D | zero_batch_test.py | 73 train_op = distribution.extended.call_for_each_replica(
75 predict_op = distribution.extended.call_for_each_replica(
|
| D | parameter_server_strategy_test.py | 210 y, z, f = d.extended.call_for_each_replica(model_fn)
259 x = d.extended.call_for_each_replica(model_fn)
360 y, z, f = d.extended.call_for_each_replica(model_fn)
407 x, y, z, train_op = d.extended.call_for_each_replica(model_fn)
480 g_v = d.extended.call_for_each_replica(grad_fn, args=(one,))
887 strategy.extended.call_for_each_replica(f)
|
| D | mirrored_run.py | 42 def call_for_each_replica(strategy, fn, args=None, kwargs=None): function
72 python_function=functools.partial(call_for_each_replica, strategy,
|
| D | multi_worker_continuous_run_test.py | 121 t_out = strategy.extended.call_for_each_replica(variable_fn)
|
| D | metrics_v1_test.py | 105 value, update = distribution.extended.call_for_each_replica(
120 value, update = distribution.extended.call_for_each_replica(
|
| D | collective_all_reduce_strategy_test.py | 151 g_v = d.extended.call_for_each_replica(grad_fn, args=[one])
201 x = distribution.extended.call_for_each_replica(model_fn)
|
| D | distribute_lib_test.py | 174 dist.extended.call_for_each_replica(run_fn)
176 dist.extended.call_for_each_replica(run_fn)
|
| /external/tensorflow/tensorflow/python/keras/distribute/ |
| D | minimize_loss_test.py | 101 distribution.extended.call_for_each_replica(
145 distribution.extended.call_for_each_replica(
196 distribution.extended.call_for_each_replica(
261 distribution.extended.call_for_each_replica(
371 distribution.extended.call_for_each_replica(
457 (train_op, loss) = distribution.extended.call_for_each_replica(
480 distribution.extended.call_for_each_replica(initial_loss))
|
| D | distributed_training_utils.py | 59 return strategy.extended.call_for_each_replica(fn, args, kwargs)
|
| D | mirrored_variable_test.py | 97 result = distribution.extended.call_for_each_replica(
|
| D | distributed_training_utils_v1.py | 768 distributed_model = strategy.extended.call_for_each_replica(
819 distributed_model = strategy.extended.call_for_each_replica(
950 grouped_session_args) = strategy.extended.call_for_each_replica(
997 grouped = strategy.extended.call_for_each_replica(
|
| D | keras_optimizer_v2_test.py | 73 train_op, optimizer = distribution.extended.call_for_each_replica(
|
| D | mirrored_strategy_test.py | 80 grads_and_vars = distribution.extended.call_for_each_replica(
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.distribute.-strategy-extended.pbtxt | 47 name: "call_for_each_replica"
|
| /external/tensorflow/tensorflow/python/training/experimental/ |
| D | loss_scale_optimizer.py | 230 update_ops = distribution.extended.call_for_each_replica(
|
| D | loss_scale.py | 411 is_finite_float = distribution.extended.call_for_each_replica(
|
| D | loss_scale_test.py | 122 return distribution.extended.call_for_each_replica(get)
|
| /external/tensorflow/tensorflow/python/keras/mixed_precision/ |
| D | loss_scale_optimizer.py | 352 is_finite_float = distribution.extended.call_for_each_replica(
730 return distribution.extended.call_for_each_replica(
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | stateful_random_ops_test.py | 656 results = strat.extended.call_for_each_replica(
683 results = strat.extended.call_for_each_replica(
|
12