Searched refs:copy_tensor_or_indexed_slices_to_device (Results 1 – 3 of 3) sorted by relevance
89 result = cross_device_utils.copy_tensor_or_indexed_slices_to_device(104 result = cross_device_utils.copy_tensor_or_indexed_slices_to_device(119 result = cross_device_utils.copy_tensor_or_indexed_slices_to_device(
205 return cross_device_utils.copy_tensor_or_indexed_slices_to_device(211 cross_device_utils.copy_tensor_or_indexed_slices_to_device(value, d))
592 def copy_tensor_or_indexed_slices_to_device(value, device): function