Searched refs:GetCompilationDevice (Results 1 – 9 of 9) sorted by relevance
125 DeviceRegistration* GetCompilationDevice(DeviceId device) const { in GetCompilationDevice() function129 xla::StatusOr<DeviceRegistration*> GetCompilationDevice( in GetCompilationDevice() function132 return GetCompilationDevice(device_id); in GetCompilationDevice()
68 if (!XlaOpRegistry::GetCompilationDevice(platform_info.device_type().type(), in BuildXlaCompilationCache()
44 CHECK(XlaOpRegistry::GetCompilationDevice(device->device_type(), in IsCompilable()
71 if (!XlaOpRegistry::GetCompilationDevice(device_type->type(), in GetIdFor()
241 if (XlaOpRegistry::GetCompilationDevice(device_type.type(), ®istration)) { in IsMustCompileDevice()
596 device_info_cache->GetCompilationDevice(n.assigned_device_name())); in IgnoreResourceOpForSafetyAnalysis()1192 if (!XlaOpRegistry::GetCompilationDevice(device_type.type(), in FindCompilationCandidates()1634 device_info_cache_.GetCompilationDevice(chosen_device); in ShouldCompileClusterImpl()
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