Searched refs:DeviceRegistration (Results 1 – 14 of 14) sorted by relevance
123 using DeviceRegistration = const XlaOpRegistry::DeviceRegistration; variable125 DeviceRegistration* GetCompilationDevice(DeviceId device) const { in GetCompilationDevice()129 xla::StatusOr<DeviceRegistration*> GetCompilationDevice( in GetCompilationDevice()154 std::vector<const XlaOpRegistry::DeviceRegistration*>
59 XlaOpRegistry::DeviceRegistration registration; in CreateDevices()
67 const XlaOpRegistry::DeviceRegistration* registration; in BuildXlaCompilationCache()
43 const XlaOpRegistry::DeviceRegistration* registration; in IsCompilable()
103 XlaOpRegistry::DeviceRegistration registration; in CreateDevices()
285 const XlaOpRegistry::DeviceRegistration& registration);
70 const XlaOpRegistry::DeviceRegistration* compilation_device; in GetIdFor()
240 const XlaOpRegistry::DeviceRegistration* registration; in IsMustCompileDevice()
595 const XlaOpRegistry::DeviceRegistration* registration, in IgnoreResourceOpForSafetyAnalysis()1191 const XlaOpRegistry::DeviceRegistration* registration; in FindCompilationCandidates()1633 const XlaOpRegistry::DeviceRegistration* registration = in ShouldCompileClusterImpl()
340 XlaOpRegistry::DeviceRegistration registration; in CreateDevices()
288 const XlaOpRegistry::DeviceRegistration* registration = in DeviceRequiresCompilation()
559 const XlaOpRegistry::DeviceRegistration& registration) { in CreateOperationFilter()
87 struct DeviceRegistration { struct154 const DeviceRegistration& registration);167 const DeviceRegistration** registration);244 std::unordered_map<string, DeviceRegistration> compilation_devices_
115 const string& device_name, const DeviceRegistration& registration) { in RegisterCompilationDevice()145 const string& device_name, const DeviceRegistration** registration) { in GetCompilationDevice()157 DeviceRegistration& registration = in GetCompilationDevice()166 DeviceRegistration& registration = in GetCompilationDevice()