Searched refs:host_allocator_ (Results 1 – 1 of 1) sorted by relevance
153 host_allocator_ = GPUProcessState::singleton()->GetGpuHostAllocator(0); in GPUDeviceTestHelper()158 Allocator* host_allocator() { return host_allocator_; } in host_allocator()170 Allocator* host_allocator_; member in tensorflow::GPUDeviceTestHelper