| /external/tensorflow/tensorflow/core/util/ |
| D | device_name_utils_test.cc | 30 if (!DeviceNameUtils::ParseFullName(original, &p)) { in RoundTripParsedName()
74 EXPECT_FALSE(DeviceNameUtils::ParseFullName("foobar", &p)); in TEST()
75 EXPECT_FALSE(DeviceNameUtils::ParseFullName( in TEST()
78 DeviceNameUtils::ParseFullName("/job:123/replica:1/task:2/gpu:", &p)); in TEST()
79 EXPECT_FALSE(DeviceNameUtils::ParseFullName( in TEST()
81 EXPECT_FALSE(DeviceNameUtils::ParseFullName( in TEST()
83 EXPECT_FALSE(DeviceNameUtils::ParseFullName( in TEST()
86 DeviceNameUtils::ParseFullName("/job:foo/replica:1/task:2/bar:3", &p)); in TEST()
87 EXPECT_FALSE(DeviceNameUtils::ParseFullName( in TEST()
89 EXPECT_TRUE(DeviceNameUtils::ParseFullName( in TEST()
[all …]
|
| D | device_name_utils.cc | 114 bool DeviceNameUtils::ParseFullName(StringPiece fullname, ParsedName* p) { in ParseFullName() function in tensorflow::DeviceNameUtils
221 if (!ParseFullName(basename, &parsed_basename)) { in CanonicalizeDeviceName()
238 if (ParseFullName(fullname, &parsed_name)) { in CanonicalizeDeviceName()
420 return ParseFullName(src, &x) && ParseFullName(dst, &y) && in IsSameAddressSpace()
440 CHECK(ParseFullName(fullname, &x)) << fullname; in LocalName()
464 if (ParseFullName(name, &pn) && pn.has_type && pn.has_id) { in SplitDeviceName()
510 if (!DeviceNameUtils::ParseFullName(device_name, &device)) { in DeviceNameToCpuDeviceName()
|
| D | device_name_utils.h | 89 static bool ParseFullName(StringPiece fullname, ParsedName* parsed);
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | virtual_placer.cc | 67 bool parsed = DeviceNameUtils::ParseFullName(lfqn, &parsed_name); in VirtualPlacer()
98 bool parsed = DeviceNameUtils::ParseFullName(lfqn, &parsed_name); in VirtualPlacer()
147 bool parsed = DeviceNameUtils::ParseFullName(lowercase_name, &parsed_name); in to_lfqn_or_empty()
|
| D | utils.cc | 241 if (DeviceNameUtils::ParseFullName(device_str, &parsed)) { in GetDeviceInfo()
411 bool parsed = DeviceNameUtils::ParseFullName(device_name, &parsed_name); in GetDeviceClassForNonChannelDevice()
421 parsed = DeviceNameUtils::ParseFullName(name, &parsed_name); in GetDeviceClassForNonChannelDevice()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | remote_device.cc | 82 if (!DeviceNameUtils::ParseFullName(worker_name, &worker_name_parsed) || in NewRemoteDevices()
93 CHECK(DeviceNameUtils::ParseFullName(da.name(), &device_name_parsed)) in NewRemoteDevices()
|
| 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()
|
| D | device_resolver_distributed_test.cc | 190 ASSERT_TRUE(DeviceNameUtils::ParseFullName(dev_name, &parsed)); in TEST_F()
|
| D | graph_mgr.cc | 93 if (!DeviceNameUtils::ParseFullName(ndef.device(), &parsed)) { in ValidateGraphDefForDevices()
|
| /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 | 26 CHECK(DeviceNameUtils::ParseFullName(new_base, &parsed_name)); in NewRenamedDevice()
|
| D | colocation_graph.cc | 181 if (!DeviceNameUtils::ParseFullName(device_name, &assigned_device_name_)) { in SetAssignedDeviceName()
191 if (!DeviceNameUtils::ParseFullName(node.requested_device(), in SetRequestedDeviceName()
412 DeviceNameUtils::ParseFullName(node.assigned_device_name(), &parsed); in AssignDevice()
748 if (DeviceNameUtils::ParseFullName(node->requested_device(), in GetDevicesForNode()
|
| D | collective_param_resolver_local.cc | 111 DeviceNameUtils::ParseFullName(device, &parsed_device); in CompleteGroupLocal()
194 if (!DeviceNameUtils::ParseFullName(dr->device, &parsed_name)) { in ParseRingOrder()
226 CHECK(DeviceNameUtils::ParseFullName(next_device, &parsed_name)); in OrderTaskDeviceMap()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_cluster_util.cc | 89 if (!DeviceNameUtils::ParseFullName(device, &parsed)) { in DeviceToDeviceType()
285 TF_RET_CHECK(DeviceNameUtils::ParseFullName(device_name, &parsed_name)) in PickDeviceForXlaImpl()
|
| D | clone_constants_for_better_clustering.cc | 67 DeviceNameUtils::ParseFullName(n->assigned_device_name(), &parsed)); in IsSmallHostConstant()
|
| D | encapsulate_xla_computations_pass.cc | 53 if (DeviceNameUtils::ParseFullName(n->requested_device(), &parsed)) { in IsCpuGpuCompile()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | rendezvous.cc | 103 DeviceNameUtils::ParseFullName(parts[0], &out->src) && in ParseKey()
105 DeviceNameUtils::ParseFullName(parts[2], &out->dst) && in ParseKey()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | sharding_util.cc | 59 if (!DeviceNameUtils::ParseFullName(device_name, &parsed_device)) { in ParseShardingFromDevice()
|
| /external/tensorflow/tensorflow/core/grappler/ |
| D | grappler_item.cc | 139 if (!DeviceNameUtils::ParseFullName(device, &name)) { in AddDevice()
|
| D | utils.cc | 494 if (!DeviceNameUtils::ParseFullName(node.device(), &parsed_name)) { in IsKernelRegisteredForNode()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | grpc_channel.cc | 270 if (!DeviceNameUtils::ParseFullName(target, &parsed)) { in TranslateTask()
|
| /external/tensorflow/tensorflow/core/grappler/clusters/ |
| D | single_machine.cc | 91 if (!DeviceNameUtils::ParseFullName(dev.name(), &parsed)) { in Provision()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_graph.cc | 223 DeviceNameUtils::ParseFullName(node_device, &parsed_name); in GetEngineInfo()
763 if (DeviceNameUtils::ParseFullName(engine.device, &parsed_name) && in GetDeviceAndAllocator()
|