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,
|