Searched refs:owned_allocator_ (Results 1 – 2 of 2) sorted by relevance
266 std::unique_ptr<se::DeviceMemoryAllocator> owned_allocator_; variable
216 owned_allocator_(std::move(allocator)), in PjRtStreamExecutorClient()223 if (owned_allocator_ != nullptr) { in PjRtStreamExecutorClient()224 allocator_ = owned_allocator_.get(); in PjRtStreamExecutorClient()