Searched refs:cpu_allocator_ (Results 1 – 6 of 6) sorted by relevance
193 : SubAllocator({}, {}), cpu_allocator_(cpu_allocator) {} in CPUSubAllocator()196 return cpu_allocator_->AllocateRaw(alignment, num_bytes); in Alloc()200 cpu_allocator_->DeallocateRaw(ptr); in Free()204 CPUAllocator* cpu_allocator_; member in tensorflow::__anonab749f280111::CPUAllocatorFactory::CPUSubAllocator
44 CHECK(cpu_allocator_) << "bad place 1"; in GetAllocator()50 return cpu_allocator_; in GetAllocator()
132 Allocator* cpu_allocator_; // not owned variable
308 cpu_allocator_(cpu_allocator), in BaseGPUDevice()
205 cpu_allocator_(cpu_allocator), in SYCLDevice()225 Allocator* cpu_allocator_; // not owned
40 return cpu_allocator_; in GetAllocator()