Searched refs:DeviceRegistration (Results 1 – 9 of 9) sorted by relevance
84 struct DeviceRegistration { struct119 const DeviceRegistration& registration);128 const DeviceRegistration** registration);200 std::unordered_map<string, DeviceRegistration> compilation_devices_
109 const string& device_name, const DeviceRegistration& registration) { in RegisterCompilationDevice()132 const string& device_name, const DeviceRegistration** registration) { in GetCompilationDevice()144 DeviceRegistration& registration = in GetCompilationDevice()154 DeviceRegistration& registration = in GetCompilationDevice()
501 const XlaOpRegistry::DeviceRegistration* registration; in FindCompilationCandidates()655 const XlaOpRegistry::DeviceRegistration* registration; in IsCompilable()715 const XlaOpRegistry::DeviceRegistration* registration; in Run()935 const XlaOpRegistry::DeviceRegistration* registration; in IgnoreResourceOpForSafetyAnalysis()1001 const XlaOpRegistry::DeviceRegistration* registration; in AreDevicesCompatible()1032 const XlaOpRegistry::DeviceRegistration* registration; in ShouldCompileClusterImpl()
46 XlaOpRegistry::DeviceRegistration registration; in CreateDevices()
43 XlaOpRegistry::DeviceRegistration registration; in CreateDevices()
65 XlaOpRegistry::DeviceRegistration registration; in CreateDevices()
228 const XlaOpRegistry::DeviceRegistration* registration; in IsMustCompileDevice()
221 const XlaOpRegistry::DeviceRegistration* registration = nullptr; in DeviceRequiresCompilation()
243 const XlaOpRegistry::DeviceRegistration* registration; in BuildCompilationCache()