Searched refs:initial_loop_values (Results 1 – 7 of 7) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/python/distribute/ |
| H A D | one_device_strategy.py | 349 initial_loop_values=None): argument
350 if initial_loop_values is None:
351 initial_loop_values = {}
352 initial_loop_values = nest.flatten(initial_loop_values)
375 cond, body, [i] + initial_loop_values, name="",
|
| H A D | tpu_strategy.py | 1049 self, fn, multi_worker_iterator, iterations, initial_loop_values=None): argument
1051 if initial_loop_values is None:
1052 initial_loop_values = {}
1053 initial_loop_values = nest.flatten(initial_loop_values)
1104 assert isinstance(initial_loop_values, list)
1105 initial_loop_values = initial_loop_values * self._num_replicas_in_sync
1113 initial_loop_values)
|
| H A D | mirrored_strategy.py | 607 initial_loop_values=None): argument
608 if initial_loop_values is None:
609 initial_loop_values = {}
610 initial_loop_values = nest.flatten(initial_loop_values)
635 cond, body, [i] + initial_loop_values, name="",
|
| H A D | distribute_lib.py | 2796 initial_loop_values=None): argument
2837 initial_loop_values)
2840 initial_loop_values): argument
|
| H A D | distribute_lib_test.py | 99 initial_loop_values=None): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/keras/engine/ |
| H A D | training_distributed_v1.py | 182 initial_loop_values = {}
183 initial_loop_values['loss'] = constant_op.constant(1e7)
186 initial_loop_values[m.name] = array_ops.zeros(tensor.shape, tensor.dtype)
190 initial_loop_values=initial_loop_values)
|
| /aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v1/ |
| H A D | tensorflow.distribute.-strategy-extended.pbtxt | 60 …argspec: "args=[\'self\', \'fn\', \'iterator\', \'iterations\', \'initial_loop_values\'], varargs=…
|