Searched refs:move_python_tensors_to_device (Results 1 – 3 of 3) sorted by relevance
34 move_python_tensors_to_device,80 move_python_tensors_to_device(84 inputs += move_python_tensors_to_device(87 inputs += move_python_tensors_to_device(
34 move_python_tensors_to_device,113 move_python_tensors_to_device(117 inputs += move_python_tensors_to_device(120 inputs += move_python_tensors_to_device(
165 def move_python_tensors_to_device(python_tensors, device): function