Searched defs:setAllocatorSettings (Results 1 – 2 of 2) sorted by relevance
/aosp_15_r20/external/pytorch/c10/cuda/ | ||
H A D | CUDAAllocatorConfig.cpp | 367 void setAllocatorSettings(const std::string& env) { in setAllocatorSettings() function |
/aosp_15_r20/external/pytorch/torch/csrc/cuda/ | ||
H A D | Module.cpp | 435 c10::cuda::CUDACachingAllocator::setAllocatorSettings( in THCPModule_cudaCachingAllocator_set_allocator_settings() member in THCPModule_cudaCachingAllocator_set_allocator_settings::c10::cuda::CUDACachingAllocator |