| /aosp_15_r20/external/tensorflow/tensorflow/python/distribute/ |
| H A D | vars_test.py | 88 update_fn = lambda: getattr(v, fn)(update_value) function
133 update_fn = lambda: getattr(v, fn)(update_value) function
183 update_fn = lambda: getattr(v, fn)(update_value) function
714 update_fn = lambda: getattr(v, fn)(update_value) function
761 update_fn = lambda: getattr(v, fn)(update_value) function
828 update_fn = lambda: getattr(v, fn)(update_value) function
871 update_fn = lambda: getattr(v, fn)(update_value) function
|
| H A D | values.py | 44 def _on_write_update_replica(var, update_fn, value, **kwargs): argument
912 def _update_cross_replica(self, update_fn, value, **kwargs): argument
928 def _update_replica(self, update_fn, value, **kwargs): argument
948 def _update(self, update_fn, value, **kwargs): argument
1097 def _update_replica(self, update_fn, value, **kwargs): argument
1213 def _update_replica(self, update_fn, value, **kwargs): argument
1465 def _update_replica(self, var, update_fn, value, **kwargs): argument
1510 def _update_replica(self, var, update_fn, value, **kwargs): argument
1634 def _update_replica(self, var, update_fn, value, **kwargs): argument
|
| H A D | tpu_replicated_variable.py | 36 def _on_device_update(update_fn, var, value, **kwargs): argument
212 def _update(self, update_fn, value, **kwargs): argument
|
| H A D | moving_averages_test.py | 152 def update_fn(v, value): function
|
| H A D | strategy_common_test.py | 276 def testDenseUpdate(self, strategy, tf_function, update_fn): argument
|
| H A D | distribute_lib_test.py | 422 def update_fn(): function
|
| H A D | packed_distributed_variable.py | 141 def _update(self, update_fn, value, **kwargs): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/training/ |
| H A D | moving_averages.py | 90 def update_fn(v, value): function
178 def _update(strategy, var, update_fn, args): argument
267 def update_fn(v, value, biased_var, local_step): function
|
| /aosp_15_r20/external/tflite-support/tensorflow_lite_support/custom_ops/kernel/sentencepiece/ |
| H A D | double_array_trie.h | 92 callback update_fn) const { in IteratePrefixMatches()
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/keras/mixed_precision/ |
| H A D | autocast_variable.py | 212 update_fn, argument
237 def _apply_update(self, update_fn, *args, **kwargs): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ |
| H A D | stateful_random_ops.py | 593 def update_fn(v): function
|