Home
last modified time | relevance | path

Searched defs:setAllocatorSettings (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/c10/cuda/
H A DCUDAAllocatorConfig.cpp367 void setAllocatorSettings(const std::string& env) { in setAllocatorSettings() function
/aosp_15_r20/external/pytorch/torch/csrc/cuda/
H A DModule.cpp435 c10::cuda::CUDACachingAllocator::setAllocatorSettings( in THCPModule_cudaCachingAllocator_set_allocator_settings() member in THCPModule_cudaCachingAllocator_set_allocator_settings::c10::cuda::CUDACachingAllocator