Searched refs:device_allocator_ (Results 1 – 4 of 4) sorted by relevance
39 device_allocator_(device_allocator) {} in XlaPlatformInfo()49 return device_allocator_; in custom_allocator()77 se::DeviceMemoryAllocator* device_allocator_; variable
27 device_allocator_ = allocator; in set_device_allocator()32 return device_allocator_; in device_allocator()
146 se::DeviceMemoryAllocator* device_allocator_ = nullptr; variable
87 return device_allocator_; in device_allocator()90 device_allocator_ = device_allocator; in set_device_allocator()126 se::DeviceMemoryAllocator* device_allocator_ = nullptr;