Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_process_state.cc46 const char* debug_allocator_str = std::getenv("TF_GPU_ALLOCATOR"); in useCudaMallocAllocator() local
47 return debug_allocator_str != nullptr && in useCudaMallocAllocator()
48 std::strcmp(debug_allocator_str, "cuda_malloc") == 0; in useCudaMallocAllocator()
52 const char* debug_allocator_str = std::getenv("TF_GPU_ALLOCATOR"); in useCudaMemoryGuardAllocator() local
53 return debug_allocator_str != nullptr && in useCudaMemoryGuardAllocator()
54 std::strcmp(debug_allocator_str, "memory_guard") == 0; in useCudaMemoryGuardAllocator()