Searched refs:validate_cuda_device (Results 1 – 2 of 2) sorted by relevance
18 from torch.serialization import validate_cuda_device185 validate_cuda_device(map_location)
379 def validate_cuda_device(location): function