Searched refs:gpu_bfc_allocator (Results 1 – 4 of 4) sorted by relevance
187 auto gpu_bfc_allocator = absl::make_unique<GPUBFCAllocator>( in GetGPUAllocator() local198 Allocator* gpu_allocator = gpu_bfc_allocator.get(); in GetGPUAllocator()203 gpu_bfc_allocator->SetTimingCounter(timing_counter); in GetGPUAllocator()218 gpu_bfc_allocator.reset(); in GetGPUAllocator()228 gpu_bfc_allocator.reset(); in GetGPUAllocator()246 gpu_bfc_allocator.release(), in GetGPUAllocator()
115 "gpu_bfc_allocator.h",153 ":gpu_bfc_allocator",221 name = "gpu_bfc_allocator",223 "gpu_bfc_allocator.cc",225 hdrs = ["gpu_bfc_allocator.h"],
255 auto gpu_bfc_allocator = std::make_unique<tensorflow::BFCAllocator>( in CreateBFCAllocator() local258 allocators.emplace_back(std::move(gpu_bfc_allocator), in CreateBFCAllocator()
58 "//tensorflow/core/common_runtime/gpu:gpu_bfc_allocator",