Searched refs:device_allocator_ (Results 1 – 4 of 4) sorted by relevance
48 device_allocator_(device_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()89 xla::DeviceMemoryAllocator* device_allocator_; variable
26 device_allocator_ = allocator; in set_device_allocator()31 return device_allocator_; in device_allocator()
80 DeviceMemoryAllocator* device_allocator_ = nullptr; variable
78 DeviceMemoryAllocator* device_allocator() const { return device_allocator_; } in device_allocator()80 device_allocator_ = device_allocator; in set_device_allocator()101 DeviceMemoryAllocator* device_allocator_ = nullptr;