| /aosp_15_r20/external/tensorflow/tensorflow/python/distribute/ |
| H A D | packed_distributed_variable.py | 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
318 def scatter_sub(self, sparse_delta, use_locking=False, name=None): argument
322 def scatter_add(self, sparse_delta, use_locking=False, name=None): argument
326 def scatter_mul(self, sparse_delta, use_locking=False, name=None): argument
[all …]
|
| H A D | tpu_values.py | 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
474 sparse_delta, argument
492 def scatter_sub(self, var, sparse_delta, use_locking=False, name=None): argument
497 def scatter_add(self, var, sparse_delta, use_locking=False, name=None): argument
[all …]
|
| H A D | values_v2.py | 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
199 def batch_scatter_update(self, sparse_delta, use_locking=False, name=None): argument
|
| H A D | sharded_variable.py | 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
683 def batch_scatter_update(self, sparse_delta, use_locking=False, name=None): argument
|
| H A D | values_util.py | 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 | 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
1661 def scatter_sub(self, var, sparse_delta, use_locking=False, name=None): argument
1665 def scatter_add(self, var, sparse_delta, use_locking=False, name=None): argument
1669 def scatter_mul(self, var, sparse_delta, use_locking=False, name=None): argument
[all …]
|
| H A D | tpu_util.py | 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 | resource_variable_ops.py | 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
1130 def batch_scatter_update(self, sparse_delta, use_locking=False, name=None): argument
2186 def scatter_sub(self, sparse_delta, use_locking=False, name=None): argument
2191 def scatter_add(self, sparse_delta, use_locking=False, name=None): argument
[all …]
|
| H A D | variables.py | 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
748 def batch_scatter_update(self, sparse_delta, use_locking=False, name=None): argument
2100 def scatter_sub(self, sparse_delta, use_locking=False, name=None): argument
2124 def scatter_add(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 | 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
283 def scatter_update(self, sparse_delta, use_locking=False, name=None): argument
287 def batch_scatter_update(self, sparse_delta, use_locking=False, name=None): argument
|