Searched defs:device_info_cache (Results 1 – 4 of 4) sorted by relevance
100 const jit::DeviceInfoCache& device_info_cache, in PickDeviceForXlaImpl()218 const jit::DeviceInfoCache& device_info_cache, in PickDeviceForXla()228 const jit::DeviceInfoCache& device_info_cache, in MaybePickDeviceForXla()
286 Status DeviceRequiresCompilation(const jit::DeviceInfoCache& device_info_cache, in DeviceRequiresCompilation()347 jit::DeviceInfoCache* device_info_cache, Node* n, in InferDeviceForCluster()467 jit::DeviceInfoCache* device_info_cache, in ReplaceNodeWithXlaCompileAndXlaRun()593 jit::DeviceInfoCache device_info_cache; in Run() local
111 jit::DeviceInfoCache device_info_cache; in SimpleRoundTripTestForDeviceSet() local
593 jit::DeviceInfoCache* device_info_cache, const Node& n, bool* ignore) { in IgnoreResourceOpForSafetyAnalysis()