Searched defs:_cuda_CUDAAllocator (Results 1 – 1 of 1) sorted by relevance
912 def __init__(self, allocator: torch._C._cuda_CUDAAllocator):987 def __init__(self, allocator: Optional[_cuda_CUDAAllocator] = None):