Searched refs:_update_non_slot (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/python/distribute/ |
| D | one_device_strategy.py | 153 return self._update_non_slot(var, fn, (var,) + tuple(args), kwargs, group)
155 def _update_non_slot(self, colocate_with, fn, args, kwargs, group): member in OneDeviceExtended
|
| D | distribute_lib.py | 1251 return self._update_non_slot(colocate_with, fn, args, kwargs, group)
1253 def _update_non_slot(self, colocate_with, fn, args, kwargs, group): member in DistributionStrategyExtended
1613 return self._update_non_slot(var, fn, (var,) + tuple(args), kwargs, group)
1615 def _update_non_slot(self, colocate_with, fn, args, kwargs, should_group): member in _DefaultDistributionExtended
|
| D | parameter_server_strategy.py | 432 def _update_non_slot(self, colocate_with, fn, args, kwargs, group): member in ParameterServerStrategyExtended
|
| D | tpu_strategy.py | 584 def _update_non_slot(self, colocate_with, fn, args, kwargs, group): member in TPUExtended
|
| D | mirrored_strategy.py | 726 def _update_non_slot(self, colocate_with, fn, args, kwargs, group): member in MirroredExtended
|