Searched refs:cpu_allocator_ (Results 1 – 7 of 7) sorted by relevance
194 : SubAllocator({}, {}), cpu_allocator_(cpu_allocator) {} in CPUSubAllocator()199 return cpu_allocator_->AllocateRaw(alignment, num_bytes); in Alloc()203 cpu_allocator_->DeallocateRaw(ptr); in Free()209 return cpu_allocator_->GetMemoryType(); in GetMemoryType()213 CPUAllocator* cpu_allocator_; member in tensorflow::__anonb46ab94d0111::CPUAllocatorFactory::CPUSubAllocator
45 CHECK(cpu_allocator_) << "bad place 1"; in GetAllocator()51 return cpu_allocator_; in GetAllocator()
153 Allocator* cpu_allocator_; // not owned variable
397 cpu_allocator_(cpu_allocator), in BaseGPUDevice()
149 cpu_allocator_(cpu_allocator), in PluggableDevice()244 DCHECK(cpu_allocator_) << "CPU allocator must be set"; in GetAllocator()251 return cpu_allocator_; in GetAllocator()
81 Allocator* cpu_allocator_; variable
295 cpu_allocator_(351 return cpu_allocator_; in GetAllocator()362 Allocator* cpu_allocator_; member in tensorflow::__anone8a2de3e0311::TfCallbackDevice