Searched refs:_distributed_apply (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/contrib/distribute/python/ |
| D | step_fn.py | 108 return self._optimizer._distributed_apply( # pylint: disable=protected-access
|
| D | mirrored_strategy_multigpu_test.py | 1411 …update_ops = optimizer._distributed_apply(distribution, grads_and_vars) # pylint: disable=protect…
|
| /external/tensorflow/tensorflow/python/keras/optimizer_v2/ |
| D | optimizer_v2.py | 411 self._distributed_apply, args=(grads_and_vars,), kwargs={"name": name})
413 def _distributed_apply(self, distribution, grads_and_vars, name): member in OptimizerV2
|
| /external/tensorflow/tensorflow/contrib/optimizer_v2/ |
| D | optimizer_v2.py | 878 self._distributed_apply, args=(filtered,),
912 def _distributed_apply(self, distribution, grads_and_vars, global_step, name): member in OptimizerV2
|
| /external/tensorflow/tensorflow/python/training/ |
| D | optimizer.py | 549 self._distributed_apply, args=(grads_and_vars, global_step, name))
619 def _distributed_apply(self, member in Optimizer
|