Searched refs:xla_allocator_ (Results 1 – 5 of 5) sorted by relevance
47 xla_allocator_(std::move(xla_allocator)), in XlaPlatformInfo()49 CHECK((device_allocator_ != nullptr) ^ (xla_allocator_.get() != nullptr)); in XlaPlatformInfo()59 return device_allocator_ ? device_allocator_ : xla_allocator_.get(); in allocator()88 std::unique_ptr<XlaAllocator> xla_allocator_; variable
250 if (xla_allocator_ == nullptr) { in GetAllocatorLocked()252 xla_allocator_ = XlaDeviceAllocatorState::GetOrCreateXlaDeviceAllocator( in GetAllocatorLocked()255 return xla_allocator_; in GetAllocatorLocked()
189 xla::DeviceMemoryAllocator* xla_allocator_; variable
206 Allocator* xla_allocator_ GUARDED_BY(mu_) = nullptr; // Not owned.
198 xla_allocator_(xla_allocator), in XlaComputationLaunchContext()