| /external/tensorflow/tensorflow/core/util/ |
| D | device_name_utils.h | 52 struct ParsedName { struct
66 bool operator==(const ParsedName& other) const { argument
97 static bool ParseFullName(StringPiece fullname, ParsedName* parsed); argument
109 static bool HasSomeDetails(const ParsedName& name) { in HasSomeDetails()
118 static bool IsSpecification(const ParsedName& less_specific,
119 const ParsedName& more_specific);
123 static void EnsureSpecification(ParsedName* more_specific,
124 const ParsedName& less_specific);
128 static bool IsCompleteSpecification(const ParsedName& pattern,
129 const ParsedName& name);
[all …]
|
| D | device_name_utils.cc | 104 bool DeviceNameUtils::ParseFullName(StringPiece fullname, ParsedName* p) { in ParseFullName()
179 void CompleteName(const DeviceNameUtils::ParsedName& parsed_basename, in CompleteName()
180 DeviceNameUtils::ParsedName* parsed_name) { in CompleteName()
210 ParsedName parsed_basename; in CanonicalizeDeviceName()
222 ParsedName parsed_name; in CanonicalizeDeviceName()
239 string DeviceNameUtils::ParsedNameToString(const ParsedName& pn) { in ParsedNameToString()
256 bool DeviceNameUtils::IsSpecification(const ParsedName& less_specific, in IsSpecification()
257 const ParsedName& more_specific) { in IsSpecification()
283 bool DeviceNameUtils::AreCompatibleDevNames(const ParsedName& a, in AreCompatibleDevNames()
284 const ParsedName& b) { in AreCompatibleDevNames()
[all …]
|
| D | device_name_utils_test.cc | 29 DeviceNameUtils::ParsedName p; in RoundTripParsedName()
74 DeviceNameUtils::ParsedName p; in TEST()
105 DeviceNameUtils::ParsedName p; in TEST()
121 DeviceNameUtils::ParsedName p; in TEST()
136 DeviceNameUtils::ParsedName p; in TEST()
147 DeviceNameUtils::ParsedName p; in TEST()
159 DeviceNameUtils::ParsedName p; in TEST()
171 DeviceNameUtils::ParsedName p; in TEST()
184 DeviceNameUtils::ParsedName p; in TEST()
193 DeviceNameUtils::ParsedName p; in TEST()
[all …]
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | colocation_graph.h | 42 const DeviceNameUtils::ParsedName* local_address_spec);
44 const DeviceNameUtils::ParsedName& requested_device_name() const { in requested_device_name()
104 DeviceNameUtils::ParsedName GetSoftDeviceName() const;
108 DeviceNameUtils::ParsedName GetPreferredSoftDeviceName() const;
152 DeviceNameUtils::ParsedName requested_device_name_;
158 DeviceNameUtils::ParsedName assigned_device_name_;
180 DeviceNameUtils::ParsedName resource_device_name_;
364 const DeviceNameUtils::ParsedName local_address_spec_;
|
| D | inspecting_placer.h | 40 DeviceNameUtils::ParsedName requested_device_name;
43 DeviceNameUtils::ParsedName resource_device_name;
|
| D | colocation_graph.cc | 140 const DeviceNameUtils::ParsedName* local_address_spec) { in SetParentAndSupportedDevices()
346 DeviceNameUtils::ParsedName assigned_device_name_copy = assigned_device_name_; in MergeDeviceNames()
350 DeviceNameUtils::ParsedName resource_device_name_copy = resource_device_name_; in MergeDeviceNames()
354 DeviceNameUtils::ParsedName requested_device_name_copy = in MergeDeviceNames()
445 DeviceNameUtils::ParsedName parsed; in AssignDevice()
530 DeviceNameUtils::ParsedName Member::GetSoftDeviceName() const { in GetSoftDeviceName()
531 DeviceNameUtils::ParsedName soft_device_name = requested_device_name_; in GetSoftDeviceName()
542 DeviceNameUtils::ParsedName Member::GetPreferredSoftDeviceName() const { in GetPreferredSoftDeviceName()
543 DeviceNameUtils::ParsedName soft_device_name = requested_device_name_; in GetPreferredSoftDeviceName()
559 static const DeviceNameUtils::ParsedName LocalAddressSpec( in LocalAddressSpec()
[all …]
|
| D | renamed_device.cc | 31 DeviceNameUtils::ParsedName parsed_name; in NewRenamedDevice()
33 DeviceNameUtils::ParsedName underlying_parsed_name = in NewRenamedDevice()
|
| D | device.h | 66 const DeviceNameUtils::ParsedName& parsed_name() const { in parsed_name()
197 DeviceNameUtils::ParsedName parsed_name_;
|
| D | device_set.h | 59 void FindMatchingDevices(const DeviceNameUtils::ParsedName& spec,
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | device_util_test.cc | 50 DeviceNameUtils::ParsedName parsed_name; in Make()
101 llvm::SmallVector<DeviceNameUtils::ParsedName, 8> devices; in TEST()
112 llvm::SmallVector<DeviceNameUtils::ParsedName, 8> devices; in TEST()
123 llvm::SmallVector<DeviceNameUtils::ParsedName, 8> devices; in TEST()
134 llvm::SmallVector<DeviceNameUtils::ParsedName, 8> devices; in TEST()
147 llvm::SmallVector<DeviceNameUtils::ParsedName, 8> devices; in TEST()
|
| D | tpu_rewrite_device_util.cc | 35 using Device = DeviceNameUtils::ParsedName;
36 using Devices = llvm::ArrayRef<DeviceNameUtils::ParsedName>;
|
| D | device_util.h | 37 llvm::SmallVectorImpl<DeviceNameUtils::ParsedName>* devices);
|
| D | tpu_rewrite_device_util.h | 55 llvm::ArrayRef<DeviceNameUtils::ParsedName> devices, int num_replicas,
|
| D | device_util.cc | 54 llvm::SmallVectorImpl<DeviceNameUtils::ParsedName>* devices) { in GetDevicesFromOp()
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | virtual_placer.cc | 64 DeviceNameUtils::ParsedName parsed_name; in VirtualPlacer()
95 DeviceNameUtils::ParsedName parsed_name; in VirtualPlacer()
143 DeviceNameUtils::ParsedName parsed_name; in to_lfqn_or_empty()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | rendezvous.h | 59 DeviceNameUtils::ParsedName src;
62 DeviceNameUtils::ParsedName dst;
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | master.cc | 160 DeviceNameUtils::ParsedName parsed; in DeviceFinder()
183 DeviceNameUtils::ParsedName local_parsed_name; in DeviceFinder()
188 for (const DeviceNameUtils::ParsedName& filter : filters_) { in DeviceFinder()
298 std::vector<DeviceNameUtils::ParsedName> filters_;
331 bool Intersects(const DeviceNameUtils::ParsedName& x, in Intersects()
332 const DeviceNameUtils::ParsedName& y) { in Intersects()
343 DeviceNameUtils::ParsedName x; in MatchFilters()
432 DeviceNameUtils::ParsedName name = d->parsed_name(); in CreateSession()
|
| D | remote_device.cc | 98 DeviceNameUtils::ParsedName worker_name_parsed; in NewRemoteDevices()
109 DeviceNameUtils::ParsedName device_name_parsed; in NewRemoteDevices()
|
| D | base_rendezvous_mgr.h | 159 virtual bool IsSameWorker(DeviceNameUtils::ParsedName src,
160 DeviceNameUtils::ParsedName dst);
|
| /external/tensorflow/tensorflow/core/grappler/clusters/ |
| D | utils.h | 34 DeviceProperties GetDeviceInfo(const DeviceNameUtils::ParsedName& device);
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | context.cc | 171 Status EagerContext::SelectDevice(DeviceNameUtils::ParsedName preferred, in SelectDevice()
208 DeviceNameUtils::ParsedName soft_device_name = preferred; in SelectDevice()
730 Status EagerContext::GetClient(const DeviceNameUtils::ParsedName& device_name, in GetClient()
837 DeviceNameUtils::ParsedName pw; in SetRemoteDeviceFilters()
851 std::vector<DeviceNameUtils::ParsedName> parsed_filters; in SetRemoteDeviceFilters()
853 DeviceNameUtils::ParsedName parsed_filter; in SetRemoteDeviceFilters()
887 const std::vector<DeviceNameUtils::ParsedName>& parsed_filters = it->second; in FilterDevicesForRemoteWorkers()
888 DeviceNameUtils::ParsedName parsed_remote_worker; in FilterDevicesForRemoteWorkers()
891 DeviceNameUtils::ParsedName pn; in FilterDevicesForRemoteWorkers()
|
| D | eager_operation.h | 87 const DeviceNameUtils::ParsedName& GetDeviceParsedName() const { in GetDeviceParsedName()
142 DeviceNameUtils::ParsedName device_parsed_name_;
|
| D | context_test.cc | 91 DeviceNameUtils::ParsedName requested; in TEST_F()
137 DeviceNameUtils::ParsedName requested; in TEST_F()
|
| /external/tensorflow/tensorflow/python/util/ |
| D | kernel_registry.cc | 44 tensorflow::DeviceNameUtils::ParsedName parsed_name; in TryFindKernelClass()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | device_util.cc | 92 DeviceNameUtils::ParsedName parsed; in DeviceNameToDeviceType()
129 DeviceNameUtils::ParsedName parsed0, parsed1; in PickDeviceForXlaImpl()
|