Searched refs:MultiDeviceAdapter (Results 1 – 2 of 2) sorted by relevance
67 class MultiDeviceAdapter : public DeviceMemoryAllocator {71 MultiDeviceAdapter(const Platform *platform, in MultiDeviceAdapter() function
90 StatusOr<std::unique_ptr<se::MultiDeviceAdapter>> CreateBFCAllocator( in CreateBFCAllocator()96 std::vector<se::MultiDeviceAdapter::AllocatorWithStream> allocators; in CreateBFCAllocator()140 return absl::make_unique<se::MultiDeviceAdapter>(platform, in CreateBFCAllocator()