Searched refs:initial_loop_values (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | training_distributed.py | 329 initial_loop_values = {}
330 initial_loop_values['loss'] = constant_op.constant(1e7)
333 initial_loop_values[name] = array_ops.zeros(tensor.shape, tensor.dtype)
348 initial_loop_values=initial_loop_values)
491 initial_loop_values = {}
492 initial_loop_values['loss'] = constant_op.constant(1e7)
495 initial_loop_values[name] = array_ops.zeros(tensor.shape, tensor.dtype)
501 initial_loop_values=initial_loop_values)
632 initial_loop_values = {}
639 initial_loop_values[name] = array_ops.zeros(shape, tensor.dtype)
[all …]
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | one_device_strategy.py | 99 initial_loop_values=None): argument
100 if initial_loop_values is None:
101 initial_loop_values = {}
102 initial_loop_values = nest.flatten(initial_loop_values)
125 cond, body, [i] + initial_loop_values, name="",
|
| D | tpu_strategy.py | 285 self, fn, multi_worker_iterator, iterations, initial_loop_values=None): argument
295 if initial_loop_values is None:
296 initial_loop_values = {}
297 initial_loop_values = nest.flatten(initial_loop_values)
343 assert isinstance(initial_loop_values, list)
344 initial_loop_values = initial_loop_values * self._num_replicas_in_sync
352 initial_loop_values)
|
| D | mirrored_strategy.py | 593 initial_loop_values=None): argument
594 if initial_loop_values is None:
595 initial_loop_values = {}
596 initial_loop_values = nest.flatten(initial_loop_values)
621 cond, body, [i] + initial_loop_values, name="",
|
| D | distribute_lib.py | 1035 initial_loop_values=None): argument
1074 fn, iterator, iterations, initial_loop_values)
1077 initial_loop_values): argument
|
| /external/tensorflow/tensorflow/contrib/distribute/python/ |
| D | minimize_loss_test.py | 426 initial_loop_values = {
433 initial_loop_values=initial_loop_values)
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.distribute.-strategy-extended.pbtxt | 59 …argspec: "args=[\'self\', \'fn\', \'iterator\', \'iterations\', \'initial_loop_values\'], varargs=…
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.distribute.-strategy-extended.pbtxt | 59 …argspec: "args=[\'self\', \'fn\', \'iterator\', \'iterations\', \'initial_loop_values\'], varargs=…
|