Searched refs:DeviceInfoCache (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | device_util.h | 32 class DeviceInfoCache; variable
55 friend class DeviceInfoCache; variable
112 class DeviceInfoCache {
202 const jit::DeviceInfoCache& device_info_cache,
211 const jit::DeviceInfoCache& device_info_cache,
|
| D | device_util.cc | 51 xla::StatusOr<DeviceId> DeviceInfoCache::GetIdFor(absl::string_view name) { in GetIdFor()
80 string DeviceInfoCache::DebugString(const DeviceSet& device_set) const { in DebugString()
101 const jit::DeviceInfoCache& device_info_cache, in PickDeviceForXlaImpl()
219 const jit::DeviceInfoCache& device_info_cache, in PickDeviceForXla()
229 const jit::DeviceInfoCache& device_info_cache, in MaybePickDeviceForXla()
|
| D | device_util_test.cc | 27 jit::DeviceInfoCache cache; in PickDeviceHelper()
111 jit::DeviceInfoCache device_info_cache; in SimpleRoundTripTestForDeviceSet()
|
| D | build_xla_ops_pass.cc | 286 Status DeviceRequiresCompilation(const jit::DeviceInfoCache& device_info_cache, in DeviceRequiresCompilation()
347 jit::DeviceInfoCache* device_info_cache, Node* n, in InferDeviceForCluster()
468 jit::DeviceInfoCache* device_info_cache, in ReplaceNodeWithXlaCompileAndXlaRun()
594 jit::DeviceInfoCache device_info_cache; in Run()
|
| D | mark_for_compilation_pass.cc | 430 jit::DeviceInfoCache device_info_cache_;
569 jit::DeviceInfoCache* device_info_cache, const Node& n, bool* ignore) { in IgnoreResourceOpForSafetyAnalysis()
|