Searched refs:shared_memory_allocator (Results 1 – 2 of 2) sorted by relevance
92 std::shared_ptr<se::DeviceMemoryAllocator> shared_memory_allocator() const { in shared_memory_allocator() function
143 xla_device_metadata->client()->backend().shared_memory_allocator(); in XlaPlatformInfoFromDevice()