Home
last modified time | relevance | path

Searched refs:device_info_cache (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/jit/
Ddevice_util.cc101 const jit::DeviceInfoCache& device_info_cache, in PickDeviceForXlaImpl() argument
126 const absl::string_view name0 = device_info_cache.GetNameFor(d0); in PickDeviceForXlaImpl()
127 const absl::string_view name1 = device_info_cache.GetNameFor(d1->value()); in PickDeviceForXlaImpl()
149 if (device_info_cache.IsGpu(device)) { in PickDeviceForXlaImpl()
156 } else if (device_info_cache.IsCpu(device)) { in PickDeviceForXlaImpl()
176 "Multiple CPU devices ", device_info_cache.DebugString(devices))); in PickDeviceForXlaImpl()
181 "Multiple GPU devices ", device_info_cache.DebugString(devices))); in PickDeviceForXlaImpl()
186 "Multiple unknown devices ", device_info_cache.DebugString(devices))); in PickDeviceForXlaImpl()
192 device_info_cache.GetNameFor(*maybe_unknown_device), ", ", in PickDeviceForXlaImpl()
193 device_info_cache.GetNameFor(*maybe_gpu_device))); in PickDeviceForXlaImpl()
[all …]
Dbuild_xla_ops_pass.cc286 Status DeviceRequiresCompilation(const jit::DeviceInfoCache& device_info_cache, in DeviceRequiresCompilation() argument
289 device_info_cache.GetCompilationDevice(device); in DeviceRequiresCompilation()
347 jit::DeviceInfoCache* device_info_cache, Node* n, in InferDeviceForCluster() argument
358 device_info_cache->GetIdFor(ndef.device())); in InferDeviceForCluster()
368 device_info_cache->GetIdFor(n->assigned_device_name())); in InferDeviceForCluster()
373 PickDeviceForXla(*device_info_cache, device_set, in InferDeviceForCluster()
376 << device_info_cache->DebugString(device_set) << ") -> " in InferDeviceForCluster()
377 << device_info_cache->GetNameFor(result); in InferDeviceForCluster()
468 jit::DeviceInfoCache* device_info_cache, in ReplaceNodeWithXlaCompileAndXlaRun() argument
477 InferDeviceForCluster(device_info_cache, n, cluster_info.function.name(), in ReplaceNodeWithXlaCompileAndXlaRun()
[all …]
Ddevice_util_test.cc111 jit::DeviceInfoCache device_info_cache; in SimpleRoundTripTestForDeviceSet() local
119 device_info_cache.GetIdFor(device_name)); in SimpleRoundTripTestForDeviceSet()
125 actual_devices.push_back(string(device_info_cache.GetNameFor(device_id))); in SimpleRoundTripTestForDeviceSet()
Ddevice_util.h202 const jit::DeviceInfoCache& device_info_cache,
211 const jit::DeviceInfoCache& device_info_cache,
Dmark_for_compilation_pass.cc569 jit::DeviceInfoCache* device_info_cache, const Node& n, bool* ignore) { in IgnoreResourceOpForSafetyAnalysis() argument
596 device_info_cache->GetCompilationDevice(n.assigned_device_name())); in IgnoreResourceOpForSafetyAnalysis()