| /external/tensorflow/tensorflow/core/util/ |
| D | device_name_utils.h | 52 struct ParsedName { struct
66 bool operator==(const ParsedName& other) const { argument
77 bool operator!=(const ParsedName& other) const {
95 static bool ParseFullOrLocalName(StringPiece fullname, ParsedName* parsed); argument
106 static bool ParseFullName(StringPiece fullname, ParsedName* parsed);
118 static bool HasSomeDetails(const ParsedName& name) { in HasSomeDetails()
127 static bool IsSpecification(const ParsedName& less_specific,
128 const ParsedName& more_specific);
132 static void EnsureSpecification(ParsedName* more_specific,
133 const ParsedName& less_specific);
[all …]
|
| D | device_name_utils.cc | 104 bool DeviceNameUtils::ParseFullName(StringPiece fullname, ParsedName* p) { in ParseFullName()
178 ParsedName* p) { in ParseFullOrLocalName()
184 void CompleteName(const DeviceNameUtils::ParsedName& parsed_basename, in CompleteName()
185 DeviceNameUtils::ParsedName* parsed_name) { in CompleteName()
215 ParsedName parsed_basename; in CanonicalizeDeviceName()
227 ParsedName parsed_name; in CanonicalizeDeviceName()
244 string DeviceNameUtils::ParsedNameToString(const ParsedName& pn) { in ParsedNameToString()
261 bool DeviceNameUtils::IsSpecification(const ParsedName& less_specific, in IsSpecification()
262 const ParsedName& more_specific) { in IsSpecification()
288 bool DeviceNameUtils::AreCompatibleDevNames(const ParsedName& a, 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()
123 DeviceNameUtils::ParsedName p; in TEST()
138 DeviceNameUtils::ParsedName p; in TEST()
149 DeviceNameUtils::ParsedName p; in TEST()
161 DeviceNameUtils::ParsedName p; in TEST()
173 DeviceNameUtils::ParsedName p; in TEST()
186 DeviceNameUtils::ParsedName p; in TEST()
195 DeviceNameUtils::ParsedName p; in TEST()
[all …]
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_structs.h | 38 using ParsedName = ::tensorflow::DeviceNameUtils::ParsedName; variable
42 void AddDevice(const ParsedName& device);
45 void AddGpuDevice(const ParsedName& device,
48 llvm::ArrayRef<ParsedName> device_names() const { return device_names_; } in device_names()
53 const ParsedName& device) const;
56 llvm::SmallVector<ParsedName, 8> device_names_;
|
| D | tf_structs.cc | 23 void RuntimeDevices::AddDevice(const ParsedName& device) { in AddDevice()
27 void RuntimeDevices::AddGpuDevice(const ParsedName& device, in AddGpuDevice()
34 const ParsedName& device) const { in GetGpuDeviceMetadata()
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | custom_device.cc | 43 const DeviceNameUtils::ParsedName* parsed = ParsedName(status); in DeviceType()
51 const DeviceNameUtils::ParsedName* parsed = ParsedName(status); in DeviceId()
71 const DeviceNameUtils::ParsedName* CustomDeviceTensorHandle::ParsedName( in ParsedName() function in tensorflow::CustomDeviceTensorHandle
74 DeviceNameUtils::ParsedName parsed_name; in ParsedName()
|
| D | custom_device.h | 113 const DeviceNameUtils::ParsedName* ParsedName(Status* status) const;
119 mutable absl::optional<DeviceNameUtils::ParsedName> parsed_name_;
|
| /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()
108 DeviceNameUtils::ParsedName GetSoftDeviceName() const;
112 DeviceNameUtils::ParsedName GetPreferredSoftDeviceName() const;
156 DeviceNameUtils::ParsedName requested_device_name_;
162 DeviceNameUtils::ParsedName assigned_device_name_;
184 DeviceNameUtils::ParsedName resource_device_name_;
376 const DeviceNameUtils::ParsedName local_address_spec_;
|
| D | composite_device.cc | 27 const DeviceNameUtils::ParsedName& host_name, Status* status) { in MakeDevice()
28 DeviceNameUtils::ParsedName parsed_name = host_name; in MakeDevice()
43 DeviceNameUtils::ParsedName parsed_name; in MakeDevice()
52 DeviceNameUtils::ParsedName name; in MakeDevice()
|
| D | inspecting_placer.h | 40 DeviceNameUtils::ParsedName requested_device_name;
43 DeviceNameUtils::ParsedName resource_device_name;
|
| D | colocation_graph.cc | 147 const DeviceNameUtils::ParsedName* local_address_spec) { in SetParentAndSupportedDevices()
230 [](const DeviceNameUtils::ParsedName& src_device, in IsEdgeFromCompositeDeviceToPhysicalDevice()
231 const DeviceNameUtils::ParsedName& dst_device) -> bool { in IsEdgeFromCompositeDeviceToPhysicalDevice()
373 DeviceNameUtils::ParsedName assigned_device_name_copy = assigned_device_name_; in MergeDeviceNames()
377 DeviceNameUtils::ParsedName resource_device_name_copy = resource_device_name_; in MergeDeviceNames()
381 DeviceNameUtils::ParsedName requested_device_name_copy = in MergeDeviceNames()
472 DeviceNameUtils::ParsedName parsed; in AssignDevice()
560 DeviceNameUtils::ParsedName Member::GetSoftDeviceName() const { in GetSoftDeviceName()
561 DeviceNameUtils::ParsedName soft_device_name = requested_device_name_; in GetSoftDeviceName()
572 DeviceNameUtils::ParsedName Member::GetPreferredSoftDeviceName() const { in GetPreferredSoftDeviceName()
[all …]
|
| D | renamed_device.cc | 31 DeviceNameUtils::ParsedName parsed_name; in NewRenamedDevice()
33 DeviceNameUtils::ParsedName underlying_parsed_name = in NewRenamedDevice()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | utils.cc | 269 absl::optional<DeviceNameUtils::ParsedName> GetDeviceParsedName( in GetDeviceParsedName()
272 DeviceNameUtils::ParsedName parsed_name; in GetDeviceParsedName()
279 absl::optional<DeviceNameUtils::ParsedName> MergeIfCompatible( in MergeIfCompatible()
280 const DeviceNameUtils::ParsedName& a, in MergeIfCompatible()
281 const DeviceNameUtils::ParsedName& b) { in MergeIfCompatible()
282 DeviceNameUtils::ParsedName merged_name = a; in MergeIfCompatible()
291 absl::optional<DeviceNameUtils::ParsedName> MergeIfCompatible( in MergeIfCompatible()
292 const DeviceNameUtils::ParsedName& a, absl::string_view b) { in MergeIfCompatible()
293 DeviceNameUtils::ParsedName b_parsed_name; in MergeIfCompatible()
|
| D | utils.h | 191 absl::optional<DeviceNameUtils::ParsedName> GetDeviceParsedName(
196 absl::optional<DeviceNameUtils::ParsedName> MergeIfCompatible(
197 const DeviceNameUtils::ParsedName& a, const DeviceNameUtils::ParsedName& b);
200 absl::optional<DeviceNameUtils::ParsedName> MergeIfCompatible(
201 const DeviceNameUtils::ParsedName& a, absl::string_view b);
|
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | distributed_tpu_rewrite_helpers.h | 41 DeviceNameUtils::ParsedName* system_spec,
50 const DeviceNameUtils::ParsedName& system_spec,
61 static Status GetTPUDevices(const DeviceNameUtils::ParsedName& system_spec,
|
| D | distributed_tpu_rewrite_helpers.cc | 31 DeviceNameUtils::ParsedName* system_spec, Device** system_device) { in GetSystemDevice()
97 const DeviceNameUtils::ParsedName& system_spec, const DeviceSet& device_set, in GetHostSystemDevices()
99 DeviceNameUtils::ParsedName host_spec; in GetHostSystemDevices()
151 const DeviceNameUtils::ParsedName& system_spec, const DeviceSet& device_set, in GetTPUDevices()
171 DeviceNameUtils::ParsedName device_spec = device->parsed_name(); in GetTPUDevices()
219 DeviceNameUtils::ParsedName spec; in ForConfigurationNodeMatchingType()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | tpu_compile_op_replication_pass.cc | 29 using ParsedName = ::tensorflow::DeviceNameUtils::ParsedName; typedef
37 ParsedName parsed_name; in GetHost()
|
| /external/llvm-project/lldb/source/Plugins/Language/CPlusPlus/ |
| D | CPlusPlusNameParser.h | 27 struct ParsedName { struct
33 ParsedName name; argument
57 llvm::Optional<ParsedName> ParseAsFullName();
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/segment/ |
| D | union_find.h | 129 const DeviceNameUtils::ParsedName& device_name);
137 const DeviceNameUtils::ParsedName& DeviceName() const { return device_name_; } in DeviceName()
143 DeviceNameUtils::ParsedName device_name_;
|
| D | union_find.cc | 128 const DeviceNameUtils::ParsedName& device_name) in ClusterProperty()
138 absl::optional<DeviceNameUtils::ParsedName> merged_device_name = in Merge()
|
| /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/compiler/mlir/tensorflow/utils/ |
| D | device_util.cc | 68 DeviceNameUtils::ParsedName device; in GetDevicesFromOp()
94 DeviceNameUtils::ParsedName device; in GetDevicesFromOp()
164 DeviceNameUtils::ParsedName parsed_name; in GetDeviceOrdinalFromDeviceString()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | device.h | 66 const DeviceNameUtils::ParsedName& parsed_name() const { in parsed_name()
192 DeviceNameUtils::ParsedName parsed_name_;
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | host_compute_ops.cc | 58 DeviceNameUtils::ParsedName parsed_name; in RecvAtHostOp()
89 DeviceNameUtils::ParsedName parsed_name; in ComputeAsync()
215 DeviceNameUtils::ParsedName parsed_name; in SendFromHostOp()
244 DeviceNameUtils::ParsedName parsed_name; in Compute()
|
| /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()
|