Searched refs:host_memory_allocator_ (Results 1 – 2 of 2) sorted by relevance
222 return host_memory_allocator_.get(); in host_memory_allocator()253 std::unique_ptr<tensorflow::Allocator> host_memory_allocator_; variable
213 host_memory_allocator_(std::move(host_memory_allocator)), in PjRtStreamExecutorClient()229 if (!host_memory_allocator_) { in PjRtStreamExecutorClient()230 host_memory_allocator_ = std::make_unique<CpuAllocator>(); in PjRtStreamExecutorClient()