Searched defs:allocator_env (Results 1 – 1 of 1) sorted by relevance
48 const char* allocator_env = std::getenv("TF_GPU_ALLOCATOR"); in UseCudaMallocAllocator() local55 const char* allocator_env = std::getenv("TF_GPU_ALLOCATOR"); in UseCudaMemoryGuardAllocator() local62 const char* allocator_env = std::getenv("TF_GPU_ALLOCATOR"); in UseCudaMallocAsyncAllocator() local