Searched refs:gradient_aggregator (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/python/keras/optimizer_v2/ |
| D | optimizer_v2.py | 321 gradient_aggregator=None, argument
411 if gradient_aggregator is None:
412 gradient_aggregator = optimizer_utils.all_reduce_sum_gradients
413 self.gradient_aggregator = gradient_aggregator
485 return self.gradient_aggregator(grads_and_vars)
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.keras.optimizers.-optimizer.pbtxt | 28 …argspec: "args=[\'self\', \'name\', \'gradient_aggregator\', \'gradient_transformers\'], varargs=N…
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.keras.optimizers.-optimizer.pbtxt | 28 …argspec: "args=[\'self\', \'name\', \'gradient_aggregator\', \'gradient_transformers\'], varargs=N…
|
| D | tensorflow.optimizers.-optimizer.pbtxt | 28 …argspec: "args=[\'self\', \'name\', \'gradient_aggregator\', \'gradient_transformers\'], varargs=N…
|
| /external/tensorflow/ |
| D | RELEASE.md | 606 * `Optimizer.__init__` now accepts a `gradient_aggregator` to allow for customization
|