Searched refs:GetDeviceToLogicalIdMap (Results 1 – 2 of 2) sorted by relevance
63 absl::flat_hash_map<GlobalDeviceId, LogicalID> GetDeviceToLogicalIdMap()
76 DeviceAssignment::GetDeviceToLogicalIdMap() const { in GetDeviceToLogicalIdMap() function in xla::DeviceAssignment