Searched refs:assign_sub_on_device (Results 1 – 1 of 1) sorted by relevance
193 return assign_on_each_device(var, assign_sub_on_device,310 def assign_sub_on_device(device, variable, tensor): function