Home
last modified time | relevance | path

Searched refs:_update_non_slot (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/python/distribute/
Done_device_strategy.py363 return self._update_non_slot(var, fn, (var,) + tuple(args), kwargs, group)
365 def _update_non_slot(self, colocate_with, fn, args, kwargs, group): member in OneDeviceExtended
Ddistribute_lib.py1628 return self._update_non_slot(colocate_with, fn, args, kwargs, group)
1630 def _update_non_slot(self, colocate_with, fn, args, kwargs, group): member in StrategyExtendedV2
2243 return self._update_non_slot(var, fn, (var,) + tuple(args), kwargs, group)
2245 def _update_non_slot(self, colocate_with, fn, args, kwargs, should_group): member in _DefaultDistributionExtended
Ddistribute_lib_test.py117 return self._update_non_slot(var, fn, (var,) + tuple(args), kwargs, group)
119 def _update_non_slot(self, colocate_with, fn, args, kwargs, group): member in _TestExtended
Dparameter_server_strategy.py521 def _update_non_slot(self, colocate_with, fn, args, kwargs, group): member in ParameterServerStrategyExtended
Dtpu_strategy.py715 def _update_non_slot(self, colocate_with, fn, args, kwargs, group): member in TPUExtended
Dmirrored_strategy.py825 def _update_non_slot(self, colocate_with, fn, args, kwargs, group): member in MirroredExtended