Searched refs:gpu_bfc_allocator (Results 1 – 3 of 3) sorted by relevance
107 "gpu_bfc_allocator.h",144 ":gpu_bfc_allocator",185 name = "gpu_bfc_allocator",187 "gpu_bfc_allocator.cc",189 hdrs = ["gpu_bfc_allocator.h"],
169 GPUBFCAllocator* gpu_bfc_allocator = in GetGPUAllocator() local172 Allocator* gpu_allocator = gpu_bfc_allocator; in GetGPUAllocator()176 gpu_bfc_allocator->SetTimingCounter(timing_counter); in GetGPUAllocator()206 gpu_bfc_allocator, sub_allocator, in GetGPUAllocator()
133 auto gpu_bfc_allocator = absl::make_unique<tensorflow::BFCAllocator>( in CreateBFCAllocator() local137 allocators.emplace_back(std::move(gpu_bfc_allocator), in CreateBFCAllocator()