Searched refs:assign_add_on_device (Results 1 – 1 of 1) sorted by relevance
205 return assign_on_each_device(var, assign_add_on_device,305 def assign_add_on_device(device, variable, tensor): function