Searched refs:_distributed_apply (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/python/training/experimental/ |
| D | loss_scale_optimizer.py | 189 self._distributed_apply, args=(grads_and_vars, global_step, name))
191 def _distributed_apply(self, member in MixedPrecisionLossScaleOptimizer
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | step_fn.py | 108 return self._optimizer._distributed_apply( # pylint: disable=protected-access
|
| /external/tensorflow/tensorflow/python/keras/distribute/ |
| D | mirrored_strategy_test.py | 84 …update_ops = optimizer._distributed_apply(distribution, grads_and_vars) # pylint: disable=protect…
|
| /external/tensorflow/tensorflow/python/training/ |
| D | optimizer.py | 577 self._distributed_apply, args=(grads_and_vars, global_step, name))
650 def _distributed_apply(self, member in Optimizer
|
| /external/tensorflow/tensorflow/python/keras/optimizer_v2/ |
| D | optimizer_v2.py | 670 functools.partial(self._distributed_apply, apply_state=apply_state),
676 def _distributed_apply(self, distribution, grads_and_vars, name, apply_state): member in OptimizerV2
|