Home
last modified time | relevance | path

Searched refs:SplitDeviceName (Results 1 – 19 of 19) sorted by relevance

/external/tensorflow/tensorflow/core/util/
Ddevice_name_utils_test.cc351 TEST(DeviceNameUtilsTest, SplitDeviceName) { in TEST() argument
354 EXPECT_TRUE(DeviceNameUtils::SplitDeviceName( in TEST()
358 EXPECT_TRUE(DeviceNameUtils::SplitDeviceName( in TEST()
363 DeviceNameUtils::SplitDeviceName("/device:GPU:3", &task, &device)); in TEST()
366 EXPECT_FALSE(DeviceNameUtils::SplitDeviceName("gpu:3", &task, &device)); in TEST()
367 EXPECT_FALSE(DeviceNameUtils::SplitDeviceName("/job:foo/task:2/replica:1", in TEST()
369 EXPECT_TRUE(DeviceNameUtils::SplitDeviceName("/device:myspecialdevice:3", in TEST()
Ddevice_name_utils.h163 static bool SplitDeviceName(StringPiece name, string* task, string* device);
Ddevice_name_utils.cc461 bool DeviceNameUtils::SplitDeviceName(StringPiece name, string* task, in SplitDeviceName() function in tensorflow::DeviceNameUtils
/external/tensorflow/tensorflow/contrib/verbs/
Drdma_rendezvous_mgr.cc53 if (!DeviceNameUtils::SplitDeviceName(parsed.src_device, &src_name, in RecvFromRemoteAsync()
55 !DeviceNameUtils::SplitDeviceName(parsed.dst_device, &dst_name, in RecvFromRemoteAsync()
/external/tensorflow/tensorflow/core/distributed_runtime/
Dworker_cache_partial.cc61 if (!DeviceNameUtils::SplitDeviceName(device_name, &task, &device)) { in RefreshDeviceStatus()
Ddevice_resolver_distributed_test.cc100 if (!DeviceNameUtils::SplitDeviceName(device, &task_name, &dev_part)) { in GetDeviceLocalityAsync()
Dcollective_rma_distributed_test.cc138 if (!DeviceNameUtils::SplitDeviceName(device, &task_name, &dev_part)) { in GetDeviceLocalityAsync()
Dcollective_param_resolver_distributed_test.cc98 if (!DeviceNameUtils::SplitDeviceName(device, &task_name, &dev_part)) { in GetDeviceLocalityAsync()
Dmaster_session.cc376 CHECK(DeviceNameUtils::SplitDeviceName(node->assigned_device_name(), &task, in SplitByWorker()
/external/tensorflow/tensorflow/contrib/gdr/
Dgdr_server_lib.cc67 DeviceNameUtils::SplitDeviceName( in Init()
Dgdr_rendezvous_mgr.cc136 if (!DeviceNameUtils::SplitDeviceName(parsed.src_device, &src_worker, in RecvFromRemoteAsync()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dimplementation_selector.cc135 if (!DeviceNameUtils::SplitDeviceName(node_def->device(), &task, &device)) { in MaybeOptimizeFunctionCall()
Dmemory_optimizer.cc717 if (!DeviceNameUtils::SplitDeviceName(node->device(), &task, &device) || in BuildSwapPair()
1236 DeviceNameUtils::SplitDeviceName(node1.device(), &task1, &device1); in CrossesTaskOrCpuGpuBoundary()
1239 DeviceNameUtils::SplitDeviceName(node2.device(), &task2, &device2); in CrossesTaskOrCpuGpuBoundary()
Darithmetic_optimizer.cc2031 return DeviceNameUtils::SplitDeviceName(node->device(), &task, &device) && in NodeIsOnCpuOrGpu()
2334 DeviceNameUtils::SplitDeviceName(node->device(), &task, &device) && in TrySimplify()
3002 return DeviceNameUtils::SplitDeviceName(node.device(), &task, &device) && in NodeIsOnCpu()
Dlayout_optimizer.cc580 if (DeviceNameUtils::SplitDeviceName(device_name, &not_used, &device) && in IsOnGPU()
/external/tensorflow/tensorflow/core/grappler/
Dutils.cc221 return DeviceNameUtils::SplitDeviceName(node->device(), &task, &device) && in NodeIsOnCpu()
227 return DeviceNameUtils::SplitDeviceName(node->device(), &task, &device) && in NodeIsOnGpu()
/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
Drpc_rendezvous_mgr.cc226 if (!DeviceNameUtils::SplitDeviceName(parsed.src_device, &call->src_worker_, in RecvFromRemoteAsync()
Dgrpc_server_lib.cc168 if (!DeviceNameUtils::SplitDeviceName(master_env_.local_devices[0]->name(), in Init()
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dcontext.cc390 if (!DeviceNameUtils::SplitDeviceName(d->name(), task_name, &ignored)) { in GetTaskName()