Searched refs:copy_tensor_or_indexed_slices_to_device (Results 1 – 3 of 3) sorted by relevance
92 result = cross_device_utils.copy_tensor_or_indexed_slices_to_device(107 result = cross_device_utils.copy_tensor_or_indexed_slices_to_device(
189 return cross_device_utils.copy_tensor_or_indexed_slices_to_device(195 cross_device_utils.copy_tensor_or_indexed_slices_to_device(value, d))
591 def copy_tensor_or_indexed_slices_to_device(value, device): function