Searched refs:assign_on_each_device (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/python/distribute/ |
| D | values_util.py | 170 def assign_on_each_device(var, assign_func, value, read_value): function
193 return assign_on_each_device(var, assign_sub_on_device,
205 return assign_on_each_device(var, assign_add_on_device,
221 return assign_on_each_device(var, assign_on_device, tensor,
|