| /aosp_15_r20/external/tensorflow/tensorflow/python/distribute/ |
| H A D | tpu_values.py | 159 def assign_sub(self, value, use_locking=False, name=None, read_value=True): argument
165 def assign_add(self, value, use_locking=False, name=None, read_value=True): argument
171 def assign(self, value, use_locking=False, name=None, read_value=True): argument
177 def scatter_sub(self, sparse_delta, use_locking=False, name=None): argument
183 def scatter_add(self, sparse_delta, use_locking=False, name=None): argument
189 def scatter_mul(self, sparse_delta, use_locking=False, name=None): argument
195 def scatter_div(self, sparse_delta, use_locking=False, name=None): argument
201 def scatter_min(self, sparse_delta, use_locking=False, name=None): argument
207 def scatter_max(self, sparse_delta, use_locking=False, name=None): argument
213 def scatter_update(self, sparse_delta, use_locking=False, name=None): argument
[all …]
|
| H A D | packed_distributed_variable.py | 147 def assign_sub(self, delta, use_locking=None, name=None, read_value=True): argument
156 def assign_add(self, delta, use_locking=None, name=None, read_value=True): argument
165 def assign(self, value, use_locking=None, name=None, read_value=True): argument
174 def scatter_sub(self, sparse_delta, use_locking=False, name=None): argument
182 def scatter_add(self, sparse_delta, use_locking=False, name=None): argument
190 def scatter_mul(self, sparse_delta, use_locking=False, name=None): argument
198 def scatter_div(self, sparse_delta, use_locking=False, name=None): argument
206 def scatter_min(self, sparse_delta, use_locking=False, name=None): argument
214 def scatter_max(self, sparse_delta, use_locking=False, name=None): argument
222 def scatter_update(self, sparse_delta, use_locking=False, name=None): argument
[all …]
|
| H A D | values_v2.py | 159 def assign_sub(self, delta, use_locking=None, name=None, read_value=True): argument
163 def assign_add(self, delta, use_locking=None, name=None, read_value=True): argument
167 def assign(self, value, use_locking=None, name=None, read_value=True): argument
171 def scatter_sub(self, sparse_delta, use_locking=False, name=None): argument
175 def scatter_add(self, sparse_delta, use_locking=False, name=None): argument
179 def scatter_mul(self, sparse_delta, use_locking=False, name=None): argument
183 def scatter_div(self, sparse_delta, use_locking=False, name=None): argument
187 def scatter_min(self, sparse_delta, use_locking=False, name=None): argument
191 def scatter_max(self, sparse_delta, use_locking=False, name=None): argument
195 def scatter_update(self, sparse_delta, use_locking=False, name=None): argument
[all …]
|
| H A D | sharded_variable.py | 537 def assign(self, value, use_locking=None, name=None, read_value=True): argument
542 def assign_add(self, delta, use_locking=False, name=None, read_value=True): argument
548 def assign_sub(self, delta, use_locking=False, name=None, read_value=True): argument
613 def scatter_add(self, sparse_delta, use_locking=False, name=None): argument
623 def scatter_div(self, sparse_delta, use_locking=False, name=None): argument
633 def scatter_max(self, sparse_delta, use_locking=False, name=None): argument
643 def scatter_min(self, sparse_delta, use_locking=False, name=None): argument
653 def scatter_mul(self, sparse_delta, use_locking=False, name=None): argument
663 def scatter_sub(self, sparse_delta, use_locking=False, name=None): argument
673 def scatter_update(self, sparse_delta, use_locking=False, name=None): argument
[all …]
|
| H A D | values_util.py | 134 def on_write_assign(var, value, use_locking=False, name=None, read_value=True): argument
144 def on_write_assign_add(var, value, use_locking=False, name=None, argument
155 def on_write_assign_sub(var, value, use_locking=False, name=None, argument
221 def scatter_sub(var, sparse_delta, use_locking=False, name=None): argument
230 def scatter_add(var, sparse_delta, use_locking=False, name=None): argument
239 def scatter_mul(var, sparse_delta, use_locking=False, name=None): argument
248 def scatter_div(var, sparse_delta, use_locking=False, name=None): argument
257 def scatter_min(var, sparse_delta, use_locking=False, name=None): argument
266 def scatter_max(var, sparse_delta, use_locking=False, name=None): argument
275 def scatter_update(var, sparse_delta, use_locking=False, name=None): argument
|
| H A D | values.py | 768 def assign_sub(self, value, use_locking=False, name=None, read_value=True): argument
781 def assign_add(self, value, use_locking=False, name=None, read_value=True): argument
794 def assign(self, value, use_locking=False, name=None, read_value=True): argument
807 def scatter_sub(self, sparse_delta, use_locking=False, name=None): argument
816 def scatter_add(self, sparse_delta, use_locking=False, name=None): argument
825 def scatter_mul(self, sparse_delta, use_locking=False, name=None): argument
834 def scatter_div(self, sparse_delta, use_locking=False, name=None): argument
843 def scatter_min(self, sparse_delta, use_locking=False, name=None): argument
852 def scatter_max(self, sparse_delta, use_locking=False, name=None): argument
861 def scatter_update(self, sparse_delta, use_locking=False, name=None): argument
[all …]
|
| H A D | tpu_replicated_variable.py | 222 def assign(self, value, use_locking=False, name=None, read_value=True): argument
240 def assign_sub(self, value, use_locking=False, name=None, read_value=True): argument
258 def assign_add(self, value, use_locking=False, name=None, read_value=True): argument
|
| H A D | tpu_util.py | 92 def assign_fn(var, value, use_locking=False, name=None, read_value=True): argument
111 …def scatter_xxx_fn(var, sparse_delta, use_locking=False, name=None): # pylint: disable=missing-do… argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ |
| H A D | state_ops.py | 133 def assign_sub(ref, value, use_locking=None, name=None): argument
205 def assign_add(ref, value, use_locking=None, name=None): argument
277 def assign(ref, value, validate_shape=None, use_locking=None, name=None): argument
383 def scatter_update(ref, indices, updates, use_locking=True, name=None): argument
437 def scatter_nd_update(ref, indices, updates, use_locking=True, name=None): argument
499 def scatter_add(ref, indices, updates, use_locking=False, name=None): argument
551 def scatter_nd_add(ref, indices, updates, use_locking=False, name=None): argument
614 def scatter_sub(ref, indices, updates, use_locking=False, name=None): argument
668 def scatter_nd_sub(ref, indices, updates, use_locking=False, name=None): argument
732 def scatter_mul(ref, indices, updates, use_locking=False, name=None): argument
[all …]
|
| H A D | resource_variable_ops.py | 845 def assign_sub(self, delta, use_locking=None, name=None, read_value=True): argument
873 def assign_add(self, delta, use_locking=None, name=None, read_value=True): argument
908 def assign(self, value, use_locking=None, name=None, read_value=True): argument
960 def scatter_sub(self, sparse_delta, use_locking=False, name=None): argument
984 def scatter_add(self, sparse_delta, use_locking=False, name=None): argument
1008 def scatter_max(self, sparse_delta, use_locking=False, name=None): argument
1033 def scatter_min(self, sparse_delta, use_locking=False, name=None): argument
1058 def scatter_mul(self, sparse_delta, use_locking=False, name=None): argument
1082 def scatter_div(self, sparse_delta, use_locking=False, name=None): argument
1106 def scatter_update(self, sparse_delta, use_locking=False, name=None): argument
[all …]
|
| H A D | variables.py | 580 def assign(self, value, use_locking=False, name=None, read_value=True): argument
598 def assign_add(self, delta, use_locking=False, name=None, read_value=True): argument
616 def assign_sub(self, delta, use_locking=False, name=None, read_value=True): argument
634 def scatter_sub(self, sparse_delta, use_locking=False, name=None): argument
650 def scatter_add(self, sparse_delta, use_locking=False, name=None): argument
666 def scatter_max(self, sparse_delta, use_locking=False, name=None): argument
683 def scatter_min(self, sparse_delta, use_locking=False, name=None): argument
700 def scatter_mul(self, sparse_delta, use_locking=False, name=None): argument
716 def scatter_div(self, sparse_delta, use_locking=False, name=None): argument
732 def scatter_update(self, sparse_delta, use_locking=False, name=None): argument
[all …]
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/keras/mixed_precision/ |
| H A D | autocast_variable.py | 214 use_locking=None, argument
247 def assign(self, value, use_locking=None, name=None, read_value=True): argument
251 def assign_add(self, delta, use_locking=None, name=None, read_value=True): argument
255 def assign_sub(self, delta, use_locking=None, name=None, read_value=True): argument
259 def scatter_sub(self, sparse_delta, use_locking=False, name=None): argument
263 def scatter_add(self, sparse_delta, use_locking=False, name=None): argument
267 def scatter_max(self, sparse_delta, use_locking=False, name=None): argument
271 def scatter_min(self, sparse_delta, use_locking=False, name=None): argument
275 def scatter_mul(self, sparse_delta, use_locking=False, name=None): argument
279 def scatter_div(self, sparse_delta, use_locking=False, name=None): argument
[all …]
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/framework/ |
| H A D | ref_var.cc | 23 int value_index, bool use_locking, bool validate_shape, in AssignRefVariable()
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/training/ |
| H A D | gradient_descent.py | 31 def __init__(self, learning_rate, use_locking=False, name="GradientDescent"): argument
|
| H A D | proximal_gradient_descent.py | 38 l2_regularization_strength=0.0, use_locking=False, argument
|
| H A D | momentum.py | 117 use_locking=False, name="Momentum", use_nesterov=False): argument
|
| H A D | adadelta.py | 107 use_locking=False, name="Adadelta"): argument
|
| H A D | proximal_adagrad.py | 41 use_locking=False, name="ProximalAdagrad"): argument
|
| H A D | rmsprop.py | 141 use_locking=False, argument
|
| H A D | adagrad_da.py | 50 use_locking=False, argument
|
| H A D | adagrad_test.py | 37 use_locking=False, argument
|
| H A D | adagrad.py | 110 use_locking=False, name="Adagrad"): argument
|
| H A D | ftrl.py | 45 use_locking=False, argument
|
| H A D | adam.py | 111 use_locking=False, argument
|
| H A D | sync_replicas_optimizer.py | 154 use_locking=False, argument
|