Searched refs:torch_tensor_on_cuda_gpu_check (Results 1 – 1 of 1) sorted by relevance
511 inline bool torch_tensor_on_cuda_gpu_check(const at::Tensor& ten) { in torch_tensor_on_cuda_gpu_check() function515 inline bool torch_tensor_on_cuda_gpu_check( in torch_tensor_on_cuda_gpu_check() function517 return !ten.has_value() || torch_tensor_on_cuda_gpu_check(ten.value()); in torch_tensor_on_cuda_gpu_check()522 torch_tensor_on_cuda_gpu_check(x), \