/external/tensorflow/tensorflow/core/util/ |
D | device_name_utils_test.cc | 351 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()
|
D | device_name_utils.h | 163 static bool SplitDeviceName(StringPiece name, string* task, string* device);
|
D | device_name_utils.cc | 461 bool DeviceNameUtils::SplitDeviceName(StringPiece name, string* task, in SplitDeviceName() function in tensorflow::DeviceNameUtils
|
/external/tensorflow/tensorflow/contrib/verbs/ |
D | rdma_rendezvous_mgr.cc | 53 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/ |
D | worker_cache_partial.cc | 61 if (!DeviceNameUtils::SplitDeviceName(device_name, &task, &device)) { in RefreshDeviceStatus()
|
D | device_resolver_distributed_test.cc | 100 if (!DeviceNameUtils::SplitDeviceName(device, &task_name, &dev_part)) { in GetDeviceLocalityAsync()
|
D | collective_rma_distributed_test.cc | 138 if (!DeviceNameUtils::SplitDeviceName(device, &task_name, &dev_part)) { in GetDeviceLocalityAsync()
|
D | collective_param_resolver_distributed_test.cc | 98 if (!DeviceNameUtils::SplitDeviceName(device, &task_name, &dev_part)) { in GetDeviceLocalityAsync()
|
D | master_session.cc | 376 CHECK(DeviceNameUtils::SplitDeviceName(node->assigned_device_name(), &task, in SplitByWorker()
|
/external/tensorflow/tensorflow/contrib/gdr/ |
D | gdr_server_lib.cc | 67 DeviceNameUtils::SplitDeviceName( in Init()
|
D | gdr_rendezvous_mgr.cc | 136 if (!DeviceNameUtils::SplitDeviceName(parsed.src_device, &src_worker, in RecvFromRemoteAsync()
|
/external/tensorflow/tensorflow/core/grappler/optimizers/ |
D | implementation_selector.cc | 135 if (!DeviceNameUtils::SplitDeviceName(node_def->device(), &task, &device)) { in MaybeOptimizeFunctionCall()
|
D | memory_optimizer.cc | 717 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()
|
D | arithmetic_optimizer.cc | 2031 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()
|
D | layout_optimizer.cc | 580 if (DeviceNameUtils::SplitDeviceName(device_name, ¬_used, &device) && in IsOnGPU()
|
/external/tensorflow/tensorflow/core/grappler/ |
D | utils.cc | 221 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/ |
D | rpc_rendezvous_mgr.cc | 226 if (!DeviceNameUtils::SplitDeviceName(parsed.src_device, &call->src_worker_, in RecvFromRemoteAsync()
|
D | grpc_server_lib.cc | 168 if (!DeviceNameUtils::SplitDeviceName(master_env_.local_devices[0]->name(), in Init()
|
/external/tensorflow/tensorflow/core/common_runtime/eager/ |
D | context.cc | 390 if (!DeviceNameUtils::SplitDeviceName(d->name(), task_name, &ignored)) { in GetTaskName()
|