Searched refs:xla_allocator_ (Results 1 – 4 of 4) sorted by relevance
262 if (xla_allocator_ == nullptr) { in GetAllocatorLocked()266 xla_allocator_ = XlaDeviceAllocatorState::GetOrCreateXlaDeviceAllocator( in GetAllocatorLocked()269 return xla_allocator_; in GetAllocatorLocked()
183 se::DeviceMemoryAllocator* xla_allocator_; variable
193 xla_allocator_(xla_allocator), in XlaComputationLaunchContext()274 xla_allocator_); in PopulateInputs()282 xla_allocator_); in PopulateInputs()
225 Allocator* xla_allocator_ TF_GUARDED_BY(mu_) = nullptr; // Not owned.