Searched refs:host_memory_allocator_ (Results 1 – 3 of 3) sorted by relevance
48 host_memory_allocator_(host_memory_allocator) { in GPUDeviceContext()64 return host_memory_allocator_; in host_memory_allocator()102 Allocator* host_memory_allocator_; variable
280 return host_memory_allocator_.get(); in host_memory_allocator()344 std::unique_ptr<tensorflow::Allocator> host_memory_allocator_; variable
215 host_memory_allocator_(std::move(host_memory_allocator)), in PjRtStreamExecutorClient()232 if (!host_memory_allocator_) { in PjRtStreamExecutorClient()233 host_memory_allocator_ = std::make_unique<CpuAllocator>(); in PjRtStreamExecutorClient()