Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/distribute/
Dvalues_util.py170 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,