Searched refs:caching_allocator_alloc (Results 1 – 3 of 3) sorted by relevance
118 caching_allocator_alloc
98 def caching_allocator_alloc(size, device: Union[Device, int] = None, stream=None): function
186 mem = torch.cuda.caching_allocator_alloc(size)3027 mem = torch.cuda.caching_allocator_alloc(1024)