| /external/tensorflow/tensorflow/core/util/ |
| device_name_utils_test.cc | 29 DeviceNameUtils::ParsedName p;
30 if (!DeviceNameUtils::ParseFullName(original, &p)) {
33 string round_tripped = DeviceNameUtils::ParsedNameToString(p);
69 EXPECT_EQ(DeviceNameUtils::FullName("hello", 1, 2, "CPU", 3),
73 DeviceNameUtils::ParsedName p;
74 EXPECT_FALSE(DeviceNameUtils::ParseFullName("foobar", &p));
75 EXPECT_FALSE(DeviceNameUtils::ParseFullName(
78 DeviceNameUtils::ParseFullName("/job:123/replica:1/task:2/gpu:", &p));
79 EXPECT_FALSE(DeviceNameUtils::ParseFullName(
81 EXPECT_FALSE(DeviceNameUtils::ParseFullName
[all...] |
| device_name_utils.cc | 102 string DeviceNameUtils::FullName(const string& job, int replica, int task,
114 bool DeviceNameUtils::ParseFullName(StringPiece fullname, ParsedName* p) {
188 string DeviceNameUtils::CanonicalizeDeviceName(StringPiece fullname) {
200 string DeviceNameUtils::ParsedNameToString(const ParsedName& pn) {
217 bool DeviceNameUtils::IsSpecification(const ParsedName& less_specific,
244 bool DeviceNameUtils::IsCompleteSpecification(const ParsedName& pattern,
258 Status DeviceNameUtils::MergeDevNames(ParsedName* target,
336 bool DeviceNameUtils::IsSameAddressSpace(const ParsedName& a,
344 bool DeviceNameUtils::IsSameAddressSpace(StringPiece src, StringPiece dst) {
352 string DeviceNameUtils::LocalName(StringPiece type, int id)
[all...] |
| device_name_utils.h | 46 class DeviceNameUtils {
|
| /external/tensorflow/tensorflow/core/grappler/clusters/ |
| utils.h | 33 DeviceProperties GetDeviceInfo(const DeviceNameUtils::ParsedName& device);
|
| utils.cc | 108 DeviceProperties GetDeviceInfo(const DeviceNameUtils::ParsedName& device) {
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| renamed_device.cc | 26 DeviceNameUtils::ParsedName parsed_name;
27 CHECK(DeviceNameUtils::ParseFullName(new_base, &parsed_name));
28 DeviceNameUtils::ParsedName underlying_parsed_name =
34 string name = DeviceNameUtils::FullName(parsed_name.job, parsed_name.replica,
|
| device_set.cc | 36 DeviceNameUtils::GetNamesForDeviceMappings(device->parsed_name())) {
41 void DeviceSet::FindMatchingDevices(const DeviceNameUtils::ParsedName& spec,
47 if (DeviceNameUtils::IsCompleteSpecification(spec, d->parsed_name())) {
|
| device.cc | 28 CHECK(DeviceNameUtils::ParseFullName(name(), &parsed_name_))
|
| device_set.h | 53 // Given a DeviceNameUtils::ParsedName (which may have some
56 void FindMatchingDevices(const DeviceNameUtils::ParsedName& spec,
|
| placer.cc | 258 Status s = DeviceNameUtils::MergeDevNames(&new_root_member.device_name,
307 if (DeviceNameUtils::HasSomeDetails(members_[node_root].device_name)) {
325 DeviceNameUtils::ParsedName soft_device_name =
343 DeviceNameUtils::ParsedName specified_device_name;
344 if (DeviceNameUtils::ParseFullName(node->requested_device(),
387 DeviceNameUtils::ParsedNameToString(
448 DeviceNameUtils::ParsedName device_name;
516 if (!DeviceNameUtils::ParseFullName(assigned_device_name,
566 if (!DeviceNameUtils::ParseFullName(node.requested_device(),
692 if (DeviceNameUtils::HasSomeDetails(source_parsed_name) &
[all...] |
| device.h | 63 const DeviceNameUtils::ParsedName& parsed_name() const {
162 DeviceNameUtils::ParsedName parsed_name_;
|
| device_mgr.cc | 34 DeviceNameUtils::GetNamesForDeviceMappings(d->parsed_name())) {
39 DeviceNameUtils::GetLocalNamesForDeviceMappings(d->parsed_name())) {
|
| /external/tensorflow/tensorflow/python/util/ |
| kernel_registry.cc | 44 tensorflow::DeviceNameUtils::ParsedName parsed_name;
45 if (!tensorflow::DeviceNameUtils::ParseFullName(node_def.device(),
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| remote_device.cc | 81 DeviceNameUtils::ParsedName worker_name_parsed;
82 if (!DeviceNameUtils::ParseFullName(worker_name, &worker_name_parsed) ||
92 DeviceNameUtils::ParsedName device_name_parsed;
93 CHECK(DeviceNameUtils::ParseFullName(da.name(), &device_name_parsed))
105 da_rewritten.set_name(DeviceNameUtils::FullName(
|
| base_rendezvous_mgr.h | 159 virtual bool IsSameWorker(DeviceNameUtils::ParsedName src,
160 DeviceNameUtils::ParsedName dst);
|
| master.cc | 158 DeviceNameUtils::ParsedName parsed;
159 if (DeviceNameUtils::ParseFullName(filter, &parsed)) {
251 std::vector<DeviceNameUtils::ParsedName> filters_;
283 bool Intersects(const DeviceNameUtils::ParsedName& x,
284 const DeviceNameUtils::ParsedName& y) {
295 DeviceNameUtils::ParsedName x;
296 if (DeviceNameUtils::ParseFullName(name, &x)) {
384 DeviceNameUtils::ParsedName name = d->parsed_name();
|
| worker_cache_partial.cc | 61 if (!DeviceNameUtils::SplitDeviceName(device_name, &task, &device)) {
|
| base_rendezvous_mgr.cc | 287 bool BaseRemoteRendezvous::IsSameWorker(DeviceNameUtils::ParsedName src,
288 DeviceNameUtils::ParsedName dst) {
289 return DeviceNameUtils::IsSameAddressSpace(src, dst);
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| virtual_placer.cc | 66 DeviceNameUtils::ParsedName parsed_name;
67 bool parsed = DeviceNameUtils::ParseFullName(lfqn, &parsed_name);
96 DeviceNameUtils::ParsedName parsed_name;
97 bool parsed = DeviceNameUtils::ParseFullName(lfqn, &parsed_name);
143 DeviceNameUtils::ParsedName parsed_name;
145 bool parsed = DeviceNameUtils::ParseFullName(lowercase_name, &parsed_name);
147 parsed = DeviceNameUtils::ParseLocalName(lowercase_name, &parsed_name);
|
| utils.cc | 203 DeviceNameUtils::ParsedName parsed;
204 if (DeviceNameUtils::ParseFullName(device_str, &parsed)) {
368 DeviceNameUtils::ParsedName parsed_name;
369 bool parsed = DeviceNameUtils::ParseFullName(device_name, &parsed_name);
379 parsed = DeviceNameUtils::ParseFullName(name, &parsed_name);
|
| /external/tensorflow/tensorflow/core/framework/ |
| rendezvous.h | 64 DeviceNameUtils::ParsedName src;
67 DeviceNameUtils::ParsedName dst;
|
| /external/tensorflow/tensorflow/contrib/verbs/ |
| rdma_rendezvous_mgr.cc | 53 if (!DeviceNameUtils::SplitDeviceName(parsed.src_device, &src_name,
55 !DeviceNameUtils::SplitDeviceName(parsed.dst_device, &dst_name,
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| sharding_util.cc | 59 DeviceNameUtils::ParsedName parsed_device;
60 if (!DeviceNameUtils::ParseFullName(device_name, &parsed_device)) {
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| grpc_channel.cc | 227 DeviceNameUtils::ParsedName parsed;
228 if (!DeviceNameUtils::ParseFullName(target, &parsed)) {
|
| grpc_channel_test.cc | 27 #define IsSameAddrSp DeviceNameUtils::IsSameAddressSpace
|
