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