Searched refs:GetDeviceTypeFor (Results 1 – 2 of 2) sorted by relevance
135 const DeviceType& GetDeviceTypeFor(DeviceId device) const { in GetDeviceTypeFor() function141 xla::StatusOr<DeviceTypeConstRef> GetDeviceTypeFor( in GetDeviceTypeFor() function
1181 device_info_cache_.GetDeviceTypeFor(node->assigned_device_name())); in FindCompilationCandidates()1632 device_info_cache_.GetDeviceTypeFor(chosen_device); in ShouldCompileClusterImpl()