Searched refs:non_slot_devices (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/contrib/optimizer_v2/ |
| D | optimizer_v2.py | 231 def _copy_with_dynamic_hyper(self, hyper, distribution, non_slot_devices): argument
245 ret._non_slot_devices = non_slot_devices
931 non_slot_devices = distribution.extended.non_slot_devices(var_list)
933 self._hyper, distribution, non_slot_devices)
990 non_slot_devices, finish, group=False)
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.distribute.-strategy-extended.pbtxt | 62 name: "non_slot_devices"
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.distribute.-strategy-extended.pbtxt | 62 name: "non_slot_devices"
|
| /external/tensorflow/tensorflow/python/training/ |
| D | optimizer.py | 694 non_slot_devices = distribution.extended.non_slot_devices(var_list)
696 non_slot_devices, finish, args=(self, update_ops), group=False)
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | one_device_strategy.py | 186 def non_slot_devices(self, var_list): member in OneDeviceExtended
|
| D | distribute_lib.py | 1309 def non_slot_devices(self, var_list): member in DistributionStrategyExtended
1648 def non_slot_devices(self, var_list): member in _DefaultDistributionExtended
|
| D | parameter_server_strategy.py | 528 def non_slot_devices(self, var_list): member in ParameterServerStrategyExtended
|
| D | tpu_strategy.py | 581 def non_slot_devices(self, var_list): member in TPUExtended
|
| D | mirrored_strategy.py | 784 def non_slot_devices(self, var_list): member in MirroredExtended
|