Searched refs:move_cpp_tensors_to_device (Results 1 – 1 of 1) sorted by relevance
189 def move_cpp_tensors_to_device(cpp_tensor_stmts, device): function225 move_cpp_tensors_to_device(230 cpp_forward_target_args_stmts = move_cpp_tensors_to_device(233 cpp_forward_extra_args_stmts = move_cpp_tensors_to_device(241 cpp_other_args_stmts = move_cpp_tensors_to_device(cpp_other_args_stmts, device)