Home
last modified time | relevance | path

Searched refs:cpu_allocator_ (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/core/framework/
Dallocator.cc193 : 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
/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_device_factory.cc44 CHECK(cpu_allocator_) << "bad place 1"; in GetAllocator()
50 return cpu_allocator_; in GetAllocator()
Dgpu_device.h132 Allocator* cpu_allocator_; // not owned variable
Dgpu_device.cc308 cpu_allocator_(cpu_allocator), in BaseGPUDevice()
/external/tensorflow/tensorflow/core/common_runtime/sycl/
Dsycl_device.h205 cpu_allocator_(cpu_allocator), in SYCLDevice()
225 Allocator* cpu_allocator_; // not owned
Dsycl_device.cc40 return cpu_allocator_; in GetAllocator()