Searched refs:cpu_allocator_ (Results 1 – 4 of 4) sorted by relevance
157 : SubAllocator({}, {}), cpu_allocator_(cpu_allocator) {} in CPUSubAllocator()162 return cpu_allocator_->AllocateRaw(alignment, num_bytes); in Alloc()166 cpu_allocator_->DeallocateRaw(ptr); in Free()172 CPUAllocator* cpu_allocator_; member in tensorflow::__anon29597ede0111::CPUAllocatorFactory::CPUSubAllocator
45 CHECK(cpu_allocator_) << "bad place 1"; in GetAllocator()51 return cpu_allocator_; in GetAllocator()
124 Allocator* cpu_allocator_; // not owned variable
389 cpu_allocator_(cpu_allocator), in BaseGPUDevice()