get_allocator_backend (reference) in projects: external

Searched refs:get_allocator_backend (Results 1 – 4 of 4) sorted by relevance

/external/pytorch/docs/source/
D	cuda.rst	`120 get_allocator_backend`
/external/pytorch/torch/cuda/
D	memory.py	`897 def get_allocator_backend() -> str: function`
/external/pytorch/test/
D	test_cuda_multigpu.py	`41 torch.cuda.get_allocator_backend() == "cudaMallocAsync"`
D	test_cuda.py	`96 torch.cuda.get_allocator_backend() == "cudaMallocAsync"`