| /external/tensorflow/tensorflow/core/util/ |
| D | device_name_utils_test.cc | 30 if (!DeviceNameUtils::ParseFullName(original, &p)) { in RoundTripParsedName()
75 EXPECT_FALSE(DeviceNameUtils::ParseFullName("foobar", &p)); in TEST()
76 EXPECT_FALSE(DeviceNameUtils::ParseFullName( in TEST()
79 DeviceNameUtils::ParseFullName("/job:123/replica:1/task:2/gpu:", &p)); in TEST()
80 EXPECT_FALSE(DeviceNameUtils::ParseFullName( in TEST()
82 EXPECT_FALSE(DeviceNameUtils::ParseFullName( in TEST()
84 EXPECT_FALSE(DeviceNameUtils::ParseFullName( in TEST()
87 DeviceNameUtils::ParseFullName("/job:foo/replica:1/task:2/bar:3", &p)); in TEST()
88 EXPECT_FALSE(DeviceNameUtils::ParseFullName( in TEST()
90 EXPECT_TRUE(DeviceNameUtils::ParseFullName( in TEST()
[all …]
|
| D | device_name_utils.cc | 104 bool DeviceNameUtils::ParseFullName(StringPiece fullname, ParsedName* p) { in ParseFullName() function in tensorflow::DeviceNameUtils
211 if (!ParseFullName(basename, &parsed_basename)) { in CanonicalizeDeviceName()
228 if (ParseFullName(fullname, &parsed_name)) { in CanonicalizeDeviceName()
453 return ParseFullName(src, &x) && ParseFullName(dst, &y) && in IsSameAddressSpace()
494 CHECK(ParseFullName(fullname, &x)) << fullname; in LocalName()
518 if (ParseFullName(name, &pn) && pn.has_type && pn.has_id) { in SplitDeviceName()
579 if (!DeviceNameUtils::ParseFullName(device_name, &device)) { in DeviceNameToCpuDeviceName()
|
| D | device_name_utils.h | 97 static bool ParseFullName(StringPiece fullname, ParsedName* parsed);
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | virtual_placer.cc | 65 bool parsed = DeviceNameUtils::ParseFullName(lfqn, &parsed_name); in VirtualPlacer()
96 bool parsed = DeviceNameUtils::ParseFullName(lfqn, &parsed_name); in VirtualPlacer()
145 bool parsed = DeviceNameUtils::ParseFullName(lowercase_name, &parsed_name); in to_lfqn_or_empty()
|
| D | utils.cc | 242 if (DeviceNameUtils::ParseFullName(device_str, &parsed)) { in GetDeviceInfo()
394 bool parsed = DeviceNameUtils::ParseFullName(device_name, &parsed_name); in GetDeviceClassForNonChannelDevice()
404 parsed = DeviceNameUtils::ParseFullName(name, &parsed_name); in GetDeviceClassForNonChannelDevice()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | device_util.cc | 93 if (!DeviceNameUtils::ParseFullName(device, &parsed)) { in DeviceNameToDeviceType()
130 if (!DeviceNameUtils::ParseFullName(name0, &parsed0) || in PickDeviceForXlaImpl()
131 !DeviceNameUtils::ParseFullName(name1, &parsed1) || in PickDeviceForXlaImpl()
|
| D | clone_constants_for_better_clustering.cc | 66 DeviceNameUtils::ParseFullName(n->assigned_device_name(), &parsed)); in IsConstantOnHost()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | remote_device.cc | 99 if (!DeviceNameUtils::ParseFullName(worker_name, &worker_name_parsed) || in NewRemoteDevices()
110 CHECK(DeviceNameUtils::ParseFullName(da.name(), &device_name_parsed)) in NewRemoteDevices()
|
| D | device_resolver_distributed_test.cc | 210 ASSERT_TRUE(DeviceNameUtils::ParseFullName(device_name, &parsed)); in ResolveIncarnationsAndValidate()
247 ASSERT_TRUE(DeviceNameUtils::ParseFullName(dev_name, &parsed)); in TEST_F()
|
| D | master.cc | 161 if (DeviceNameUtils::ParseFullName(filter, &parsed)) { in DeviceFinder()
184 CHECK(DeviceNameUtils::ParseFullName(local_device_name, in DeviceFinder()
344 if (DeviceNameUtils::ParseFullName(name, &x)) { in MatchFilters()
|
| D | test_utils.h | 147 CHECK(DeviceNameUtils::ParseFullName(it.first, &device_name)); in ListWorkersInJob()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | rendezvous.cc | 105 DeviceNameUtils::ParseFullName(parts[0], &out->src) && in ParseKey()
107 DeviceNameUtils::ParseFullName(parts[2], &out->dst) && in ParseKey()
|
| /external/tensorflow/tensorflow/python/util/ |
| D | kernel_registry.cc | 45 if (!tensorflow::DeviceNameUtils::ParseFullName(node_def.device(), in TryFindKernelClass()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | device.cc | 28 CHECK(DeviceNameUtils::ParseFullName(name(), &parsed_name_)) in Device()
|
| D | renamed_device.cc | 32 CHECK(DeviceNameUtils::ParseFullName(new_base, &parsed_name)); in NewRenamedDevice()
|
| D | colocation_graph.cc | 157 if (!DeviceNameUtils::ParseFullName(device_name, &assigned_device_name_)) { in SetAssignedDeviceName()
173 if (!DeviceNameUtils::ParseFullName(node.requested_device(), in SetResourceDeviceName()
197 if (!DeviceNameUtils::ParseFullName(node.requested_device(), in SetRequestedDeviceName()
446 DeviceNameUtils::ParseFullName(node.assigned_device_name(), &parsed); in AssignDevice()
1082 if (DeviceNameUtils::ParseFullName(node->requested_device(), in GetDevicesForNode()
|
| D | collective_param_resolver_local.cc | 163 DeviceNameUtils::ParseFullName(device, &parsed_device); in CompleteGroupLocal()
261 if (!DeviceNameUtils::ParseFullName(dr->device, &parsed_name)) { in ParseRingOrder()
293 CHECK(DeviceNameUtils::ParseFullName(next_device, &parsed_name)); in OrderTaskDeviceMap()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | device_util.cc | 71 if (!DeviceNameUtils::ParseFullName(string_attr.getValue().str(), in GetDevicesFromOp()
|
| D | tpu_rewrite_device_util_test.cc | 38 if (!DeviceNameUtils::ParseFullName(device_name, &parsed_name)) in DeviceNamesToParsedNames()
|
| D | eval_util.cc | 68 if (!DeviceNameUtils::ParseFullName(node_def->device(), &parsed_name)) return; in ForceUseLocalhostDevice()
|
| D | device_util_test.cc | 51 DeviceNameUtils::ParseFullName(name, &parsed_name); in Make()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | sharding_util.cc | 59 if (!DeviceNameUtils::ParseFullName(device_name, &parsed_device)) { in ParseShardingFromDevice()
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | eager_operation.cc | 130 if (!DeviceNameUtils::ParseFullName(device, &device_parsed_name_)) { in SetDeviceName()
|
| D | context.cc | 838 if (!DeviceNameUtils::ParseFullName(remote_worker, &pw)) { in SetRemoteDeviceFilters()
854 if (DeviceNameUtils::ParseFullName(filter, &parsed_filter)) { in SetRemoteDeviceFilters()
889 DeviceNameUtils::ParseFullName(remote_worker, &parsed_remote_worker); in FilterDevicesForRemoteWorkers()
892 DeviceNameUtils::ParseFullName(device_attrs[i].name(), &pn); in FilterDevicesForRemoteWorkers()
|
| /external/tensorflow/tensorflow/core/grappler/ |
| D | grappler_item.cc | 152 if (!DeviceNameUtils::ParseFullName(device, &name)) { in AddDevice()
|