Searched refs:GetCompilationDevice (Results 1 – 8 of 8) sorted by relevance
125 DeviceRegistration* GetCompilationDevice(DeviceId device) const { in GetCompilationDevice() function129 StatusOr<DeviceRegistration*> GetCompilationDevice(absl::string_view name) { in GetCompilationDevice() function131 return GetCompilationDevice(device_id); in GetCompilationDevice()
620 device_info_cache->GetCompilationDevice(n.assigned_device_name())); in IgnoreResourceOpForSafetyAnalysis()1303 if (!XlaOpRegistry::GetCompilationDevice(device_type.type(), in FindCompilationCandidates()1760 device_info_cache_.GetCompilationDevice(chosen_device); in ShouldCompileClusterImpl()
106 if (!XlaOpRegistry::GetCompilationDevice(platform_info.device_type().type(), in BuildXlaCompilationCache()
70 if (!XlaOpRegistry::GetCompilationDevice(device_type->type(), in GetIdFor()
239 if (XlaOpRegistry::GetCompilationDevice(device_type.type(), ®istration)) { in IsMustCompileDevice()
289 device_info_cache.GetCompilationDevice(device); in DeviceRequiresCompilation()
166 static bool GetCompilationDevice(const string& device_name,
144 /* static */ bool XlaOpRegistry::GetCompilationDevice( in GetCompilationDevice() function in tensorflow::XlaOpRegistry