Searched refs:set_end_allocate_to_pool_fn (Results 1 – 3 of 3) sorted by relevance
| /external/pytorch/torch/csrc/cuda/ | ||
| D | CUDAPluggableAllocator.h | 95 void set_end_allocate_to_pool_fn( |
| D | CUDAPluggableAllocator.cpp | 92 void CUDAPluggableAllocator::set_end_allocate_to_pool_fn( in set_end_allocate_to_pool_fn() function in torch::cuda::CUDAPluggableAllocator::CUDAPluggableAllocator |
| D | Module.cpp | 1187 self.set_end_allocate_to_pool_fn(func); in registerCudaPluggableAllocator() |